philincloud commited on
Commit
66ffcbf
·
verified ·
1 Parent(s): d64b354

Update langgraph_agent.py

Browse files
Files changed (1) hide show
  1. langgraph_agent.py +1 -1
langgraph_agent.py CHANGED
@@ -87,7 +87,7 @@ 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,
93
  api_key=GEMINI_API_KEY,
 
87
  """Build the LangGraph agent with chosen LLM (default: Gemini)."""
88
  if provider == "gemini":
89
  llm = ChatGoogleGenerativeAI(
90
+ model= "gemini-2.5-flash-preview-05-20",
91
  temperature=1.0,
92
  max_retries=2,
93
  api_key=GEMINI_API_KEY,