haiku.rag 0.7.0__py3-none-any.whl → 0.7.1__py3-none-any.whl

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 (7) hide show
  1. haiku/rag/logging.py +1 -0
  2. haiku/rag/qa/agent.py +2 -2
  3. {haiku_rag-0.7.0.dist-info → haiku_rag-0.7.1.dist-info}/METADATA +1 -1
  4. {haiku_rag-0.7.0.dist-info → haiku_rag-0.7.1.dist-info}/RECORD +7 -7
  5. {haiku_rag-0.7.0.dist-info → haiku_rag-0.7.1.dist-info}/WHEEL +0 -0
  6. {haiku_rag-0.7.0.dist-info → haiku_rag-0.7.1.dist-info}/entry_points.txt +0 -0
  7. {haiku_rag-0.7.0.dist-info → haiku_rag-0.7.1.dist-info}/licenses/LICENSE +0 -0
haiku/rag/logging.py CHANGED
@@ -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:
haiku/rag/qa/agent.py CHANGED
@@ -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
  )
{haiku_rag-0.7.0.dist-info → haiku_rag-0.7.1.dist-info}/METADATA RENAMED
@@ -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
{haiku_rag-0.7.0.dist-info → haiku_rag-0.7.1.dist-info}/RECORD RENAMED
@@ -4,7 +4,7 @@ haiku/rag/chunker.py,sha256=PVe6ysv8UlacUd4Zb3_8RFWIaWDXnzBAy2VDJ4TaUsE,1555
4
4
  haiku/rag/cli.py,sha256=UY9Vh5RsIxSCV14eQbNOiwToKmbFAvqTOAnxjieaYBs,6399
5
5
  haiku/rag/client.py,sha256=N4zkWjE9Rsw9YgPvNo83xptHUQR2ognfOnjkoV_w6hc,20999
6
6
  haiku/rag/config.py,sha256=9Mv0QJ3c6VF1oVRSXJlSsG234dCd_sKnJO-ybMaTpDQ,1690
7
- haiku/rag/logging.py,sha256=9RjS931Dkp0nSdJaCSCydi33jq4gXmLmIidONvD9qD4,731
7
+ haiku/rag/logging.py,sha256=DOQi9QMpQRl8h17Vu4nQh8HxpHdeIu29n8-HZaT3SRQ,786
8
8
  haiku/rag/mcp.py,sha256=bR9Y-Nz-hvjiql20Y0KE0hwNGwyjmPGX8K9d-qmXptY,4683
9
9
  haiku/rag/migration.py,sha256=gWxQwiKo0YulRhogYz4K8N98kHN9LQXIx9FeTmT24v4,10915
10
10
  haiku/rag/monitor.py,sha256=r386nkhdlsU8UECwIuVwnrSlgMk3vNIuUZGNIzkZuec,2770
@@ -16,7 +16,7 @@ haiku/rag/embeddings/ollama.py,sha256=y6-lp0XpbnyIjoOEdtSzMdEVkU5glOwnWQ1FkpUZnp
16
16
  haiku/rag/embeddings/openai.py,sha256=iA-DewCOSip8PLU_RhEJHFHBle4DtmCCIGNfGs58Wvk,357
17
17
  haiku/rag/embeddings/voyageai.py,sha256=0hiRTIqu-bpl-4OaCtMHvWfPdgbrzhnfZJowSV8pLRA,415
18
18
  haiku/rag/qa/__init__.py,sha256=Sl7Kzrg9CuBOcMF01wc1NtQhUNWjJI0MhIHfCWrb8V4,434
19
- haiku/rag/qa/agent.py,sha256=r6tYKvOW4W1HxBRHH1kmzlzb1bIJcQSuHd6cG9ANqXY,2594
19
+ haiku/rag/qa/agent.py,sha256=gZ12vLUSHHCMl0HyPoLlPDbhUWoyEUydXG7u8lG1eqg,2602
20
20
  haiku/rag/qa/prompts.py,sha256=xdT4cyrOrAK9UDgVqyev1wHF49jD57Bh40gx2sH4NPI,3341
21
21
  haiku/rag/reranking/__init__.py,sha256=IRXHs4qPu6VbGJQpzSwhgtVWWumURH_vEoVFE-extlo,894
22
22
  haiku/rag/reranking/base.py,sha256=LM9yUSSJ414UgBZhFTgxGprlRqzfTe4I1vgjricz2JY,405
@@ -32,8 +32,8 @@ haiku/rag/store/repositories/chunk.py,sha256=5S77mGh6pWxPHjaXriJGmvbSOhoNM8tLwyg
32
32
  haiku/rag/store/repositories/document.py,sha256=lP8Lo82KTP-qwXFRpYZ46WjeAdAsHwZ5pJcrXdz4g0U,6988
33
33
  haiku/rag/store/repositories/settings.py,sha256=dqnAvm-98nQrWpLBbf9QghJw673QD80-iqQhRMP5t0c,5025
34
34
  haiku/rag/store/upgrades/__init__.py,sha256=wUiEoSiHTahvuagx93E4FB07v123AhdbOjwUkPusiIg,14
35
- haiku_rag-0.7.0.dist-info/METADATA,sha256=FGx5ufhc35VUXqXLdbhmUEa-3hi7aaBuDeyZYf8xaaQ,4597
36
- haiku_rag-0.7.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
37
- haiku_rag-0.7.0.dist-info/entry_points.txt,sha256=G1U3nAkNd5YDYd4v0tuYFbriz0i-JheCsFuT9kIoGCI,48
38
- haiku_rag-0.7.0.dist-info/licenses/LICENSE,sha256=eXZrWjSk9PwYFNK9yUczl3oPl95Z4V9UXH7bPN46iPo,1065
39
- haiku_rag-0.7.0.dist-info/RECORD,,
35
+ haiku_rag-0.7.1.dist-info/METADATA,sha256=yIz4nATa_b2vDstrCZe9CzV_77Vw74QIhRWqCZCdaes,4597
36
+ haiku_rag-0.7.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
37
+ haiku_rag-0.7.1.dist-info/entry_points.txt,sha256=G1U3nAkNd5YDYd4v0tuYFbriz0i-JheCsFuT9kIoGCI,48
38
+ haiku_rag-0.7.1.dist-info/licenses/LICENSE,sha256=eXZrWjSk9PwYFNK9yUczl3oPl95Z4V9UXH7bPN46iPo,1065
39
+ haiku_rag-0.7.1.dist-info/RECORD,,