danvics/haiku.rag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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

Browse source 
Post 0.7.0 release minor fixes
This commit is contained in:
Yiorgis Gozadinos 2025-09-03 11:44:20 +03:00 committed by GitHub
commit 246860b281
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
src/haiku/rag/logging.py
View file

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

4
src/haiku/rag/qa/agent.py
View file

@ -1,6 +1,6 @@
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
from pydantic_ai import Agent, RunContext 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 pydantic_ai.providers.ollama import OllamaProvider
from haiku.rag.client import HaikuRAG from haiku.rag.client import HaikuRAG
@ -61,7 +61,7 @@ class QuestionAnswerAgent:
def _get_model(self, provider: str, model: str): def _get_model(self, provider: str, model: str):
"""Get the appropriate model object for the provider.""" """Get the appropriate model object for the provider."""
if provider == "ollama": if provider == "ollama":
return OpenAIModel( return OpenAIChatModel(
model_name=model, model_name=model,
provider=OllamaProvider(base_url=f"{Config.OLLAMA_BASE_URL}/v1"), provider=OllamaProvider(base_url=f"{Config.OLLAMA_BASE_URL}/v1"),
) )

1
tests/generate_benchmark_db.py
View file

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