trongld commited on
Commit
c3c0da2
·
1 Parent(s): 7ebc204

Update LangGraphAgent to use ChatGoogleGenerativeAI

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -3,7 +3,7 @@ from typing import Optional
3
 
4
  from langgraph.graph import StateGraph, START, END, MessagesState
5
  from langchain_core.messages import HumanMessage, AIMessage, SystemMessage
6
- from langchain_google_genai import ChatGoogleGenAI
7
  from langgraph.prebuilt import ToolNode
8
  from pydantic import Field, SecretStr # add
9
 
@@ -74,7 +74,7 @@ class LangGraphAgent:
74
 
75
  # Choose an LLM if not provided
76
  if model is None:
77
- model = ChatGoogleGenAI(
78
  model="gemini-2.5-flash",
79
  )
80
  if model is None and ChatOpenAI is not None:
 
3
 
4
  from langgraph.graph import StateGraph, START, END, MessagesState
5
  from langchain_core.messages import HumanMessage, AIMessage, SystemMessage
6
+ from langchain_google_genai import ChatGoogleGenerativeAI
7
  from langgraph.prebuilt import ToolNode
8
  from pydantic import Field, SecretStr # add
9
 
 
74
 
75
  # Choose an LLM if not provided
76
  if model is None:
77
+ model = ChatGoogleGenerativeAI(
78
  model="gemini-2.5-flash",
79
  )
80
  if model is None and ChatOpenAI is not None: