IamRulo commited on
Commit
b189c98
·
verified ·
1 Parent(s): 4cbc21e

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
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: