Prasanthkumar commited on
Commit
0b61a31
·
verified ·
1 Parent(s): e4495f7

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +2 -1
model.py CHANGED
@@ -163,7 +163,8 @@ def build_graph(provider: str = "groq"):
163
  llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
164
  elif provider == "groq":
165
  # Groq https://console.groq.com/docs/models
166
- llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
 
167
  elif provider == "huggingface":
168
  # TODO: Add huggingface endpoint
169
  llm = ChatHuggingFace(
 
163
  llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
164
  elif provider == "groq":
165
  # Groq https://console.groq.com/docs/models
166
+ api_key = os.getenv('GROQ_API')
167
+ llm = ChatGroq(model="qwen-qwq-32b", temperature=0,api_key = api_key) # optional : qwen-qwq-32b gemma2-9b-it
168
  elif provider == "huggingface":
169
  # TODO: Add huggingface endpoint
170
  llm = ChatHuggingFace(