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

Fix double summarization, cleanup

Browse source
This commit is contained in:
Yiorgis Gozadinos 2026-02-13 12:01:28 +02:00
parent b64d9721bb
commit f5c562db9b
No known key found for this signature in database
2 changed files with 13 additions and 52 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
haiku_rag_slim/haiku/rag/agents/chat/agent.py
View file

@ -12,11 +12,7 @@ from haiku.rag.config.models import AppConfig
from haiku.rag.tools.context import ToolContext, prepare_context from haiku.rag.tools.context import ToolContext, prepare_context
from haiku.rag.tools.deps import AgentDeps from haiku.rag.tools.deps import AgentDeps
from haiku.rag.tools.document import create_document_toolset from haiku.rag.tools.document import create_document_toolset
from haiku.rag.tools.qa import ( from haiku.rag.tools.qa import QA_SESSION_NAMESPACE, QASessionState, create_qa_toolset
QA_SESSION_NAMESPACE,
QASessionState,
create_qa_toolset,
)
from haiku.rag.tools.search import create_search_toolset from haiku.rag.tools.search import create_search_toolset
from haiku.rag.tools.session import SessionContext from haiku.rag.tools.session import SessionContext
from haiku.rag.utils import get_model from haiku.rag.utils import get_model
@ -167,9 +163,7 @@ async def run_chat_agent(
deps: ChatDeps, deps: ChatDeps,
message: str, message: str,
) -> str: ) -> str:
"""Run the chat agent and trigger background summarization. """Run the chat agent.
This wrapper handles post-processing like background summarization.
Args: Args:
agent: The chat agent. agent: The chat agent.
@ -180,7 +174,6 @@ async def run_chat_agent(
Agent response. Agent response.
""" """
result = await agent.run(message, deps=deps) result = await agent.run(message, deps=deps)
trigger_background_summarization(deps)
return result.output return result.output

54
haiku_rag_slim/haiku/rag/tools/__init__.py
View file

@ -6,12 +6,7 @@ from haiku.rag.tools.context import (
prepare_context, prepare_context,
) )
from haiku.rag.tools.deps import AgentDeps from haiku.rag.tools.deps import AgentDeps
from haiku.rag.tools.document import ( from haiku.rag.tools.document import create_document_toolset
DocumentInfo,
DocumentListResponse,
create_document_toolset,
find_document,
)
from haiku.rag.tools.filters import ( from haiku.rag.tools.filters import (
build_document_filter, build_document_filter,
build_multi_document_filter, build_multi_document_filter,
@ -20,51 +15,24 @@ from haiku.rag.tools.filters import (
) )
from haiku.rag.tools.models import AnalysisResult, QAResult from haiku.rag.tools.models import AnalysisResult, QAResult
from haiku.rag.tools.prompts import build_tools_prompt from haiku.rag.tools.prompts import build_tools_prompt
from haiku.rag.tools.qa import ( from haiku.rag.tools.qa import create_qa_toolset
QA_SESSION_NAMESPACE, from haiku.rag.tools.search import create_search_toolset
QAHistoryEntry,
QASessionState,
create_qa_toolset,
run_qa_core,
)
from haiku.rag.tools.search import SEARCH_NAMESPACE, SearchState, create_search_toolset
from haiku.rag.tools.session import (
SESSION_NAMESPACE,
SessionContext,
SessionState,
compute_combined_state_delta,
compute_state_delta,
)
__all__ = [ __all__ = [
"AgentDeps", "AgentDeps",
"build_tools_prompt", "AnalysisResult",
"QAResult",
"RAGDeps", "RAGDeps",
"ToolContext", "ToolContext",
"ToolContextCache", "ToolContextCache",
"prepare_context",
"QAResult",
"AnalysisResult",
"build_document_filter", "build_document_filter",
"build_multi_document_filter", "build_multi_document_filter",
"build_tools_prompt",
"combine_filters", "combine_filters",
"get_session_filter",
"SEARCH_NAMESPACE",
"SearchState",
"create_search_toolset",
"DocumentInfo",
"DocumentListResponse",
"create_document_toolset",
"find_document",
"QA_SESSION_NAMESPACE",
"QASessionState",
"QAHistoryEntry",
"create_qa_toolset",
"run_qa_core",
"create_analysis_toolset", "create_analysis_toolset",
"SESSION_NAMESPACE", "create_document_toolset",
"SessionContext", "create_qa_toolset",
"SessionState", "create_search_toolset",
"compute_state_delta", "get_session_filter",
"compute_combined_state_delta", "prepare_context",
] ]