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

Upload agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -189,7 +189,7 @@ def build_graph(provider: str = "openai"):
189
  llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
190
  elif provider == "groq":
191
  # Groq https://console.groq.com/docs/models
192
- llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
193
  elif provider == "huggingface":
194
  # TODO: Add huggingface endpoint
195
  llm = ChatHuggingFace(
 
189
  llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
190
  elif provider == "groq":
191
  # Groq https://console.groq.com/docs/models
192
+ llm = ChatGroq(model="gemma2-9b-it", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
193
  elif provider == "huggingface":
194
  # TODO: Add huggingface endpoint
195
  llm = ChatHuggingFace(