KhalilWang commited on
Commit
5d25dec
·
verified ·
1 Parent(s): e9c2329

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -151,10 +151,10 @@ tools = [
151
  ]
152
 
153
  # Build graph function
154
- def build_graph(provider: str = "google"):
155
  """Build the graph"""
156
  # Load environment variables from .env file
157
- if provider == "google":
158
  # Google Gemini
159
  llm = ChatOpenAI(
160
  model="deepseek-chat",
 
151
  ]
152
 
153
  # Build graph function
154
+ def build_graph(provider: str = "deepseek"):
155
  """Build the graph"""
156
  # Load environment variables from .env file
157
+ if provider == "deepseek":
158
  # Google Gemini
159
  llm = ChatOpenAI(
160
  model="deepseek-chat",