philincloud commited on
Commit
3ea980f
·
verified ·
1 Parent(s): 8a158c4

Update langgraph_agent.py

Browse files
Files changed (1) hide show
  1. 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 = "openai"):
87
  """Build the LangGraph agent with chosen LLM (default: Gemini)."""
88
- if provider == "openai":
89
  lllm = ChatGoogleGenerativeAI(
90
  model= "gemini-2.5-pro-preview-05-06",
91
  temperature=1.0,
92
  max_retries=2,
93
- google_api_key=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