Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -165,10 +165,9 @@ def build_graph(provider: str = "huggingface"):
|
|
| 165 |
llm = ChatHuggingFace(
|
| 166 |
llm=HuggingFaceEndpoint(endpoint_url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf"),
|
| 167 |
temperature=0,
|
| 168 |
-
)
|
| 169 |
|
| 170 |
|
| 171 |
-
)
|
| 172 |
else:
|
| 173 |
raise ValueError("Invalid provider. Choose 'google', 'groq' or 'huggingface'.")
|
| 174 |
# Bind tools to LLM
|
|
|
|
| 165 |
llm = ChatHuggingFace(
|
| 166 |
llm=HuggingFaceEndpoint(endpoint_url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf"),
|
| 167 |
temperature=0,
|
| 168 |
+
)
|
| 169 |
|
| 170 |
|
|
|
|
| 171 |
else:
|
| 172 |
raise ValueError("Invalid provider. Choose 'google', 'groq' or 'huggingface'.")
|
| 173 |
# Bind tools to LLM
|