ama2aifusion commited on
Commit
3cd55ca
·
verified ·
1 Parent(s): f4c3d3d

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -0
agent.py CHANGED
@@ -149,6 +149,7 @@ def build_graph(provider: str = "huggingface"):
149
  # Groq https://console.groq.com/docs/models
150
  chat = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
151
  elif provider == "openai":
 
152
  model_openai = "gpt-4o"
153
  chat = ChatOpenAI(
154
  model=model_openai,
 
149
  # Groq https://console.groq.com/docs/models
150
  chat = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
151
  elif provider == "openai":
152
+ # Set the model from openai here
153
  model_openai = "gpt-4o"
154
  chat = ChatOpenAI(
155
  model=model_openai,