lakshinav commited on
Commit
11fad68
·
verified ·
1 Parent(s): 6ea7b5c

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -163,7 +163,7 @@ def build_graph(provider: str = "huggingface"):
163
  # llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
164
  if provider == "huggingface":
165
  # TODO: Add huggingface endpoint
166
- HUGGINGFACEHUB_API_TOKEN = os.getenv['HUGGINGFACEHUB_API_TOKEN']
167
  repo_id = "llama-2-7b-chat-hf"
168
  llm = ChatHuggingFace(
169
  llm=HuggingFaceEndpoint(
 
163
  # llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
164
  if provider == "huggingface":
165
  # TODO: Add huggingface endpoint
166
+ HUGGINGFACEHUB_API_TOKEN = os.getenv('HUGGINGFACEHUB_API_TOKEN')
167
  repo_id = "llama-2-7b-chat-hf"
168
  llm = ChatHuggingFace(
169
  llm=HuggingFaceEndpoint(