unique-toolkit 2026.20.0.dev30__tar.gz → 2026.20.0.dev31__tar.gz

{unique_toolkit-2026.20.0.dev30 → unique_toolkit-2026.20.0.dev31}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.20.0.dev30
+Version: 2026.20.0.dev31
 Summary: 
 Author: Cedric Klinkert, Martin Fadler, Sadique Sheik, Fabian Schläpfer, Pascal Hauri, Ejiroghene Oki-Peter, Rami Azouz, Ricardo Visini, Sanzio Monti
 Author-email: Cedric Klinkert <cedric.klinkert@unique.ch>, Martin Fadler <martin.fadler@unique.ch>, Sadique Sheik <sadique@unique.ch>, Fabian Schläpfer <fabian@unique.ch>, Pascal Hauri <pascal@unique.ch>, Ejiroghene Oki-Peter <oki-peter.ejiroghene@unique.ai>, Rami Azouz <rami.ext@unique.ch>, Ricardo Visini <ricardo.visini@unique.ch>, Sanzio Monti <sanzio.monti@unique.ai>

{unique_toolkit-2026.20.0.dev30 → unique_toolkit-2026.20.0.dev31}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "2026.20.0.dev30"
+version = "2026.20.0.dev31"
 description = ""
 readme = "README.md"
 requires-python = ">=3.12"

{unique_toolkit-2026.20.0.dev30 → unique_toolkit-2026.20.0.dev31}/unique_toolkit/agentic/tools/a2a/postprocessing/references.py

@@ -62,7 +62,7 @@ def _add_sub_agent_references_in_place(
     responses: list[SubAgentResponse],
 ) -> None:
     text = loop_response.message.text
-    refs = []
+    refs = list(loop_response.message.references or [])
 
     for response in responses:
         sub_agent_refs = []

unique_toolkit-2026.20.0.dev31/unique_toolkit/agentic/tools/a2a/postprocessing/test/test_references.py

@@ -0,0 +1,93 @@
+"""
+Unit tests for SubAgentReferencesPostprocessor in references.py.
+
+Focus: regression coverage for the bug where existing references on the
+loop response (e.g. those added by DisplayCodeInterpreterFilesPostProcessor)
+were silently wiped out when SubAgentReferencesPostprocessor ran.
+"""
+
+import pytest
+
+from unique_toolkit.agentic.tools.a2a.postprocessing.references import (
+    _add_sub_agent_references_in_place,
+)
+from unique_toolkit.chat.schemas import ChatMessage, ChatMessageRole
+from unique_toolkit.content import ContentReference
+from unique_toolkit.language_model.schemas import LanguageModelStreamResponse
+
+
+@pytest.fixture
+def existing_doc_reference() -> ContentReference:
+    """A pre-existing ContentReference such as one added by a prior postprocessor."""
+    return ContentReference(
+        id="cont_existing_doc",
+        message_id="msg_1",
+        name="generated_chart.png",
+        sequence_number=1,
+        source_id="cont_existing_doc",
+        source="node-ingestion-chunks",
+        url="unique://content/cont_existing_doc",
+    )
+
+
+def _make_loop_response(
+    text: str,
+    references: list[ContentReference] | None,
+) -> LanguageModelStreamResponse:
+    return LanguageModelStreamResponse(
+        message=ChatMessage(
+            id="msg_1",
+            chat_id="chat_1",
+            role=ChatMessageRole.ASSISTANT,
+            text=text,
+            original_text=text,
+            references=references,
+        ),
+    )
+
+
+@pytest.mark.ai
+def test_add_sub_agent_references_in_place__preserves_existing_refs__when_no_sub_agent_responses(
+    existing_doc_reference: ContentReference,
+) -> None:
+    """
+    Purpose: Verify pre-existing references on the loop response are not wiped
+    when there are no sub-agent responses to merge in.
+    Why this matters: A previous postprocessor (e.g. DisplayCodeInterpreterFilesPostProcessor)
+    may have already populated message.references; this postprocessor must not overwrite
+    them with an empty list, otherwise the user sees `<sup>1</sup>` in text with no chip.
+    Setup summary: Loop response carrying one existing ref, empty sub-agent responses,
+    assert the existing ref survives unchanged.
+    """
+    # Arrange
+    loop_response = _make_loop_response(
+        text="See <sup>1</sup> for the chart.",
+        references=[existing_doc_reference],
+    )
+
+    # Act
+    _add_sub_agent_references_in_place(loop_response=loop_response, responses=[])
+
+    # Assert
+    assert loop_response.message.references == [existing_doc_reference]
+
+
+@pytest.mark.ai
+def test_add_sub_agent_references_in_place__handles_none_references__without_error() -> (
+    None
+):
+    """
+    Purpose: Verify the function tolerates message.references being None.
+    Why this matters: ChatMessage.references is Optional; the seeding step must coerce
+    None to an empty list rather than crashing.
+    Setup summary: Loop response with references=None and no sub-agent responses,
+    assert references becomes an empty list.
+    """
+    # Arrange
+    loop_response = _make_loop_response(text="No refs here.", references=None)
+
+    # Act
+    _add_sub_agent_references_in_place(loop_response=loop_response, responses=[])
+
+    # Assert
+    assert loop_response.message.references == []