haystack-experimental 0.15.2__py3-none-any.whl → 0.17.0__py3-none-any.whl

haystack_experimental/components/agents/human_in_the_loop/dataclasses.py

@@ -3,7 +3,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 from dataclasses import asdict, dataclass
-from typing import Any, Optional
+from typing import Any
 
 
 @dataclass
@@ -23,8 +23,8 @@ class ConfirmationUIResult:
     """
 
     action: str  # "confirm", "reject", "modify"
-    feedback: Optional[str] = None
-    new_tool_params: Optional[dict[str, Any]] = None
+    feedback: str | None = None
+    new_tool_params: dict[str, Any] | None = None
 
 
 @dataclass
@@ -49,9 +49,9 @@ class ToolExecutionDecision:
 
     tool_name: str
     execute: bool
-    tool_call_id: Optional[str] = None
-    feedback: Optional[str] = None
-    final_tool_params: Optional[dict[str, Any]] = None
+    tool_call_id: str | None = None
+    feedback: str | None = None
+    final_tool_params: dict[str, Any] | None = None
 
     def to_dict(self) -> dict[str, Any]:
         """

haystack_experimental/components/agents/human_in_the_loop/errors.py

@@ -2,17 +2,13 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-from typing import Optional
-
 
 class HITLBreakpointException(Exception):
     """
     Exception raised when a tool execution is paused by a ConfirmationStrategy (e.g. BreakpointConfirmationStrategy).
     """
 
-    def __init__(
-        self, message: str, tool_name: str, snapshot_file_path: str, tool_call_id: Optional[str] = None
-    ) -> None:
+    def __init__(self, message: str, tool_name: str, snapshot_file_path: str, tool_call_id: str | None = None) -> None:
         """
         Initialize the HITLBreakpointException.
 

haystack_experimental/components/agents/human_in_the_loop/strategies.py

@@ -3,7 +3,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 from dataclasses import replace
-from typing import TYPE_CHECKING, Any, Optional
+from typing import TYPE_CHECKING, Any
 
 from haystack.components.agents.state import State
 from haystack.components.tools.tool_invoker import ToolInvoker
@@ -52,8 +52,8 @@ class BlockingConfirmationStrategy:
         tool_name: str,
         tool_description: str,
         tool_params: dict[str, Any],
-        tool_call_id: Optional[str] = None,
-        confirmation_strategy_context: Optional[dict[str, Any]] = None,
+        tool_call_id: str | None = None,
+        confirmation_strategy_context: dict[str, Any] | None = None,
     ) -> ToolExecutionDecision:
         """
         Run the human-in-the-loop strategy for a given tool and its parameters.
@@ -125,8 +125,8 @@ class BlockingConfirmationStrategy:
         tool_name: str,
         tool_description: str,
         tool_params: dict[str, Any],
-        tool_call_id: Optional[str] = None,
-        confirmation_strategy_context: Optional[dict[str, Any]] = None,
+        tool_call_id: str | None = None,
+        confirmation_strategy_context: dict[str, Any] | None = None,
     ) -> ToolExecutionDecision:
         """
         Async version of run. Calls the sync run() method by default.
@@ -210,8 +210,8 @@ class BreakpointConfirmationStrategy:
         tool_name: str,
         tool_description: str,
         tool_params: dict[str, Any],
-        tool_call_id: Optional[str] = None,
-        confirmation_strategy_context: Optional[dict[str, Any]] = None,
+        tool_call_id: str | None = None,
+        confirmation_strategy_context: dict[str, Any] | None = None,
     ) -> ToolExecutionDecision:
         """
         Run the breakpoint confirmation strategy for a given tool and its parameters.
@@ -248,8 +248,8 @@ class BreakpointConfirmationStrategy:
         tool_name: str,
         tool_description: str,
         tool_params: dict[str, Any],
-        tool_call_id: Optional[str] = None,
-        confirmation_strategy_context: Optional[dict[str, Any]] = None,
+        tool_call_id: str | None = None,
+        confirmation_strategy_context: dict[str, Any] | None = None,
     ) -> ToolExecutionDecision:
         """
         Async version of run. Calls the sync run() method.
@@ -304,7 +304,7 @@ def _prepare_tool_args(
     tool: Tool,
     tool_call_arguments: dict[str, Any],
     state: State,
-    streaming_callback: Optional[StreamingCallbackT] = None,
+    streaming_callback: StreamingCallbackT | None = None,
     enable_streaming_passthrough: bool = False,
 ) -> dict[str, Any]:
     """

haystack_experimental/components/agents/human_in_the_loop/types.py

@@ -2,7 +2,7 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-from typing import Any, Optional, Protocol
+from typing import Any, Protocol
 
 from haystack.core.serialization import default_from_dict, default_to_dict
 
@@ -67,8 +67,8 @@ class ConfirmationStrategy(Protocol):
         tool_name: str,
         tool_description: str,
         tool_params: dict[str, Any],
-        tool_call_id: Optional[str] = None,
-        **kwargs: Optional[dict[str, Any]],
+        tool_call_id: str | None = None,
+        **kwargs: dict[str, Any] | None,
     ) -> ToolExecutionDecision:
         """
         Run the confirmation strategy for a given tool and its parameters.
@@ -92,8 +92,8 @@ class ConfirmationStrategy(Protocol):
         tool_name: str,
         tool_description: str,
         tool_params: dict[str, Any],
-        tool_call_id: Optional[str] = None,
-        **kwargs: Optional[dict[str, Any]],
+        tool_call_id: str | None = None,
+        **kwargs: dict[str, Any] | None,
     ) -> ToolExecutionDecision:
         """
         Async version of run. Run the confirmation strategy for a given tool and its parameters.

haystack_experimental/components/agents/human_in_the_loop/user_interfaces.py

@@ -4,7 +4,7 @@
 
 import json
 from threading import Lock
-from typing import Any, Optional
+from typing import Any
 
 from haystack.core.serialization import default_to_dict
 from rich.console import Console
@@ -20,7 +20,7 @@ _ui_lock = Lock()
 class RichConsoleUI(ConfirmationUI):
     """Rich console interface for user interaction."""
 
-    def __init__(self, console: Optional[Console] = None):
+    def __init__(self, console: Console | None = None):
         self.console = console or Console()
 
     def get_user_confirmation(

haystack_experimental/components/generators/chat/openai.py

@@ -3,7 +3,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 from dataclasses import replace
-from typing import Any, Optional
+from typing import Any
 
 from haystack import component
 from haystack.components.generators.chat.openai import OpenAIChatGenerator as BaseOpenAIChatGenerator
@@ -56,12 +56,12 @@ class OpenAIChatGenerator(BaseOpenAIChatGenerator):
     def run(
         self,
         messages: list[ChatMessage],
-        streaming_callback: Optional[StreamingCallbackT] = None,
-        generation_kwargs: Optional[dict[str, Any]] = None,
+        streaming_callback: StreamingCallbackT | None = None,
+        generation_kwargs: dict[str, Any] | None = None,
         *,
-        tools: Optional[ToolsType] = None,
-        tools_strict: Optional[bool] = None,
-        hallucination_score_config: Optional[HallucinationScoreConfig] = None,
+        tools: ToolsType | None = None,
+        tools_strict: bool | None = None,
+        hallucination_score_config: HallucinationScoreConfig | None = None,
     ) -> dict[str, list[ChatMessage]]:
         """
         Invokes chat completion based on the provided messages and generation parameters.
@@ -123,12 +123,12 @@ class OpenAIChatGenerator(BaseOpenAIChatGenerator):
     async def run_async(
         self,
         messages: list[ChatMessage],
-        streaming_callback: Optional[StreamingCallbackT] = None,
-        generation_kwargs: Optional[dict[str, Any]] = None,
+        streaming_callback: StreamingCallbackT | None = None,
+        generation_kwargs: dict[str, Any] | None = None,
         *,
-        tools: Optional[ToolsType] = None,
-        tools_strict: Optional[bool] = None,
-        hallucination_score_config: Optional[HallucinationScoreConfig] = None,
+        tools: ToolsType | None = None,
+        tools_strict: bool | None = None,
+        hallucination_score_config: HallucinationScoreConfig | None = None,
     ) -> dict[str, list[ChatMessage]]:
         """
         Asynchronously invokes chat completion based on the provided messages and generation parameters.

haystack_experimental/components/preprocessors/__init__.py

@@ -8,13 +8,11 @@ from typing import TYPE_CHECKING
 from lazy_imports import LazyImporter
 
 _import_structure = {
-    "embedding_based_document_splitter": ["EmbeddingBasedDocumentSplitter"],
     "md_header_level_inferrer": ["MarkdownHeaderLevelInferrer"],
 }
 
 if TYPE_CHECKING:
-    from .embedding_based_document_splitter import EmbeddingBasedDocumentSplitter
-    from .md_header_level_inferrer import MarkdownHeaderLevelInferrer
+    from .md_header_level_inferrer import MarkdownHeaderLevelInferrer as MarkdownHeaderLevelInferrer
 
 else:
     sys.modules[__name__] = LazyImporter(name=__name__, module_file=__file__, import_structure=_import_structure)

haystack_experimental/components/retrievers/chat_message_retriever.py

@@ -2,7 +2,7 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-from typing import Any, Optional
+from typing import Any
 
 from haystack import DeserializationError, component, default_from_dict, default_to_dict, logging
 from haystack.core.serialization import import_class_by_name
@@ -39,7 +39,7 @@ class ChatMessageRetriever:
     ```
     """
 
-    def __init__(self, chat_message_store: ChatMessageStore, last_k: Optional[int] = 10):
+    def __init__(self, chat_message_store: ChatMessageStore, last_k: int | None = 10):
         """
         Create the ChatMessageRetriever component.
 
@@ -94,8 +94,8 @@ class ChatMessageRetriever:
         self,
         chat_history_id: str,
         *,
-        last_k: Optional[int] = None,
-        current_messages: Optional[list[ChatMessage]] = None,
+        last_k: int | None = None,
+        current_messages: list[ChatMessage] | None = None,
     ) -> dict[str, list[ChatMessage]]:
         """
         Run the ChatMessageRetriever

haystack_experimental/components/retrievers/types/protocol.py

@@ -2,7 +2,7 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-from typing import Any, Optional, Protocol
+from typing import Any, Protocol
 
 # Ellipsis are needed to define the Protocol but pylint complains. See https://github.com/pylint-dev/pylint/issues/9319.
 # pylint: disable=unnecessary-ellipsis
@@ -16,7 +16,7 @@ class TextRetriever(Protocol):
     store or other data source. They return a dictionary with a list of Document objects.
     """
 
-    def run(self, query: str, filters: Optional[dict[str, Any]] = None, top_k: Optional[int] = None) -> dict[str, Any]:
+    def run(self, query: str, filters: dict[str, Any] | None = None, top_k: int | None = None) -> dict[str, Any]:
         """
         Retrieve documents that are relevant to the query.
 
@@ -42,7 +42,7 @@ class EmbeddingRetriever(Protocol):
     """
 
     def run(
-        self, query_embedding: list[float], filters: Optional[dict[str, Any]] = None, top_k: Optional[int] = None
+        self, query_embedding: list[float], filters: dict[str, Any] | None = None, top_k: int | None = None
     ) -> dict[str, Any]:
         """
         Retrieve documents that are relevant to the query.

haystack_experimental/components/summarizers/llm_summarizer.py

@@ -2,7 +2,7 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-from typing import Any, Optional
+from typing import Any
 
 from haystack import Document, component, default_from_dict, default_to_dict, logging
 from haystack.components.generators.chat.types import ChatGenerator
@@ -49,9 +49,9 @@ class LLMSummarizer:
     def __init__(  # pylint: disable=too-many-positional-arguments
         self,
         chat_generator: ChatGenerator,
-        system_prompt: Optional[str] = "Rewrite this text in summarized form.",
+        system_prompt: str | None = "Rewrite this text in summarized form.",
         summary_detail: float = 0,
-        minimum_chunk_size: Optional[int] = 500,
+        minimum_chunk_size: int | None = 500,
         chunk_delimiter: str = ".",
         summarize_recursively: bool = False,
         split_overlap: int = 0,
@@ -274,10 +274,10 @@ class LLMSummarizer:
         self,
         *,
         documents: list[Document],
-        detail: Optional[float] = None,
-        minimum_chunk_size: Optional[int] = None,
-        summarize_recursively: Optional[bool] = None,
-        system_prompt: Optional[str] = None,
+        detail: float | None = None,
+        minimum_chunk_size: int | None = None,
+        summarize_recursively: bool | None = None,
+        system_prompt: str | None = None,
     ) -> dict[str, list[Document]]:
         """
         Run the summarizer on a list of documents.

haystack_experimental/core/pipeline/breakpoint.py

@@ -4,7 +4,7 @@
 
 from dataclasses import replace
 from datetime import datetime
-from typing import TYPE_CHECKING, Any, Optional
+from typing import TYPE_CHECKING, Any
 
 from haystack import logging
 from haystack.core.pipeline.utils import _deepcopy_with_exceptions
@@ -27,7 +27,7 @@ def _create_agent_snapshot(
     component_visits: dict[str, int],
     agent_breakpoint: AgentBreakpoint,
     component_inputs: dict[str, Any],
-    tool_execution_decisions: Optional[list["ToolExecutionDecision"]] = None,
+    tool_execution_decisions: list["ToolExecutionDecision"] | None = None,
 ) -> AgentSnapshot:
     """
     Create a snapshot of the agent's state.
@@ -59,10 +59,10 @@ def _create_agent_snapshot(
 def _create_pipeline_snapshot_from_tool_invoker(
     *,
     execution_context: "_ExecutionContext",
-    tool_name: Optional[str] = None,
-    agent_name: Optional[str] = None,
-    break_point: Optional[AgentBreakpoint] = None,
-    parent_snapshot: Optional[PipelineSnapshot] = None,
+    tool_name: str | None = None,
+    agent_name: str | None = None,
+    break_point: AgentBreakpoint | None = None,
+    parent_snapshot: PipelineSnapshot | None = None,
 ) -> PipelineSnapshot:
     """
     Create a pipeline snapshot when a tool invoker breakpoint is raised or an exception during execution occurs.

haystack_experimental/dataclasses/breakpoints.py

@@ -4,7 +4,7 @@
 
 from dataclasses import dataclass
 from datetime import datetime
-from typing import Any, Optional
+from typing import Any
 
 from haystack.dataclasses.breakpoints import AgentBreakpoint
 from haystack.dataclasses.breakpoints import AgentSnapshot as HaystackAgentSnapshot
@@ -14,7 +14,7 @@ from haystack_experimental.components.agents.human_in_the_loop.dataclasses impor
 
 @dataclass
 class AgentSnapshot(HaystackAgentSnapshot):
-    tool_execution_decisions: Optional[list[ToolExecutionDecision]] = None
+    tool_execution_decisions: list[ToolExecutionDecision] | None = None
 
     def to_dict(self) -> dict[str, Any]:
         """

haystack_experimental/memory_stores/__init__.py

@@ -0,0 +1,7 @@
+# SPDX-FileCopyrightText: 2022-present deepset GmbH <info@deepset.ai>
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from .types import MemoryStore
+
+__all__ = ["MemoryStore"]

haystack_experimental/memory_stores/mem0/__init__.py

@@ -0,0 +1,16 @@
+# SPDX-FileCopyrightText: 2022-present deepset GmbH <info@deepset.ai>
+#
+# SPDX-License-Identifier: Apache-2.0
+
+import sys
+from typing import TYPE_CHECKING
+
+from lazy_imports import LazyImporter
+
+_import_structure = {"memory_store": ["Mem0MemoryStore"]}
+
+if TYPE_CHECKING:
+    from .memory_store import Mem0MemoryStore as Mem0MemoryStore
+
+else:
+    sys.modules[__name__] = LazyImporter(name=__name__, module_file=__file__, import_structure=_import_structure)