samoye16 commited on
Commit
831badd
·
verified ·
1 Parent(s): 6676eec

Update agents.py

Browse files
Files changed (1) hide show
  1. agents.py +3 -1
agents.py CHANGED
@@ -18,6 +18,8 @@ from langchain.tools.retriever import create_retriever_tool
18
  from supabase.client import create_client, Client
19
 
20
 
 
 
21
  # ---- Basic Arithmetic Utilities ---- #
22
  @tool
23
  def multiply(a: int, b: int) -> int:
@@ -106,7 +108,7 @@ def create_agent_flow(provider: str = "groq"):
106
  if provider == "google":
107
  llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
108
  elif provider == "groq":
109
- llm = ChatGroq(model="qwen-qwq-32b", temperature=0)
110
  elif provider == "huggingface":
111
  llm = ChatHuggingFace(llm=HuggingFaceEndpoint(
112
  url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
 
18
  from supabase.client import create_client, Client
19
 
20
 
21
+ # Load environment variables
22
+
23
  # ---- Basic Arithmetic Utilities ---- #
24
  @tool
25
  def multiply(a: int, b: int) -> int:
 
108
  if provider == "google":
109
  llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
110
  elif provider == "groq":
111
+ llm = ChatGroq(api_key="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" , model="qwen-qwq-32b", temperature=0)
112
  elif provider == "huggingface":
113
  llm = ChatHuggingFace(llm=HuggingFaceEndpoint(
114
  url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",