Update tools.py
Browse files
tools.py
CHANGED
@@ -186,7 +186,7 @@ def build_graph(provider: str = "huggingface"):
|
|
186 |
llm = ChatHuggingFace(
|
187 |
llm=HuggingFaceEndpoint(
|
188 |
endpoint_url="https://api-inference.huggingface.co/models/HuggingFaceTB/SmolLM2-1.7B-Instruct",
|
189 |
-
huggingfacehub_api_token=os.getenv("
|
190 |
)
|
191 |
)
|
192 |
else:
|
|
|
186 |
llm = ChatHuggingFace(
|
187 |
llm=HuggingFaceEndpoint(
|
188 |
endpoint_url="https://api-inference.huggingface.co/models/HuggingFaceTB/SmolLM2-1.7B-Instruct",
|
189 |
+
huggingfacehub_api_token=os.getenv("HF_INFERENCE_ENDPOINT") # Ensure you have this in your .env file
|
190 |
)
|
191 |
)
|
192 |
else:
|