haiku.rag-slim 0.16.0__py3-none-any.whl → 0.24.0__py3-none-any.whl

haiku/rag/graph/deep_qa/prompts.py

@@ -1,59 +0,0 @@
-"""Deep QA specific prompts."""
-
-SYNTHESIS_PROMPT = """You are an expert at synthesizing information into clear, concise answers.
-
-Task:
-- Combine the gathered information from sub-questions into a single comprehensive answer
-- Answer the original question directly and completely
-- Base your answer strictly on the provided evidence
-- Be clear, accurate, and well-structured
-
-Output format:
-- answer: The complete answer to the original question (2-4 paragraphs)
-- sources: List of document titles/URIs used (extract from the sub-answers)
-
-Guidelines:
-- Start directly with the answer - no preamble like "Based on the research..."
-- Use a clear, professional tone
-- Organize information logically
-- If evidence is incomplete, state limitations clearly
-- Do not include any claims not supported by the gathered information"""
-
-SYNTHESIS_PROMPT_WITH_CITATIONS = """You are an expert at synthesizing information into clear, concise answers with proper citations.
-
-Task:
-- Combine the gathered information from sub-questions into a single comprehensive answer
-- Answer the original question directly and completely
-- Base your answer strictly on the provided evidence
-- Include inline citations using [Source Title] format
-
-Output format:
-- answer: The complete answer with inline citations (2-4 paragraphs)
-- sources: List of document titles/URIs used (extract from the sub-answers)
-
-Guidelines:
-- Start directly with the answer - no preamble like "Based on the research..."
-- Add citations after each claim: [Source Title]
-- Use a clear, professional tone
-- Organize information logically
-- If evidence is incomplete, state limitations clearly
-- Do not include any claims not supported by the gathered information"""
-
-DECISION_PROMPT = """You are an expert at evaluating whether gathered information is sufficient to answer a question.
-
-Task:
-- Review the original question and all gathered sub-question answers
-- Determine if we have enough information to provide a comprehensive answer
-- If insufficient, suggest specific new sub-questions to fill the gaps
-
-Output format:
-- is_sufficient: Boolean indicating if we can answer the question comprehensively
-- reasoning: Clear explanation of your assessment
-- new_questions: List of specific follow-up questions needed (empty if sufficient)
-
-Guidelines:
-- Be strict but reasonable in your assessment
-- Focus on whether core aspects of the question are addressed
-- New questions should be specific and distinct from what's been asked
-- Limit new questions to 2-3 maximum
-- Consider whether additional searches would meaningfully improve the answer"""

haiku/rag/graph/deep_qa/state.py

@@ -1,56 +0,0 @@
-import asyncio
-from dataclasses import dataclass
-from typing import TYPE_CHECKING
-
-from pydantic import BaseModel, Field
-
-from haiku.rag.client import HaikuRAG
-from haiku.rag.graph.deep_qa.dependencies import DeepQAContext
-from haiku.rag.graph.deep_qa.models import DeepQAAnswer
-
-if TYPE_CHECKING:
-    from haiku.rag.config.models import AppConfig
-    from haiku.rag.graph.agui.emitter import AGUIEmitter
-
-
-@dataclass
-class DeepQADeps:
-    client: HaikuRAG
-    agui_emitter: "AGUIEmitter[DeepQAState, DeepQAAnswer] | None" = None
-    semaphore: asyncio.Semaphore | None = None
-
-
-class DeepQAState(BaseModel):
-    """Deep QA state for multi-agent question answering."""
-
-    model_config = {"arbitrary_types_allowed": True}
-
-    context: DeepQAContext = Field(description="Shared QA context")
-    max_sub_questions: int = Field(
-        default=3, description="Maximum number of sub-questions"
-    )
-    max_iterations: int = Field(
-        default=2, description="Maximum number of QA iterations"
-    )
-    max_concurrency: int = Field(
-        default=1, description="Maximum parallel sub-question searches"
-    )
-    iterations: int = Field(default=0, description="Current iteration number")
-
-    @classmethod
-    def from_config(cls, context: DeepQAContext, config: "AppConfig") -> "DeepQAState":
-        """Create a DeepQAState from an AppConfig.
-
-        Args:
-            context: The DeepQAContext containing the question and settings
-            config: The AppConfig object (uses config.qa for state parameters)
-
-        Returns:
-            A configured DeepQAState instance
-        """
-        return cls(
-            context=context,
-            max_sub_questions=config.qa.max_sub_questions,
-            max_iterations=config.qa.max_iterations,
-            max_concurrency=config.qa.max_concurrency,
-        )

haiku/rag/graph/research/common.py

@@ -1,87 +0,0 @@
-from pydantic_ai import format_as_xml
-
-from haiku.rag.graph.research.dependencies import ResearchContext
-from haiku.rag.graph.research.models import InsightAnalysis
-
-
-def format_context_for_prompt(context: ResearchContext) -> str:
-    """Format the research context as XML for inclusion in prompts."""
-
-    context_data = {
-        "original_question": context.original_question,
-        "unanswered_questions": context.sub_questions,
-        "qa_responses": [
-            {
-                "question": qa.query,
-                "answer": qa.answer,
-                "context_snippets": qa.context,
-                "sources": qa.sources,  # pyright: ignore[reportAttributeAccessIssue]
-            }
-            for qa in context.qa_responses
-        ],
-        "insights": [
-            {
-                "id": insight.id,
-                "summary": insight.summary,
-                "status": insight.status.value,
-                "supporting_sources": insight.supporting_sources,
-                "originating_questions": insight.originating_questions,
-                "notes": insight.notes,
-            }
-            for insight in context.insights
-        ],
-        "gaps": [
-            {
-                "id": gap.id,
-                "description": gap.description,
-                "severity": gap.severity.value,
-                "blocking": gap.blocking,
-                "resolved": gap.resolved,
-                "resolved_by": gap.resolved_by,
-                "supporting_sources": gap.supporting_sources,
-                "notes": gap.notes,
-            }
-            for gap in context.gaps
-        ],
-    }
-    return format_as_xml(context_data, root_tag="research_context")
-
-
-def format_analysis_for_prompt(
-    analysis: InsightAnalysis | None,
-) -> str:
-    """Format the latest insight analysis as XML for prompts."""
-
-    if analysis is None:
-        return "<latest_analysis />"
-
-    data = {
-        "commentary": analysis.commentary,
-        "highlights": [
-            {
-                "id": insight.id,
-                "summary": insight.summary,
-                "status": insight.status.value,
-                "supporting_sources": insight.supporting_sources,
-                "originating_questions": insight.originating_questions,
-                "notes": insight.notes,
-            }
-            for insight in analysis.highlights
-        ],
-        "gap_assessments": [
-            {
-                "id": gap.id,
-                "description": gap.description,
-                "severity": gap.severity.value,
-                "blocking": gap.blocking,
-                "resolved": gap.resolved,
-                "resolved_by": gap.resolved_by,
-                "supporting_sources": gap.supporting_sources,
-                "notes": gap.notes,
-            }
-            for gap in analysis.gap_assessments
-        ],
-        "resolved_gaps": analysis.resolved_gaps,
-        "new_questions": analysis.new_questions,
-    }
-    return format_as_xml(data, root_tag="latest_analysis")

haiku/rag/reader.py

@@ -1,135 +0,0 @@
-from pathlib import Path
-from typing import ClassVar
-
-from docling_core.types.doc.document import DoclingDocument
-
-from haiku.rag.utils import text_to_docling_document
-
-# Check if docling is available
-try:
-    import docling  # noqa: F401
-
-    DOCLING_AVAILABLE = True
-except ImportError:
-    DOCLING_AVAILABLE = False
-
-
-class FileReader:
-    # Extensions supported by docling
-    docling_extensions: ClassVar[list[str]] = [
-        ".adoc",
-        ".asc",
-        ".asciidoc",
-        ".bmp",
-        ".csv",
-        ".docx",
-        ".html",
-        ".xhtml",
-        ".jpeg",
-        ".jpg",
-        ".md",
-        ".pdf",
-        ".png",
-        ".pptx",
-        ".tiff",
-        ".xlsx",
-        ".xml",
-        ".webp",
-    ]
-
-    # Plain text extensions that we'll read directly
-    text_extensions: ClassVar[list[str]] = [
-        ".astro",
-        ".c",
-        ".cpp",
-        ".css",
-        ".go",
-        ".h",
-        ".hpp",
-        ".java",
-        ".js",
-        ".json",
-        ".kt",
-        ".mdx",
-        ".mjs",
-        ".php",
-        ".py",
-        ".rb",
-        ".rs",
-        ".svelte",
-        ".swift",
-        ".ts",
-        ".tsx",
-        ".txt",
-        ".vue",
-        ".yaml",
-        ".yml",
-    ]
-
-    # Code file extensions with their markdown language identifiers for syntax highlighting
-    code_markdown_identifier: ClassVar[dict[str, str]] = {
-        ".astro": "astro",
-        ".c": "c",
-        ".cpp": "cpp",
-        ".css": "css",
-        ".go": "go",
-        ".h": "c",
-        ".hpp": "cpp",
-        ".java": "java",
-        ".js": "javascript",
-        ".json": "json",
-        ".kt": "kotlin",
-        ".mjs": "javascript",
-        ".php": "php",
-        ".py": "python",
-        ".rb": "ruby",
-        ".rs": "rust",
-        ".svelte": "svelte",
-        ".swift": "swift",
-        ".ts": "typescript",
-        ".tsx": "tsx",
-        ".vue": "vue",
-        ".yaml": "yaml",
-        ".yml": "yaml",
-    }
-
-    extensions: ClassVar[list[str]] = docling_extensions + text_extensions
-
-    @staticmethod
-    def parse_file(path: Path) -> DoclingDocument:
-        try:
-            file_extension = path.suffix.lower()
-
-            if file_extension in FileReader.docling_extensions:
-                # Use docling for complex document formats
-                if not DOCLING_AVAILABLE:
-                    raise ImportError(
-                        "Docling is required for processing this file type. "
-                        "Install with: pip install haiku.rag-slim[docling]"
-                    )
-                from docling.document_converter import DocumentConverter
-
-                converter = DocumentConverter()
-                result = converter.convert(path)
-                return result.document
-            elif file_extension in FileReader.text_extensions:
-                # Read plain text files directly
-                content = path.read_text(encoding="utf-8")
-
-                # Wrap code files (but not plain txt) in markdown code blocks for better presentation
-                if file_extension in FileReader.code_markdown_identifier:
-                    language = FileReader.code_markdown_identifier[file_extension]
-                    content = f"```{language}\n{content}\n```"
-
-                # Convert text to DoclingDocument by wrapping as markdown
-                return text_to_docling_document(content, name=f"{path.stem}.md")
-            else:
-                # Fallback: try to read as text and convert to DoclingDocument
-                content = path.read_text(encoding="utf-8")
-                return text_to_docling_document(content, name=f"{path.stem}.md")
-        except ImportError:
-            raise
-        except Exception as e:
-            raise ValueError(
-                f"Failed to parse file: {path} - {type(e).__name__}: {e}"
-            ) from e

haiku_rag_slim-0.16.0.dist-info/RECORD

@@ -1,71 +0,0 @@
-haiku/rag/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-haiku/rag/app.py,sha256=pBauUXYs6Bme4c4qoeYvOVxQxuLL-BYmorA3TSeTZKQ,22363
-haiku/rag/chunker.py,sha256=V3Sm47BY9vAQlzXuGKJiUMA22-XvAZXLkHX37pwvXcY,1926
-haiku/rag/cli.py,sha256=eVj2Muo07TiI5Tssx6rYMxoMvrtfzvYvt2YX0Trj1a4,12593
-haiku/rag/client.py,sha256=cYaThPAf9sdBxdC-WusnC_2MMcjcFHHkEOTDMjTJHWs,27848
-haiku/rag/logging.py,sha256=dm65AwADpcQsH5OAPtRA-4hsw0w5DK-sGOvzYkj6jzw,1720
-haiku/rag/mcp.py,sha256=ld3xLkPDzLGcoo2tUmN-TXY6XIxfIky3PUZdA24VZPQ,8729
-haiku/rag/monitor.py,sha256=aM92Q36-RDYMabaOk-TK7ok3C865E9aB63ObvLOZkNU,6978
-haiku/rag/reader.py,sha256=dK3zayFyNQcay-ROB_M3n8r0s4sLj3Y2sNCFoF4Ot9I,3846
-haiku/rag/utils.py,sha256=VExKww-W2ZQU0peougb89JNfsU6JVr9goad1qUb66dc,6411
-haiku/rag/config/__init__.py,sha256=U12iAoS5V1oW7wOSkAOvuy_cB5bbSsajtsye57Hu5B8,1636
-haiku/rag/config/loader.py,sha256=mhvoPoTXYGPx0ASeJ22rvRl0vQGtNrTzcKm--vXGBfQ,2752
-haiku/rag/config/models.py,sha256=AwgEuIFiKkhxxZ6bR4oBRVbrjEc1bSjI1Wg0zIv2rkg,2701
-haiku/rag/embeddings/__init__.py,sha256=zwWRU9S5YGEJxlgPv5haHBgj3LUJMe-dEwr3LKLa9RY,1731
-haiku/rag/embeddings/base.py,sha256=kzca54e2HGzS_0YKt9OLESM9lrFKpBm_97V07jx0aas,783
-haiku/rag/embeddings/ollama.py,sha256=_uIIObbZX9QVU1lcgWQFooA3b-AeZRNncM7yQ2TxlEU,825
-haiku/rag/embeddings/openai.py,sha256=BfmPni567DH8KqwLCPiOmr3q-dpzpOJkvFFoUuTR5as,731
-haiku/rag/embeddings/vllm.py,sha256=IZFS3pbvLXkhvdT7pFVi2csFlNTSza5bpybwz7ud3Po,847
-haiku/rag/embeddings/voyageai.py,sha256=6vEuk6q510AJv-K2lL93P2dVrziAjELTOe_w_Zp5YT4,917
-haiku/rag/graph/__init__.py,sha256=rQJKwsr52NLhqjLvdGlA9Svpm7iAYIPutH_8w-aATWM,646
-haiku/rag/graph/agui/__init__.py,sha256=Mh3l373-LAT1sR7pubTbNQk5UOqkxWXlp-_VAb5bb_8,1351
-haiku/rag/graph/agui/cli_renderer.py,sha256=_IVkPZWUrFJsKtA2fNI_GRMx-tfOUmBfKFZwttqiZZY,5187
-haiku/rag/graph/agui/emitter.py,sha256=k5lEAASa0lC0Gthsxi-mNFWyI6GpR697EKkoL9E1SkE,6445
-haiku/rag/graph/agui/events.py,sha256=5XGVEfFX2KqywkrQOgW5v3dtluwGswb5LgDyEnIK1GQ,6033
-haiku/rag/graph/agui/server.py,sha256=muLq9D-ekejlfT7p1NpPy7RDFIX5CoE5fN87_wEFN28,10497
-haiku/rag/graph/agui/state.py,sha256=LkuuAY9w32pc0kkXkLJvyNGC0JzhXn05IfIVZzCXAv0,965
-haiku/rag/graph/agui/stream.py,sha256=Z3AVzgcPf2S89c7JBdd_p62VDj7AuMKKr3aroZahnsw,2693
-haiku/rag/graph/common/__init__.py,sha256=nqxh6QViWuUw-cq1E8CWfqWciAh17dbfufIABK1wPsw,127
-haiku/rag/graph/common/models.py,sha256=Nwzw8TZ_J0aEdenU5to14OIMyDROGYvAbBtVIlC4ZXM,1373
-haiku/rag/graph/common/nodes.py,sha256=r9GxXPaGPAF_442pQuMrD6P12GI1ZRTNxGg43hFrfmk,9119
-haiku/rag/graph/common/prompts.py,sha256=DTEk4QN1uRHUZRSPyAhGPJjQNLVASQOWkb4Rhc_gnv4,2276
-haiku/rag/graph/common/utils.py,sha256=QoOIkgIRxIJ8nwFrcbHGzJC7MlzzUb8hfVZlyzHa7dY,1458
-haiku/rag/graph/deep_qa/__init__.py,sha256=BajDKA9m47hoc-alGzQjhOPFirnl6z3vtJod8iYSLQg,56
-haiku/rag/graph/deep_qa/dependencies.py,sha256=Shdj7z9Q1iHpGsuZbiFXVMHjx5_lZA4agHHMOHd__Lk,937
-haiku/rag/graph/deep_qa/graph.py,sha256=2PpJgHQHsgZ4nsVWuG3ytGPs4tEiDRsWMXlh1y1NjIQ,8017
-haiku/rag/graph/deep_qa/models.py,sha256=siZMQXD21_3nk8kaLCv0BCuD9TydLYo-yC4-9CxQy3E,683
-haiku/rag/graph/deep_qa/prompts.py,sha256=XtESQvlAvMpJdE2TX9OZYY55yMFmsFT1h6V6QhnvTCA,2598
-haiku/rag/graph/deep_qa/state.py,sha256=EPUGHbLFUWeJzfx43koQctL7HiB9EsMtO8hfw3t3JzY,1823
-haiku/rag/graph/research/__init__.py,sha256=eNTG4ujgxnY9-7pTJ-fw7xcqOr4MHXQx-P_BH6cA15Q,220
-haiku/rag/graph/research/common.py,sha256=3QtESionc7hxwaInAwk7dFnNpqAoG7h4briTdCf3gDk,3000
-haiku/rag/graph/research/dependencies.py,sha256=6Swm-XJ37GBdJe8Ynd1ZwRjM4CJODmNSfNgrStC9yMo,6100
-haiku/rag/graph/research/graph.py,sha256=llka6emUBAyRUTsBCUX5Jwj7_JUFaUDW4e2kbFH3rH8,10408
-haiku/rag/graph/research/models.py,sha256=U68SdBKvNz8CSX9V3cj3PTevMyiShm6JyASUD0t2kCU,5375
-haiku/rag/graph/research/prompts.py,sha256=opz4MXjoDHH1wjG6bPyiqT0LVzk3pBA6y_a9zpBW8yM,4834
-haiku/rag/graph/research/state.py,sha256=fmuL30bNO5xY4YjZx8vmcr8N2DcoIBKJ6i5BDnO6MoU,2777
-haiku/rag/qa/__init__.py,sha256=Q18B5cjgYSuOdzwsJkXDeqcclAI2pu3tBIcWLcMTT5M,949
-haiku/rag/qa/agent.py,sha256=ReuvluxVzaH82PhrFLNAAM3rVrSj-sKHkhki266SsGI,3181
-haiku/rag/qa/prompts.py,sha256=Lqwn3m4zCsu_CJiC4s9cLsuPNbb9nq6j2PqEF3lw1eA,3380
-haiku/rag/reranking/__init__.py,sha256=cwkydVEJr7Tgs4uAWB057y9j5N3F1BDO-71YJNVkL-s,1900
-haiku/rag/reranking/base.py,sha256=Yji15nAR8LyIJGqZvEZifTWmortNQ4k_7ZHst_5mRYk,408
-haiku/rag/reranking/cohere.py,sha256=BhBPPnaSnDoVlkL_MHF74kegXQBrsZGKnWqC40ztiAk,1050
-haiku/rag/reranking/mxbai.py,sha256=qR55dmpaBz15lSN_wXD3-Z6Kqr_bmNKU9q4Pwef_wB8,911
-haiku/rag/reranking/vllm.py,sha256=Ip83qzV2RM7qXTj0mE2St66hvXykovoNW8Hu3AUebDc,1489
-haiku/rag/reranking/zeroentropy.py,sha256=bVW5gcdSEz8A97xVSD0jhWGN1l4lUZ10I-5vufINGKE,1913
-haiku/rag/store/__init__.py,sha256=R2IRcxtkFDxqa2sgMirqLq3l2-FPdWr6ydYStaqm5OQ,104
-haiku/rag/store/engine.py,sha256=LTv2_QXqkTJqwHTOKIwzvEBodBZAax3zkW9MtJciMjo,11696
-haiku/rag/store/models/__init__.py,sha256=kc7Ctf53Jr483tk4QTIrcgqBbXDz4ZoeYSkFXfPnpks,89
-haiku/rag/store/models/chunk.py,sha256=3EuZav4QekJIeHBCub48EM8SjNX8HEJ6wVDXGot4PEQ,421
-haiku/rag/store/models/document.py,sha256=cZXy_jEti-hnhq7FKhuhCfd99ccY9fIHMLovB_Thbb8,425
-haiku/rag/store/repositories/__init__.py,sha256=Olv5dLfBQINRV3HrsfUpjzkZ7Qm7goEYyMNykgo_DaY,291
-haiku/rag/store/repositories/chunk.py,sha256=wlC_D_Vu_CjHvr2dfXV6qG7lyQExERHSsOlJMtAwldk,15930
-haiku/rag/store/repositories/document.py,sha256=JKpDUQmYgSqbc2eUKPYz9MX0x--oOHCdI0tNEib1Yqw,8704
-haiku/rag/store/repositories/settings.py,sha256=15gS7Xj7cG4qetv_ioxZO_r31by7GuSqtpowOsMkHmc,6129
-haiku/rag/store/upgrades/__init__.py,sha256=RQ8A6rEXBASLb5PD9vdDnEas_m_GgRzzdVu4B88Snqc,1975
-haiku/rag/store/upgrades/v0_10_1.py,sha256=qNGnxj6hoHaHJ1rKTiALfw0c9NQOi0KAK-VZCD_073A,1959
-haiku/rag/store/upgrades/v0_9_3.py,sha256=NrjNilQSgDtFWRbL3ZUtzQzJ8tf9u0dDRJtnDFwwbdw,3322
-haiku_rag_slim-0.16.0.dist-info/METADATA,sha256=7BjfptrvrZgWin4NuyVLmZUGMwzIpeNfw5-_CakZJk4,4192
-haiku_rag_slim-0.16.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-haiku_rag_slim-0.16.0.dist-info/entry_points.txt,sha256=G1U3nAkNd5YDYd4v0tuYFbriz0i-JheCsFuT9kIoGCI,48
-haiku_rag_slim-0.16.0.dist-info/licenses/LICENSE,sha256=eXZrWjSk9PwYFNK9yUczl3oPl95Z4V9UXH7bPN46iPo,1065
-haiku_rag_slim-0.16.0.dist-info/RECORD,,