Update langgraph_agent.py
Browse files- langgraph_agent.py +3 -3
langgraph_agent.py
CHANGED
@@ -83,14 +83,14 @@ with open("prompt.txt", "r", encoding="utf-8") as f:
|
|
83 |
sys_msg = SystemMessage(content=system_prompt)
|
84 |
|
85 |
|
86 |
-
def build_graph(provider: str = "
|
87 |
"""Build the LangGraph agent with chosen LLM (default: Gemini)."""
|
88 |
-
if provider == "
|
89 |
lllm = ChatGoogleGenerativeAI(
|
90 |
model= "gemini-2.5-pro-preview-05-06",
|
91 |
temperature=1.0,
|
92 |
max_retries=2,
|
93 |
-
google_api_key=
|
94 |
)
|
95 |
|
96 |
|
|
|
83 |
sys_msg = SystemMessage(content=system_prompt)
|
84 |
|
85 |
|
86 |
+
def build_graph(provider: str = "gemini"):
|
87 |
"""Build the LangGraph agent with chosen LLM (default: Gemini)."""
|
88 |
+
if provider == "gemini":
|
89 |
lllm = ChatGoogleGenerativeAI(
|
90 |
model= "gemini-2.5-pro-preview-05-06",
|
91 |
temperature=1.0,
|
92 |
max_retries=2,
|
93 |
+
google_api_key=GEMINI_API_KEY,
|
94 |
)
|
95 |
|
96 |
|