From 450a58eb522539f11d18dfd3e773cbaa1160ddff Mon Sep 17 00:00:00 2001
From: Yiorgis Gozadinos <ggozadinos@gmail.com>
Date: Wed, 3 Sep 2025 11:41:48 +0300
Subject: [PATCH] Update OpenAI pydantic model

---
 src/haiku/rag/qa/agent.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/haiku/rag/qa/agent.py b/src/haiku/rag/qa/agent.py
index b76112bc..711106ba 100644
--- a/src/haiku/rag/qa/agent.py
+++ b/src/haiku/rag/qa/agent.py
@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field
 from pydantic_ai import Agent, RunContext
-from pydantic_ai.models.openai import OpenAIModel
+from pydantic_ai.models.openai import OpenAIChatModel
 from pydantic_ai.providers.ollama import OllamaProvider
 
 from haiku.rag.client import HaikuRAG
@@ -61,7 +61,7 @@ class QuestionAnswerAgent:
     def _get_model(self, provider: str, model: str):
         """Get the appropriate model object for the provider."""
         if provider == "ollama":
-            return OpenAIModel(
+            return OpenAIChatModel(
                 model_name=model,
                 provider=OllamaProvider(base_url=f"{Config.OLLAMA_BASE_URL}/v1"),
             )