Update agent.py
Browse files
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 |
-
#
|
151 |
chat = ChatHuggingFace(
|
152 |
llm=HuggingFaceEndpoint(
|
153 |
-
|
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 |
)
|