ama2aifusion commited on
Commit
3889a5e
·
verified ·
1 Parent(s): 17789e9

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -147,10 +147,10 @@ def build_graph(provider: str = "huggingface"):
147
  # Groq https://console.groq.com/docs/models
148
  chat = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
149
  elif provider == "huggingface":
150
- # TODO: Add huggingface endpoint
151
  chat = ChatHuggingFace(
152
  llm=HuggingFaceEndpoint(
153
- url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
154
  temperature=0,
155
  ),
156
  )
 
147
  # Groq https://console.groq.com/docs/models
148
  chat = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
149
  elif provider == "huggingface":
150
+ # Add huggingface endpoint
151
  chat = ChatHuggingFace(
152
  llm=HuggingFaceEndpoint(
153
+ endpoint_url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
154
  temperature=0,
155
  ),
156
  )