sebasmos commited on
Commit
d9765af
·
1 Parent(s): 2bfa0ff

Add HF token fallback in agent.py

Browse files
Files changed (1) hide show
  1. 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=os.getenv("HF_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()