haiku.rag 0.3.1__tar.gz → 0.3.2__tar.gz

{haiku_rag-0.3.1 → haiku_rag-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: haiku.rag
-Version: 0.3.1
+Version: 0.3.2
 Summary: Retrieval Augmented Generation (RAG) with SQLite
 Author-email: Yiorgis Gozadinos <ggozadinos@gmail.com>
 License: MIT

{haiku_rag-0.3.1 → haiku_rag-0.3.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "haiku.rag"
-version = "0.3.1"
+version = "0.3.2"
 description = "Retrieval Augmented Generation (RAG) with SQLite"
 authors = [{ name = "Yiorgis Gozadinos", email = "ggozadinos@gmail.com" }]
 license = { text = "MIT" }

{haiku_rag-0.3.1 → haiku_rag-0.3.2}/src/haiku/rag/qa/__init__.py

@@ -8,7 +8,6 @@ def get_qa_agent(client: HaikuRAG, model: str = "") -> QuestionAnswerAgentBase:
     """
     Factory function to get the appropriate QA agent based on the configuration.
     """
-
     if Config.QA_PROVIDER == "ollama":
         return QuestionAnswerOllamaAgent(client, model or Config.QA_MODEL)
 
@@ -21,7 +20,7 @@ def get_qa_agent(client: HaikuRAG, model: str = "") -> QuestionAnswerAgentBase:
                 "Please install haiku.rag with the 'openai' extra:"
                 "uv pip install haiku.rag --extra openai"
             )
-        return QuestionAnswerOpenAIAgent(client, model or "gpt-4o-mini")
+        return QuestionAnswerOpenAIAgent(client, model or Config.QA_MODEL)
 
     if Config.QA_PROVIDER == "anthropic":
         try:
@@ -32,8 +31,6 @@ def get_qa_agent(client: HaikuRAG, model: str = "") -> QuestionAnswerAgentBase:
                 "Please install haiku.rag with the 'anthropic' extra:"
                 "uv pip install haiku.rag --extra anthropic"
             )
-        return QuestionAnswerAnthropicAgent(
-            client, model or "claude-3-5-haiku-20241022"
-        )
+        return QuestionAnswerAnthropicAgent(client, model or Config.QA_MODEL)
 
     raise ValueError(f"Unsupported QA provider: {Config.QA_PROVIDER}")

{haiku_rag-0.3.1 → haiku_rag-0.3.2}/tests/generate_benchmark_db.py

@@ -6,7 +6,7 @@ from llm_judge import LLMJudge
 from tqdm import tqdm
 
 from haiku.rag.client import HaikuRAG
-from haiku.rag.qa.ollama import QuestionAnswerOllamaAgent
+from haiku.rag.qa import get_qa_agent
 
 db_path = Path(__file__).parent / "data" / "benchmark.sqlite"
 
@@ -88,7 +88,7 @@ async def run_qa_benchmark(k: int | None = None):
     total_questions = 0
 
     async with HaikuRAG(db_path) as rag:
-        qa = QuestionAnswerOllamaAgent(rag)
+        qa = get_qa_agent(rag)
 
         for i, doc in enumerate(tqdm(corpus, desc="QA Benchmarking")):
             question = doc["question"]  # type: ignore

{haiku_rag-0.3.1 → haiku_rag-0.3.2}/uv.lock

@@ -816,7 +816,7 @@ wheels = [
 
 [[package]]
 name = "haiku-rag"
-version = "0.3.1"
+version = "0.3.2"
 source = { editable = "." }
 dependencies = [
     { name = "fastmcp" },