mdicio commited on
Commit
16fe3a2
·
1 Parent(s): b4f5b89
Files changed (1) hide show
  1. agent.py +4 -6
agent.py CHANGED
@@ -156,12 +156,10 @@ def build_graph(provider: str = "groq"):
156
  ) # optional : qwen-qwq-32b gemma2-9b-it
157
  elif provider == "huggingface":
158
  # TODO: Add huggingface endpoint
159
- llm = ChatHuggingFace(
160
- llm=HuggingFaceEndpoint(
161
- url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
162
- temperature=0,
163
- ),
164
- )
165
  else:
166
  raise ValueError("Invalid provider. Choose 'google', 'groq' or 'huggingface'.")
167
  # Bind tools to LLM
 
156
  ) # optional : qwen-qwq-32b gemma2-9b-it
157
  elif provider == "huggingface":
158
  # TODO: Add huggingface endpoint
159
+ llm=HuggingFaceEndpoint(
160
+ repo_id="Meta-DeepLearning/llama-2-7b-chat-hf",
161
+ temperature=0,
162
+ )
 
 
163
  else:
164
  raise ValueError("Invalid provider. Choose 'google', 'groq' or 'huggingface'.")
165
  # Bind tools to LLM