Update langgraph_agent.py
Browse files- langgraph_agent.py +1 -1
langgraph_agent.py
CHANGED
|
@@ -86,7 +86,7 @@ sys_msg = SystemMessage(content=system_prompt)
|
|
| 86 |
def build_graph(provider: str = "gemini"):
|
| 87 |
"""Build the LangGraph agent with chosen LLM (default: Gemini)."""
|
| 88 |
if provider == "gemini":
|
| 89 |
-
|
| 90 |
model= "gemini-2.5-pro-preview-05-06",
|
| 91 |
temperature=1.0,
|
| 92 |
max_retries=2,
|
|
|
|
| 86 |
def build_graph(provider: str = "gemini"):
|
| 87 |
"""Build the LangGraph agent with chosen LLM (default: Gemini)."""
|
| 88 |
if provider == "gemini":
|
| 89 |
+
llm = ChatGoogleGenerativeAI(
|
| 90 |
model= "gemini-2.5-pro-preview-05-06",
|
| 91 |
temperature=1.0,
|
| 92 |
max_retries=2,
|