Update agent.py
Browse files
agent.py
CHANGED
|
@@ -24,7 +24,7 @@ def build_graph():
|
|
| 24 |
# llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
|
| 25 |
# llm = HuggingFaceInferenceAPI(model_name="Qwen/Qwen2.5-Coder-32B-Instruct")
|
| 26 |
llm = HuggingFaceEndpoint(
|
| 27 |
-
url="https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.
|
| 28 |
huggingfacehub_api_token=os.environ["HF_TOKEN"]
|
| 29 |
)
|
| 30 |
llm_with_tools = llm.bind_tools(TOOLS)
|
|
|
|
| 24 |
# llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
|
| 25 |
# llm = HuggingFaceInferenceAPI(model_name="Qwen/Qwen2.5-Coder-32B-Instruct")
|
| 26 |
llm = HuggingFaceEndpoint(
|
| 27 |
+
url="https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.3",
|
| 28 |
huggingfacehub_api_token=os.environ["HF_TOKEN"]
|
| 29 |
)
|
| 30 |
llm_with_tools = llm.bind_tools(TOOLS)
|