Merge pull request #46 from ggozad/chore/post-release-fixes

Post 0.7.0 release minor fixes

src/haiku/rag/logging.py

@@ -6,6 +6,7 @@ from rich.logging import RichHandler
 logging.basicConfig(level=logging.DEBUG)
 logging.getLogger("httpx").setLevel(logging.WARNING)
 logging.getLogger("httpcore").setLevel(logging.WARNING)
+logging.getLogger("docling").setLevel(logging.WARNING)
 
 
 def get_logger() -> logging.Logger:

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"),
             )

tests/generate_benchmark_db.py

@@ -6,6 +6,7 @@ from llm_judge import LLMJudge
 from rich.console import Console
 from rich.progress import Progress
 
+from haiku.rag import logging  # noqa
 from haiku.rag.client import HaikuRAG
 from haiku.rag.qa import get_qa_agent