From dbb8b7f5cedb979a0a712d4d59e5bb5d19a03275 Mon Sep 17 00:00:00 2001 From: Aymeric Roucher <69208727+aymeric-roucher@users.noreply.github.com> Date: Mon, 6 Jan 2025 14:59:21 +0100 Subject: [PATCH] Fix examples/tool_calling_agent_ollama.py --- examples/tool_calling_agent_ollama.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/tool_calling_agent_ollama.py b/examples/tool_calling_agent_ollama.py index a3ca6eb..ad914f8 100644 --- a/examples/tool_calling_agent_ollama.py +++ b/examples/tool_calling_agent_ollama.py @@ -4,7 +4,7 @@ from typing import Optional model = LiteLLMModel( 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 ) @@ -22,4 +22,4 @@ def get_weather(location: str, celsius: Optional[bool] = False) -> str: agent = ToolCallingAgent(tools=[get_weather], model=model) -print(agent.run("What's the weather like in Paris?")) \ No newline at end of file +print(agent.run("What's the weather like in Paris?"))