KeyArgo
/
smolagents
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix examples/tool_calling_agent_ollama.py

This commit is contained in:
Aymeric Roucher 2025-01-06 14:59:21 +01:00 committed by GitHub
parent c3a11e09da
commit dbb8b7f5ce
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/tool_calling_agent_ollama.py
View File

@ -4,7 +4,7 @@ from typing import Optional
model = LiteLLMModel( model = LiteLLMModel(
model_id="ollama_chat/llama3.2", model_id="ollama_chat/llama3.2",
api_base="http://localhost:11434/v1", # replace with remote open-ai compatible server if necessary api_base="http://localhost:11434", # replace with remote open-ai compatible server if necessary
api_key="your-api-key" # replace with API key if necessary api_key="your-api-key" # replace with API key if necessary
) )
@ -22,4 +22,4 @@ def get_weather(location: str, celsius: Optional[bool] = False) -> str:
agent = ToolCallingAgent(tools=[get_weather], model=model) agent = ToolCallingAgent(tools=[get_weather], model=model)
print(agent.run("What's the weather like in Paris?")) print(agent.run("What's the weather like in Paris?"))