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 |
|