Spaces:
Sleeping
Sleeping
Add HF token fallback in agent.py
Browse files- src/agent.py +3 -1
src/agent.py
CHANGED
|
@@ -28,11 +28,13 @@ class ProjectAgent:
|
|
| 28 |
"""Initialize the agent."""
|
| 29 |
self.rag = rag
|
| 30 |
# Use HF Inference API (free tier)
|
|
|
|
|
|
|
| 31 |
llm = HuggingFaceEndpoint(
|
| 32 |
repo_id=model_name,
|
| 33 |
temperature=0.1,
|
| 34 |
max_new_tokens=512,
|
| 35 |
-
huggingfacehub_api_token=
|
| 36 |
)
|
| 37 |
self.llm = ChatHuggingFace(llm=llm)
|
| 38 |
self.graph = self._build_graph()
|
|
|
|
| 28 |
"""Initialize the agent."""
|
| 29 |
self.rag = rag
|
| 30 |
# Use HF Inference API (free tier)
|
| 31 |
+
# HF Spaces provides token as HF_TOKEN or HUGGING_FACE_HUB_TOKEN
|
| 32 |
+
hf_token = os.getenv("HF_TOKEN") or os.getenv("HUGGING_FACE_HUB_TOKEN")
|
| 33 |
llm = HuggingFaceEndpoint(
|
| 34 |
repo_id=model_name,
|
| 35 |
temperature=0.1,
|
| 36 |
max_new_tokens=512,
|
| 37 |
+
huggingfacehub_api_token=hf_token
|
| 38 |
)
|
| 39 |
self.llm = ChatHuggingFace(llm=llm)
|
| 40 |
self.graph = self._build_graph()
|