Jialun He commited on
Commit
c97e291
·
1 Parent(s): 1ff2bf8

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -159,7 +159,7 @@ tools = [
159
  ]
160
 
161
  # Build graph function
162
- def build_graph(provider: str = "google"):
163
  """Build the graph"""
164
  # Load environment variables from .env file
165
  if provider == "google":
@@ -170,7 +170,7 @@ def build_graph(provider: str = "google"):
170
  llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
171
  elif provider == "openai":
172
  # OpenAI
173
- llm = ChatOpenAI(model="gpt-4", temperature=0)
174
  elif provider == "huggingface":
175
  llm = ChatHuggingFace(
176
  llm=HuggingFaceEndpoint(
 
159
  ]
160
 
161
  # Build graph function
162
+ def build_graph(provider: str = "openai"):
163
  """Build the graph"""
164
  # Load environment variables from .env file
165
  if provider == "google":
 
170
  llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
171
  elif provider == "openai":
172
  # OpenAI
173
+ llm = ChatOpenAI(model="gpt-4o-mini", temperature=0)
174
  elif provider == "huggingface":
175
  llm = ChatHuggingFace(
176
  llm=HuggingFaceEndpoint(