Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -71,9 +71,6 @@ def build_graph(provider: str = "google"):
|
|
| 71 |
if provider == "google":
|
| 72 |
# Google Gemini
|
| 73 |
llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
|
| 74 |
-
elif provider == "groq":
|
| 75 |
-
# Groq https://console.groq.com/docs/models
|
| 76 |
-
llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
|
| 77 |
elif provider == "huggingface":
|
| 78 |
llm = ChatHuggingFace(
|
| 79 |
llm=HuggingFaceEndpoint(
|
|
|
|
| 71 |
if provider == "google":
|
| 72 |
# Google Gemini
|
| 73 |
llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
|
|
|
|
|
|
|
|
|
|
| 74 |
elif provider == "huggingface":
|
| 75 |
llm = ChatHuggingFace(
|
| 76 |
llm=HuggingFaceEndpoint(
|