Spaces:
Sleeping
Sleeping
Update LangGraphAgent to use ChatGoogleGenerativeAI
Browse files
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
|
| 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 =
|
| 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:
|