rsobieski commited on
Commit
58a60d8
·
verified ·
1 Parent(s): d7bbf02

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
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.1",
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)