Spaces:
Sleeping
Sleeping
bug llm
Browse files
agent.py
CHANGED
|
@@ -156,12 +156,10 @@ def build_graph(provider: str = "groq"):
|
|
| 156 |
) # optional : qwen-qwq-32b gemma2-9b-it
|
| 157 |
elif provider == "huggingface":
|
| 158 |
# TODO: Add huggingface endpoint
|
| 159 |
-
llm
|
| 160 |
-
|
| 161 |
-
|
| 162 |
-
|
| 163 |
-
),
|
| 164 |
-
)
|
| 165 |
else:
|
| 166 |
raise ValueError("Invalid provider. Choose 'google', 'groq' or 'huggingface'.")
|
| 167 |
# Bind tools to LLM
|
|
|
|
| 156 |
) # optional : qwen-qwq-32b gemma2-9b-it
|
| 157 |
elif provider == "huggingface":
|
| 158 |
# TODO: Add huggingface endpoint
|
| 159 |
+
llm=HuggingFaceEndpoint(
|
| 160 |
+
repo_id="Meta-DeepLearning/llama-2-7b-chat-hf",
|
| 161 |
+
temperature=0,
|
| 162 |
+
)
|
|
|
|
|
|
|
| 163 |
else:
|
| 164 |
raise ValueError("Invalid provider. Choose 'google', 'groq' or 'huggingface'.")
|
| 165 |
# Bind tools to LLM
|