Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -163,10 +163,11 @@ def build_graph(provider: str = "huggingface"):
|
|
| 163 |
elif provider == "huggingface":
|
| 164 |
# TODO: Add huggingface endpoint
|
| 165 |
llm = ChatHuggingFace(
|
| 166 |
-
|
| 167 |
-
|
| 168 |
-
|
| 169 |
-
|
|
|
|
| 170 |
)
|
| 171 |
else:
|
| 172 |
raise ValueError("Invalid provider. Choose 'google', 'groq' or 'huggingface'.")
|
|
|
|
| 163 |
elif provider == "huggingface":
|
| 164 |
# TODO: Add huggingface endpoint
|
| 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'.")
|