supratipb commited on
Commit
f31ff03
Β·
verified Β·
1 Parent(s): d5e9128

Upload agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -177,11 +177,11 @@ tools = [
177
  ]
178
 
179
  # Build graph function
180
- def build_graph(provider: str = "groq"):
181
  """Build the graph"""
182
  # Load environment variables from .env file
183
  if provider == "openai":
184
- llm = ChatOpenAI(model_name="gpt-4", temperature=0)
185
  elif provider == "anthropic":
186
  llm = ChatAnthropic(model="claude-v1", temperature=0)
187
  elif provider == "google":
 
177
  ]
178
 
179
  # Build graph function
180
+ def build_graph(provider: str = "openai"):
181
  """Build the graph"""
182
  # Load environment variables from .env file
183
  if provider == "openai":
184
+ llm = ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0)
185
  elif provider == "anthropic":
186
  llm = ChatAnthropic(model="claude-v1", temperature=0)
187
  elif provider == "google":