Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 5 additions & 3 deletions haystack_experimental/core/pipeline/breakpoint.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,12 +2,12 @@
#
# SPDX-License-Identifier: Apache-2.0

from copy import deepcopy
from dataclasses import replace
from datetime import datetime
from typing import TYPE_CHECKING, Any, Optional

from haystack import logging
from haystack.core.pipeline.utils import _deepcopy_with_exceptions
from haystack.dataclasses.breakpoints import AgentBreakpoint, PipelineSnapshot, PipelineState, ToolBreakpoint
from haystack.utils.base_serialization import _serialize_value_with_schema
from haystack.utils.misc import _get_output_dir
Expand Down Expand Up @@ -44,8 +44,10 @@ def _create_agent_snapshot(
"""
return AgentSnapshot(
component_inputs={
"chat_generator": _serialize_value_with_schema(deepcopy(component_inputs["chat_generator"])),
"tool_invoker": _serialize_value_with_schema(deepcopy(component_inputs["tool_invoker"])),
"chat_generator": _serialize_value_with_schema(
_deepcopy_with_exceptions(component_inputs["chat_generator"])
),
"tool_invoker": _serialize_value_with_schema(_deepcopy_with_exceptions(component_inputs["tool_invoker"])),
},
component_visits=component_visits,
break_point=agent_breakpoint,
Expand Down
Loading