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.

Files changed (74) hide show
  1. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/PKG-INFO +1 -1
  2. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/pyproject.toml +1 -1
  3. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/logging.py +1 -0
  4. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/qa/agent.py +2 -2
  5. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/generate_benchmark_db.py +1 -0
  6. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/uv.lock +1 -1
  7. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/.github/FUNDING.yml +0 -0
  8. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/.github/workflows/build-docs.yml +0 -0
  9. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/.github/workflows/build-publish.yml +0 -0
  10. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/.gitignore +0 -0
  11. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/.pre-commit-config.yaml +0 -0
  12. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/.python-version +0 -0
  13. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/LICENSE +0 -0
  14. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/README.md +0 -0
  15. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/docs/benchmarks.md +0 -0
  16. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/docs/cli.md +0 -0
  17. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/docs/configuration.md +0 -0
  18. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/docs/index.md +0 -0
  19. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/docs/installation.md +0 -0
  20. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/docs/mcp.md +0 -0
  21. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/docs/python.md +0 -0
  22. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/docs/server.md +0 -0
  23. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/mkdocs.yml +0 -0
  24. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/__init__.py +0 -0
  25. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/app.py +0 -0
  26. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/chunker.py +0 -0
  27. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/cli.py +0 -0
  28. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/client.py +0 -0
  29. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/config.py +0 -0
  30. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/embeddings/__init__.py +0 -0
  31. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/embeddings/base.py +0 -0
  32. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/embeddings/ollama.py +0 -0
  33. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/embeddings/openai.py +0 -0
  34. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/embeddings/voyageai.py +0 -0
  35. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/mcp.py +0 -0
  36. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/migration.py +0 -0
  37. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/monitor.py +0 -0
  38. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/qa/__init__.py +0 -0
  39. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/qa/prompts.py +0 -0
  40. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/reader.py +0 -0
  41. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/reranking/__init__.py +0 -0
  42. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/reranking/base.py +0 -0
  43. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/reranking/cohere.py +0 -0
  44. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/reranking/mxbai.py +0 -0
  45. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/store/__init__.py +0 -0
  46. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/store/engine.py +0 -0
  47. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/store/models/__init__.py +0 -0
  48. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/store/models/chunk.py +0 -0
  49. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/store/models/document.py +0 -0
  50. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/store/repositories/__init__.py +0 -0
  51. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/store/repositories/chunk.py +0 -0
  52. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/store/repositories/document.py +0 -0
  53. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/store/repositories/settings.py +0 -0
  54. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/store/upgrades/__init__.py +0 -0
  55. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/src/haiku/rag/utils.py +0 -0
  56. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/__init__.py +0 -0
  57. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/conftest.py +0 -0
  58. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/llm_judge.py +0 -0
  59. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_app.py +0 -0
  60. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_chunk.py +0 -0
  61. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_chunker.py +0 -0
  62. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_cli.py +0 -0
  63. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_client.py +0 -0
  64. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_document.py +0 -0
  65. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_embedder.py +0 -0
  66. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_lancedb_connection.py +0 -0
  67. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_monitor.py +0 -0
  68. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_qa.py +0 -0
  69. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_reader.py +0 -0
  70. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_rebuild.py +0 -0
  71. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_reranker.py +0 -0
  72. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_search.py +0 -0
  73. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_settings.py +0 -0
  74. {haiku_rag-0.7.0 → haiku_rag-0.7.1}/tests/test_utils.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: haiku.rag
3
- Version: 0.7.0
3
+ Version: 0.7.1
4
4
  Summary: Retrieval Augmented Generation (RAG) with LanceDB
5
5
  Author-email: Yiorgis Gozadinos <ggozadinos@gmail.com>
6
6
  License: MIT
@@ -1,6 +1,6 @@
1
1
  [project]
2
2
  name = "haiku.rag"
3
- version = "0.7.0"
3
+ version = "0.7.1"
4
4
  description = "Retrieval Augmented Generation (RAG) with LanceDB"
5
5
  authors = [{ name = "Yiorgis Gozadinos", email = "ggozadinos@gmail.com" }]
6
6
  license = { text = "MIT" }
@@ -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 OpenAIModel
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 OpenAIModel(
64
+ return OpenAIChatModel(
65
65
  model_name=model,
66
66
  provider=OllamaProvider(base_url=f"{Config.OLLAMA_BASE_URL}/v1"),
67
67
  )
@@ -6,6 +6,7 @@ from llm_judge import LLMJudge
6
6
  from rich.console import Console
7
7
  from rich.progress import Progress
8
8
 
9
+ from haiku.rag import logging # noqa
9
10
  from haiku.rag.client import HaikuRAG
10
11
  from haiku.rag.qa import get_qa_agent
11
12
 
@@ -951,7 +951,7 @@ wheels = [
951
951
 
952
952
  [[package]]
953
953
  name = "haiku-rag"
954
- version = "0.7.0"
954
+ version = "0.7.1"
955
955
  source = { editable = "." }
956
956
  dependencies = [
957
957
  { name = "docling" },
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