Update agent.py
Browse files
agent.py
CHANGED
|
@@ -160,7 +160,7 @@ def build_graph(provider: str = "huggingface"):
|
|
| 160 |
elif provider == "groq":
|
| 161 |
# Groq https://console.groq.com/docs/models
|
| 162 |
llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
|
| 163 |
-
elif provider == "huggingface":
|
| 164 |
llm = ChatHuggingFace(
|
| 165 |
llm=HuggingFaceEndpoint(
|
| 166 |
repo_id = "Qwen/Qwen2.5-Coder-32B-Instruct"
|
|
|
|
| 160 |
elif provider == "groq":
|
| 161 |
# Groq https://console.groq.com/docs/models
|
| 162 |
llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
|
| 163 |
+
elif provider == "huggingface": #free acess
|
| 164 |
llm = ChatHuggingFace(
|
| 165 |
llm=HuggingFaceEndpoint(
|
| 166 |
repo_id = "Qwen/Qwen2.5-Coder-32B-Instruct"
|