Update agent.py
Browse files
agent.py
CHANGED
@@ -183,9 +183,9 @@ def build_graph(provider: str = "huggingface"):
|
|
183 |
# TODO: Add huggingface endpoint
|
184 |
llm = ChatHuggingFace(
|
185 |
llm=HuggingFaceEndpoint(
|
186 |
-
endpoint_url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
|
|
|
187 |
temperature=0.2,
|
188 |
-
#token="HF_TOKEN",
|
189 |
),
|
190 |
)
|
191 |
else:
|
|
|
183 |
# TODO: Add huggingface endpoint
|
184 |
llm = ChatHuggingFace(
|
185 |
llm=HuggingFaceEndpoint(
|
186 |
+
#endpoint_url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
|
187 |
+
endpoint_url="https://api-inference.huggingface.co/models/Qwen/Qwen3-30B-A3B",
|
188 |
temperature=0.2,
|
|
|
189 |
),
|
190 |
)
|
191 |
else:
|