danvics/haiku.rag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add max retries=3 back

Browse source
This commit is contained in:
Yiorgis Gozadinos 2026-07-19 12:12:52 +03:00
parent 9e7db72997
commit 22ffd5e92e
No known key found for this signature in database
3 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
haiku_rag_slim/haiku/rag/capabilities/analysis.py
View file

@ -116,7 +116,9 @@ class AnalysisCapability(RAGCapabilityBase[AnalysisState]):
return await self._with_state(self._cite(chunk_ids)) return await self._with_state(self._cite(chunk_ids))
return FunctionToolset( return FunctionToolset(
[analysis_search, analysis_execute_code, analysis_cite], id=_CAPABILITY_ID [analysis_search, analysis_execute_code, analysis_cite],
id=_CAPABILITY_ID,
max_retries=3,
) )

2
haiku_rag_slim/haiku/rag/capabilities/rag.py
View file

@ -57,7 +57,7 @@ class RAGCapability(RAGCapabilityBase[RAGState]):
"""Register exact search-result chunk IDs as citations for the answer.""" """Register exact search-result chunk IDs as citations for the answer."""
return await self._with_state(self._cite(chunk_ids)) return await self._with_state(self._cite(chunk_ids))
return FunctionToolset([rag_search, rag_cite], id=_CAPABILITY_ID) return FunctionToolset([rag_search, rag_cite], id=_CAPABILITY_ID, max_retries=3)
def create_capability( def create_capability(

2
tests/capabilities/test_capabilities.py
View file

@ -44,6 +44,7 @@ def test_rag_capability_api(temp_db_path):
assert capability.id == "haiku-rag" assert capability.id == "haiku-rag"
assert capability.defer_loading is True assert capability.defer_loading is True
assert set(capability.get_toolset().tools) == {"rag_search", "rag_cite"} assert set(capability.get_toolset().tools) == {"rag_search", "rag_cite"}
assert capability.get_toolset().max_retries == 3
assert capability.state_type is RAGState assert capability.state_type is RAGState
assert capability.state_namespace == "rag" assert capability.state_namespace == "rag"
@ -59,6 +60,7 @@ def test_analysis_capability_api(temp_db_path):
"analysis_execute_code", "analysis_execute_code",
"analysis_cite", "analysis_cite",
} }
assert capability.get_toolset().max_retries == 3
assert capability.state_type is AnalysisState assert capability.state_type is AnalysisState