by-framework-langgraph 0.0.3.dev2__tar.gz → 0.0.3.dev3__tar.gz

{by_framework_langgraph-0.0.3.dev2 → by_framework_langgraph-0.0.3.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: by-framework-langgraph
-Version: 0.0.3.dev2
+Version: 0.0.3.dev3
 Summary: LangGraph integration for by-framework
 Requires-Python: >=3.12
 Requires-Dist: by-framework>=0.2.1

{by_framework_langgraph-0.0.3.dev2 → by_framework_langgraph-0.0.3.dev3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "by-framework-langgraph"
-version = "0.0.3.dev2"
+version = "0.0.3.dev3"
 description = "LangGraph integration for by-framework"
 readme = "README.md"
 requires-python = ">=3.12"

{by_framework_langgraph-0.0.3.dev2 → by_framework_langgraph-0.0.3.dev3}/src/by_framework_langgraph/adapter.py

@@ -525,6 +525,7 @@ class LangGraphAdapter:
                 self._context, "parent_message_id", ""
             ),
             "by_framework_agent_id": getattr(self._context, "current_agent_id", ""),
+            "worker_id": getattr(self._context, "worker_id", ""),
             "langgraph_thread_id": self._thread_id,
         }
         return {
@@ -538,10 +539,31 @@ class LangGraphAdapter:
 
         with (
             self._phoenix_context_manager(),
+            self._langfuse_attribute_propagation_context_manager(),
             self._langfuse_callback_manager(callbacks),
         ):
             yield callbacks
 
+    @contextmanager
+    def _langfuse_attribute_propagation_context_manager(self) -> Iterator[None]:
+        """Propagate stable framework metadata to Langfuse child observations."""
+        worker_id = str(getattr(self._context, "worker_id", "") or "")
+        if not worker_id:
+            yield
+            return
+
+        try:
+            propagate_attributes = getattr(
+                import_module("langfuse"),
+                "propagate_attributes",
+            )
+        except (ImportError, AttributeError):
+            yield
+            return
+
+        with propagate_attributes(metadata={"worker_id": worker_id}):
+            yield
+
     @contextmanager
     def _phoenix_context_manager(self) -> Iterator[None]:
         """Prepare OpenTelemetry context for Phoenix tracing."""

{by_framework_langgraph-0.0.3.dev2 → by_framework_langgraph-0.0.3.dev3}/tests/test_adapter.py

@@ -23,6 +23,7 @@ def _make_mock_context(session_id: str = "test-session"):
     ctx.message_id = "msg-ctx"
     ctx.parent_message_id = "parent-ctx"
     ctx.current_agent_id = "planner"
+    ctx.worker_id = "worker-langgraph-1"
     ctx.redis = AsyncMock()
     ctx.emit_chunk = AsyncMock()
     ctx.ask_user = AsyncMock()
@@ -225,6 +226,7 @@ class TestAdapterRun:
         assert config["metadata"]["langfuse_session_id"] == "test-session"
         assert config["metadata"]["langfuse_user_id"] == "user-1"
         assert config["metadata"]["by_framework_message_id"] == "msg-ctx"
+        assert config["metadata"]["worker_id"] == "worker-langgraph-1"
         assert config["metadata"]["langgraph_thread_id"] == "test-session"
         # Callbacks list includes the Langfuse handler(s) plus the token accumulator.
         assert callback_handler in config["callbacks"]
@@ -239,6 +241,63 @@ class TestAdapterRun:
             }
         ]
 
+    @pytest.mark.asyncio
+    async def test_propagates_worker_id_to_langfuse_child_observations(
+        self, monkeypatch
+    ):
+        """Langfuse attribute propagation covers nested LangGraph observations."""
+        ctx = _make_mock_context()
+        graph = MagicMock()
+        snapshot = MagicMock()
+        snapshot.next = ()
+        graph.get_state.return_value = snapshot
+
+        propagation_events: list[tuple[str, dict[str, str]]] = []
+        propagation_active = False
+
+        class FakePropagation:
+
+            def __init__(self, metadata):
+                self.metadata = metadata
+
+            def __enter__(self):
+                nonlocal propagation_active
+                propagation_active = True
+                propagation_events.append(("enter", self.metadata))
+
+            def __exit__(self, exc_type, exc_val, exc_tb):
+                nonlocal propagation_active
+                propagation_events.append(("exit", self.metadata))
+                propagation_active = False
+
+        def fake_propagate_attributes(**kwargs):
+            return FakePropagation(metadata=kwargs["metadata"])
+
+        async def fake_ainvoke(input_data, *, config):
+            del input_data
+            assert config["metadata"]["worker_id"] == "worker-langgraph-1"
+            assert propagation_active is True
+            return {"messages": [MagicMock(content="hello")]}
+
+        graph.ainvoke = fake_ainvoke
+        monkeypatch.setitem(
+            sys.modules,
+            "langfuse",
+            SimpleNamespace(propagate_attributes=fake_propagate_attributes),
+        )
+
+        adapter = LangGraphAdapter(graph, ctx, stream=False)
+
+        result = await adapter.run(
+            AskAgentCommand(header=_make_header(), content="write a poem")
+        )
+
+        assert result == "hello"
+        assert propagation_events == [
+            ("enter", {"worker_id": "worker-langgraph-1"}),
+            ("exit", {"worker_id": "worker-langgraph-1"}),
+        ]
+
     @pytest.mark.asyncio
     async def test_skips_langfuse_callback_when_provider_package_missing(
         self, monkeypatch