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 +2 -5
- {haiku_rag-0.3.1.dist-info → haiku_rag-0.3.2.dist-info}/METADATA +1 -1
- {haiku_rag-0.3.1.dist-info → haiku_rag-0.3.2.dist-info}/RECORD +6 -6
- {haiku_rag-0.3.1.dist-info → haiku_rag-0.3.2.dist-info}/WHEEL +0 -0
- {haiku_rag-0.3.1.dist-info → haiku_rag-0.3.2.dist-info}/entry_points.txt +0 -0
- {haiku_rag-0.3.1.dist-info → haiku_rag-0.3.2.dist-info}/licenses/LICENSE +0 -0
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
|
|
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}")
|
|
@@ -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=
|
|
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.
|
|
33
|
-
haiku_rag-0.3.
|
|
34
|
-
haiku_rag-0.3.
|
|
35
|
-
haiku_rag-0.3.
|
|
36
|
-
haiku_rag-0.3.
|
|
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,,
|
|
File without changes
|
|
File without changes
|
|
File without changes
|