Ollama support for openai and langgraph examples

Author: pamelafox

examples/langgraph_agent.py

@@ -43,12 +43,10 @@ def play_song_on_apple(song: str):
 elif API_HOST == "github":
     model = ChatOpenAI(model=os.getenv("GITHUB_MODEL", "gpt-4o"), base_url="https://models.inference.ai.azure.com", api_key=os.environ["GITHUB_TOKEN"])
 elif API_HOST == "ollama":
-    model = ChatOpenAI(model=os.environ["OLLAMA_MODEL"], base_url=os.environ["OLLAMA_ENDPOINT"], api_key="none")
+    model = ChatOpenAI(model=os.environ["OLLAMA_MODEL"], base_url=os.environ.get("OLLAMA_ENDPOINT", "http://localhost:11434/v1"), api_key="none")
 
 model = model.bind_tools(tools, parallel_tool_calls=False)
 
-# Define nodes and conditional edges
-
 
 # Define the function that determines whether to continue or not
 def should_continue(state):

examples/openai_agents_tools.py

@@ -31,6 +31,9 @@
         azure_ad_token_provider=token_provider,
     )
     MODEL_NAME = os.environ["AZURE_OPENAI_CHAT_DEPLOYMENT"]
+elif API_HOST == "ollama":
+    client = openai.AsyncOpenAI(base_url=os.environ.get("OLLAMA_ENDPOINT", "http://localhost:11434/v1"), api_key="none")
+    MODEL_NAME = os.environ["OLLAMA_MODEL"]
 
 
 @function_tool