Spaces:
Build error
Build error
Update agents.py
Browse files
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",
|