aelyazid commited on
Commit
99a1f90
·
verified ·
1 Parent(s): 3153352

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +5 -4
agent.py CHANGED
@@ -163,10 +163,11 @@ def build_graph(provider: str = "huggingface"):
163
  elif provider == "huggingface":
164
  # TODO: Add huggingface endpoint
165
  llm = ChatHuggingFace(
166
- llm=HuggingFaceEndpoint(
167
- url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
168
- temperature=0,
169
- ),
 
170
  )
171
  else:
172
  raise ValueError("Invalid provider. Choose 'google', 'groq' or 'huggingface'.")
 
163
  elif provider == "huggingface":
164
  # TODO: Add huggingface endpoint
165
  llm = ChatHuggingFace(
166
+ llm=HuggingFaceEndpoint(endpoint_url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf"),
167
+ temperature=0,
168
+ )
169
+
170
+
171
  )
172
  else:
173
  raise ValueError("Invalid provider. Choose 'google', 'groq' or 'huggingface'.")