haiku.rag 0.7.0__tar.gz → 0.7.1__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of haiku.rag might be problematic. Click here for more details.
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/PKG-INFO +1 -1
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/pyproject.toml +1 -1
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/logging.py +1 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/qa/agent.py +2 -2
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/generate_benchmark_db.py +1 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/uv.lock +1 -1
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/.github/FUNDING.yml +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/.github/workflows/build-docs.yml +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/.github/workflows/build-publish.yml +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/.gitignore +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/.pre-commit-config.yaml +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/.python-version +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/LICENSE +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/README.md +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/docs/benchmarks.md +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/docs/cli.md +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/docs/configuration.md +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/docs/index.md +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/docs/installation.md +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/docs/mcp.md +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/docs/python.md +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/docs/server.md +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/mkdocs.yml +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/__init__.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/app.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/chunker.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/cli.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/client.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/config.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/embeddings/__init__.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/embeddings/base.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/embeddings/ollama.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/embeddings/openai.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/embeddings/voyageai.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/mcp.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/migration.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/monitor.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/qa/__init__.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/qa/prompts.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/reader.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/reranking/__init__.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/reranking/base.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/reranking/cohere.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/reranking/mxbai.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/store/__init__.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/store/engine.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/store/models/__init__.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/store/models/chunk.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/store/models/document.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/store/repositories/__init__.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/store/repositories/chunk.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/store/repositories/document.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/store/repositories/settings.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/store/upgrades/__init__.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/utils.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/__init__.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/conftest.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/llm_judge.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_app.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_chunk.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_chunker.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_cli.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_client.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_document.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_embedder.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_lancedb_connection.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_monitor.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_qa.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_reader.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_rebuild.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_reranker.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_search.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_settings.py +0 -0
- {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_utils.py +0 -0
|
@@ -6,6 +6,7 @@ from rich.logging import RichHandler
|
|
|
6
6
|
logging.basicConfig(level=logging.DEBUG)
|
|
7
7
|
logging.getLogger("httpx").setLevel(logging.WARNING)
|
|
8
8
|
logging.getLogger("httpcore").setLevel(logging.WARNING)
|
|
9
|
+
logging.getLogger("docling").setLevel(logging.WARNING)
|
|
9
10
|
|
|
10
11
|
|
|
11
12
|
def get_logger() -> logging.Logger:
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
from pydantic import BaseModel, Field
|
|
2
2
|
from pydantic_ai import Agent, RunContext
|
|
3
|
-
from pydantic_ai.models.openai import
|
|
3
|
+
from pydantic_ai.models.openai import OpenAIChatModel
|
|
4
4
|
from pydantic_ai.providers.ollama import OllamaProvider
|
|
5
5
|
|
|
6
6
|
from haiku.rag.client import HaikuRAG
|
|
@@ -61,7 +61,7 @@ class QuestionAnswerAgent:
|
|
|
61
61
|
def _get_model(self, provider: str, model: str):
|
|
62
62
|
"""Get the appropriate model object for the provider."""
|
|
63
63
|
if provider == "ollama":
|
|
64
|
-
return
|
|
64
|
+
return OpenAIChatModel(
|
|
65
65
|
model_name=model,
|
|
66
66
|
provider=OllamaProvider(base_url=f"{Config.OLLAMA_BASE_URL}/v1"),
|
|
67
67
|
)
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|