aleixlopezpascual commited on
Commit
237dd50
·
1 Parent(s): 999f7f9

remove groq

Browse files
Files changed (1) hide show
  1. agent.py +4 -4
agent.py CHANGED
@@ -152,15 +152,15 @@ tools = [
152
  ]
153
 
154
  # Build graph function
155
- def build_graph(provider: str = "groq"):
156
  """Build the graph"""
157
  # Load environment variables from .env file
158
  if provider == "google":
159
  # Google Gemini
160
  llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
161
- elif provider == "groq":
162
- # Groq https://console.groq.com/docs/models
163
- llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
164
  elif provider == "huggingface":
165
  # TODO: Add huggingface endpoint
166
  llm = ChatHuggingFace(
 
152
  ]
153
 
154
  # Build graph function
155
+ def build_graph(provider: str = "google"):
156
  """Build the graph"""
157
  # Load environment variables from .env file
158
  if provider == "google":
159
  # Google Gemini
160
  llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
161
+ # elif provider == "groq":
162
+ # # Groq https://console.groq.com/docs/models
163
+ # llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
164
  elif provider == "huggingface":
165
  # TODO: Add huggingface endpoint
166
  llm = ChatHuggingFace(