haiku.rag 0.3.1__py3-none-any.whl → 0.3.2__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.
haiku/rag/qa/__init__.py CHANGED
@@ -8,7 +8,6 @@ def get_qa_agent(client: HaikuRAG, model: str = "") -> QuestionAnswerAgentBase:
8
8
  """
9
9
  Factory function to get the appropriate QA agent based on the configuration.
10
10
  """
11
-
12
11
  if Config.QA_PROVIDER == "ollama":
13
12
  return QuestionAnswerOllamaAgent(client, model or Config.QA_MODEL)
14
13
 
@@ -21,7 +20,7 @@ def get_qa_agent(client: HaikuRAG, model: str = "") -> QuestionAnswerAgentBase:
21
20
  "Please install haiku.rag with the 'openai' extra:"
22
21
  "uv pip install haiku.rag --extra openai"
23
22
  )
24
- return QuestionAnswerOpenAIAgent(client, model or "gpt-4o-mini")
23
+ return QuestionAnswerOpenAIAgent(client, model or Config.QA_MODEL)
25
24
 
26
25
  if Config.QA_PROVIDER == "anthropic":
27
26
  try:
@@ -32,8 +31,6 @@ def get_qa_agent(client: HaikuRAG, model: str = "") -> QuestionAnswerAgentBase:
32
31
  "Please install haiku.rag with the 'anthropic' extra:"
33
32
  "uv pip install haiku.rag --extra anthropic"
34
33
  )
35
- return QuestionAnswerAnthropicAgent(
36
- client, model or "claude-3-5-haiku-20241022"
37
- )
34
+ return QuestionAnswerAnthropicAgent(client, model or Config.QA_MODEL)
38
35
 
39
36
  raise ValueError(f"Unsupported QA provider: {Config.QA_PROVIDER}")
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: haiku.rag
3
- Version: 0.3.1
3
+ Version: 0.3.2
4
4
  Summary: Retrieval Augmented Generation (RAG) with SQLite
5
5
  Author-email: Yiorgis Gozadinos <ggozadinos@gmail.com>
6
6
  License: MIT
@@ -14,7 +14,7 @@ haiku/rag/embeddings/base.py,sha256=PTAWKTU-Q-hXIhbRK1o6pIdpaW7DFdzJXQ0Nzc6VI-w,
14
14
  haiku/rag/embeddings/ollama.py,sha256=hWdrTiuJwNSRYCqP0WP-z6XXA3RBGkAiknZMsPLH0qU,441
15
15
  haiku/rag/embeddings/openai.py,sha256=reh8AykG2f9f5hhRDmqSsjiuCPi9SsXfe2YEZFlxXk8,550
16
16
  haiku/rag/embeddings/voyageai.py,sha256=jc0JywdLJD3Ee1MUv1m8MhWCEo0enNnVcrIBtUvD-Ss,534
17
- haiku/rag/qa/__init__.py,sha256=xN36Sw5xj3rHiI3D9YGSoX4ywK0sSkmHnG0uf_3bj08,1534
17
+ haiku/rag/qa/__init__.py,sha256=oso98Ypti7mBLTJ6Zk71YaSJ9Rgc89QXp9RSB6zSpYs,1501
18
18
  haiku/rag/qa/anthropic.py,sha256=lzHRQxpEv6Qd6iBIqexUgWnq-ITqytppwkfOuRGWdDs,4556
19
19
  haiku/rag/qa/base.py,sha256=4ZTM_l5FAZ9cA0f8NeqRJiUAmjatwCTmSoclFw0gTFQ,1349
20
20
  haiku/rag/qa/ollama.py,sha256=poShrse-RgLTwa5gbVzoERNTrn5QRpovJCZKYkIpOZI,2393
@@ -29,8 +29,8 @@ haiku/rag/store/repositories/__init__.py,sha256=uIBhxjQh-4o3O-ck8b7BQ58qXQTuJdPv
29
29
  haiku/rag/store/repositories/base.py,sha256=cm3VyQXhtxvRfk1uJHpA0fDSxMpYN-mjQmRiDiLsQ68,1008
30
30
  haiku/rag/store/repositories/chunk.py,sha256=gik7ZPOK3gCoG6tU1pGueAZBPmJxIb7obYFUhwINrYg,16497
31
31
  haiku/rag/store/repositories/document.py,sha256=xpWOpjHFbhVwNJ1gpusEKNY6l_Qyibg9y_bdHCwcfpk,7133
32
- haiku_rag-0.3.1.dist-info/METADATA,sha256=_1rJ4s0aq82EkBRPfaPmRZ84QGYTfACyV5V_hk3F118,3931
33
- haiku_rag-0.3.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
34
- haiku_rag-0.3.1.dist-info/entry_points.txt,sha256=G1U3nAkNd5YDYd4v0tuYFbriz0i-JheCsFuT9kIoGCI,48
35
- haiku_rag-0.3.1.dist-info/licenses/LICENSE,sha256=eXZrWjSk9PwYFNK9yUczl3oPl95Z4V9UXH7bPN46iPo,1065
36
- haiku_rag-0.3.1.dist-info/RECORD,,
32
+ haiku_rag-0.3.2.dist-info/METADATA,sha256=0A8BVZDp38i_xLznvkrZBq3f3OYtWPtqBx_U2eHRIow,3931
33
+ haiku_rag-0.3.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
34
+ haiku_rag-0.3.2.dist-info/entry_points.txt,sha256=G1U3nAkNd5YDYd4v0tuYFbriz0i-JheCsFuT9kIoGCI,48
35
+ haiku_rag-0.3.2.dist-info/licenses/LICENSE,sha256=eXZrWjSk9PwYFNK9yUczl3oPl95Z4V9UXH7bPN46iPo,1065
36
+ haiku_rag-0.3.2.dist-info/RECORD,,