by-framework-langgraph 0.0.2__tar.gz → 0.0.3.dev0__tar.gz

{by_framework_langgraph-0.0.2 → by_framework_langgraph-0.0.3.dev0}/.gitignore

@@ -3,6 +3,7 @@ __pycache__/
 *.py[cod]
 *$py.class
 *.so
+.ruff_cache/
 .Python
 build/
 develop-eggs/
@@ -66,3 +67,7 @@ Thumbs.db
 *gateway-sdk.log*
 .claude/settings.local.json
 .claude/settings.json
+
+core
+
+.coderfleet-uploads/

{by_framework_langgraph-0.0.2 → by_framework_langgraph-0.0.3.dev0}/PKG-INFO

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

{by_framework_langgraph-0.0.2 → by_framework_langgraph-0.0.3.dev0}/pyproject.toml

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

{by_framework_langgraph-0.0.2 → by_framework_langgraph-0.0.3.dev0}/src/by_framework_langgraph/_utils.py

@@ -2,7 +2,6 @@
 
 from __future__ import annotations
 
-import hashlib
 from typing import Any
 
 from by_framework.core.protocol.commands import ResumeCommand
@@ -52,13 +51,3 @@ def extract_resume_data(command: ResumeCommand) -> str:
         return extract_content_text(command.content)
 
     return ""
-
-
-def str_to_uint128(s: str) -> int:
-    """Convert a string to a 128-bit integer (for OTEL TraceId)."""
-    return int(hashlib.md5(s.encode()).hexdigest(), 16)
-
-
-def str_to_uint64(s: str) -> int:
-    """Convert a string to a 64-bit integer (for OTEL SpanId)."""
-    return int(hashlib.md5(s.encode()).hexdigest()[:16], 16)

{by_framework_langgraph-0.0.2 → by_framework_langgraph-0.0.3.dev0}/src/by_framework_langgraph/adapter.py

@@ -16,15 +16,11 @@ from by_framework.common.logger import logger
 from by_framework.core.protocol.agent_state import AgentState
 from by_framework.core.protocol.commands import ResumeCommand
 from by_framework.core.protocol.events import StreamChunkEvent
+from by_framework.observability.span_recorder import (str_to_uint64, str_to_uint128)
 from langchain_core.messages import HumanMessage
 from langgraph.types import Command
 
-from ._utils import (
-    extract_content_text,
-    extract_resume_data,
-    str_to_uint64,
-    str_to_uint128,
-)
+from ._utils import extract_content_text, extract_resume_data
 
 if TYPE_CHECKING:
     from by_framework.core.protocol.commands import GatewayCommand
@@ -383,6 +379,26 @@ class LangGraphAdapter:
     @contextmanager
     def _langfuse_callback_manager(self, callbacks: list[Any]) -> Iterator[None]:
         """Prepare Langfuse callback and observation for LangChain."""
+        # Prefer AgentContext's callback factory so trace and parent ids align.
+        # Filter out auto-generated MagicMock attributes when tests use a mock
+        # context — real callback objects always come from a non-test module.
+        langfuse_callback_value = getattr(self._context, "langfuse_callback", None)
+        is_real_callback = langfuse_callback_value is not None and type(
+            langfuse_callback_value
+        ).__module__ not in ("unittest.mock",)
+
+        if is_real_callback:
+            handler = (
+                langfuse_callback_value()
+                if callable(langfuse_callback_value)
+                else langfuse_callback_value
+            )
+            if handler is not None:
+                callbacks.append(handler)
+                yield
+                return
+
+        # Fallback to local import if context method is missing
         # pylint: disable=import-outside-toplevel
         try:
             langfuse_config = import_module(
@@ -414,6 +430,17 @@ class LangGraphAdapter:
             },
             metadata=self._default_metadata(),
         ):
+            # Prevent the generated OTel span from being promoted to a trace root.
+            # The native LangfusePlugin sets the same attribute on its own path
+            # (via _SdkLangfuseTracer); this covers the LangGraph fallback path.
+            try:
+                from opentelemetry import trace
+
+                current_span = trace.get_current_span()
+                if current_span and hasattr(current_span, "set_attribute"):
+                    current_span.set_attribute("langfuse.internal.as_root", False)
+            except Exception:  # pylint: disable=broad-exception-caught
+                pass
             yield
 
     @staticmethod

{by_framework_langgraph-0.0.2 → by_framework_langgraph-0.0.3.dev0}/tests/test_adapter.py

@@ -259,6 +259,50 @@ class TestAdapterRun:
         }
         assert observation_calls[0]["name"] == "planner:langgraph"
 
+    @pytest.mark.asyncio
+    async def test_uses_context_langfuse_callback_property(self):
+        """Verify AgentContext.langfuse_callback property value is used directly as a handler."""
+        handler = object()
+
+        # pylint: disable=too-few-public-methods,missing-class-docstring,missing-function-docstring
+        class ContextWithCallbackProperty:
+            session_id = "test-session"
+            trace_id = "trace-ctx"
+            message_id = "msg-ctx"
+            parent_message_id = ""
+            current_agent_id = "planner"
+            current_command = SimpleNamespace(
+                header=MessageHeader(
+                    message_id="msg-ctx",
+                    session_id="test-session",
+                    trace_id="trace-ctx",
+                    target_agent_type="planner",
+                )
+            )
+
+            def __init__(self):
+                self.emit_chunk = AsyncMock()
+
+            @property
+            def langfuse_callback(self):
+                return handler
+
+        ctx = ContextWithCallbackProperty()
+        graph = MagicMock()
+        graph.ainvoke = AsyncMock(
+            return_value={"messages": [MagicMock(content="hello")]}
+        )
+        snapshot = MagicMock()
+        snapshot.next = ()
+        graph.get_state.return_value = snapshot
+
+        adapter = LangGraphAdapter(graph, ctx, stream=False)
+        result = await adapter.run(AskAgentCommand(header=_make_header(), content="hi"))
+
+        assert result == "hello"
+        _, kwargs = graph.ainvoke.call_args
+        assert kwargs["config"]["callbacks"] == [handler]
+
     @pytest.mark.asyncio
     async def test_skips_langfuse_tracing_silently_when_not_configured(
         self, monkeypatch, caplog