Feature/ollama local model integration

.env.sample

@@ -1,8 +1,12 @@
-# API_HOST can be either azure or openai:
+# API_HOST can be either azure, ollama or openai:
 API_HOST=azure
 # Configure for Azure:
 AZURE_OPENAI_ENDPOINT=https://YOUR-AZURE-OPENAI-SERVICE-NAME.openai.azure.com
 AZURE_OPENAI_CHAT_DEPLOYMENT=YOUR-AZURE-DEPLOYMENT-NAME
+# Config for Ollama
+OLLAMA_BASE_URL=http://localhost:11434/v1
+OLLAMA_MODEL=qwen3.5:0.8b
+OLLAMA_API_KEY="dummy_key"
 # Configure for OpenAI.com:
 OPENAI_API_KEY=YOUR-OPENAI-KEY
 OPENAI_MODEL=gpt-5.4

README.md

@@ -66,15 +66,32 @@ The dev container includes a Redis server, which is used by the `agent_history_r
     * [Python 3.10+](https://www.python.org/downloads/)
     * [uv](https://docs.astral.sh/uv/getting-started/installation/)
     * Git
+    * Ollama - (optional for local Ollama model api call)
 
 2. Clone the repository:
 
     ```shell
     git clone https://github.com/Azure-Samples/python-agentframework-demos
     cd python-agentframework-demos
     ```
+3. Create a virtual environment:
 
-3. Install the dependencies:
+    ```shell
+    python -m venv .venv
+    ```
+3. Activate the virtual environment:
+
+    ```shell
+    source .venv/bin/activate  
+    # On Windows: .venv\Scripts\activate
+    ```
+4. Install the uv:
+
+    ```shell
+    pip uv
+    ```
+
+4. Install the dependencies:
 
     ```shell
     uv sync

examples/agent_basic.py

@@ -20,6 +20,12 @@
         api_key=token_provider,
         model=os.environ["AZURE_OPENAI_CHAT_DEPLOYMENT"],
     )
+elif API_HOST == "ollama":
+    client = OpenAIChatClient(
+        base_url=os.environ['OLLAMA_BASE_URL'],
+        model=os.environ["OLLAMA_MODEL"],
+        api_key=os.environ['OLLAMA_API_KEY'],        
+    )
 else:
     client = OpenAIChatClient(
         api_key=os.environ["OPENAI_API_KEY"], model=os.environ.get("OPENAI_MODEL", "gpt-5.4")

examples/agent_tool.py

@@ -31,6 +31,12 @@
         api_key=token_provider,
         model=os.environ["AZURE_OPENAI_CHAT_DEPLOYMENT"],
     )
+elif API_HOST == "ollama":
+    client = OpenAIChatClient(
+        base_url=os.environ['OLLAMA_BASE_URL'],
+        model=os.environ["OLLAMA_MODEL"],
+        api_key=os.environ['OLLAMA_API_KEY'],        
+    )
 else:
     client = OpenAIChatClient(
         api_key=os.environ["OPENAI_API_KEY"], model=os.environ.get("OPENAI_MODEL", "gpt-5.4")

examples/agent_tools.py

@@ -33,6 +33,12 @@
         api_key=token_provider,
         model=os.environ["AZURE_OPENAI_CHAT_DEPLOYMENT"],
     )
+elif API_HOST == "ollama":
+    client = OpenAIChatClient(
+        base_url=os.environ['OLLAMA_BASE_URL'],
+        model=os.environ["OLLAMA_MODEL"],
+        api_key=os.environ['OLLAMA_API_KEY'],        
+    )
 else:
     client = OpenAIChatClient(
         api_key=os.environ["OPENAI_API_KEY"], model=os.environ.get("OPENAI_MODEL", "gpt-5.4")

examples/workflow_fan_out_fan_in_edges.py

@@ -34,6 +34,12 @@
         api_key=token_provider,
         model=os.environ["AZURE_OPENAI_CHAT_DEPLOYMENT"],
     )
+elif API_HOST == "ollama":
+    client = OpenAIChatClient(
+        base_url=os.environ['OLLAMA_BASE_URL'],
+        model=os.environ["OLLAMA_MODEL"],
+        api_key=os.environ['OLLAMA_API_KEY'],        
+    )
 else:
     client = OpenAIChatClient(api_key=os.environ["OPENAI_API_KEY"], model=os.environ.get("OPENAI_MODEL", "gpt-5.4"))