PradeepBodhi commited on
Commit
27a3848
·
verified ·
1 Parent(s): f578b84

Update agent.py

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