agentx-python 0.6.25__tar.gz → 0.6.27__tar.gz

{agentx_python-0.6.25/agentx_python.egg-info → agentx_python-0.6.27}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentx-python
-Version: 0.6.25
+Version: 0.6.27
 Summary: Official Python SDK for AgentX (https://www.agentx.so/)
 Home-page: https://github.com/AgentX-ai/AgentX-python
 Author: Robin Wang and AgentX Team

{agentx_python-0.6.25 → agentx_python-0.6.27}/agentx/integrations/langchain.py

@@ -489,6 +489,7 @@ class AgentXCallbackHandler(BaseCallbackHandler):
                 duration_ms=step.get("duration_ms"),
                 input=step.get("query"),
                 output=step.get("output"),
+                metadata={"kind": "retrieval"},
             )
 
     def on_chain_end(

{agentx_python-0.6.25 → agentx_python-0.6.27}/agentx/tracing/tracer.py

@@ -252,6 +252,7 @@ class _TraceSpan:
         cache_write_tokens: Optional[int] = None,
         error: Optional[str] = None,
         tool_calls: Optional[List[Dict[str, Any]]] = None,
+        metadata: Optional[Dict[str, Any]] = None,
     ) -> "_TraceSpan":
         """
         Send one real child-span row parented to this span, with explicit timing (the caller's
@@ -306,6 +307,8 @@ class _TraceSpan:
             wire["model"] = model
         if child.tool_calls:
             wire["tool_calls"] = child.tool_calls
+        if metadata:
+            wire["metadata"] = _safe_serialize(metadata)
         if child._session_id:
             wire["session_id"] = child._session_id
         wire["span_id"] = child._span_id
@@ -411,6 +414,7 @@ class _TraceSpan:
                     duration_ms=step.get("duration_ms"),
                     input=step.get("query"),
                     output=step.get("output"),
+                    metadata={"kind": "retrieval"},
                 )
 
             if self.input is None and input is not None:
@@ -558,6 +562,7 @@ class Tracer:
     def __init__(self, ingest_client: IngestClient) -> None:
         self._client = ingest_client
         self._pending_tool_calls: List[Dict[str, Any]] = []
+        self._pending_retrievals: List[Dict[str, Any]] = []
         self._local = threading.local()
 
     # ------------------------------------------------------------------
@@ -733,15 +738,40 @@ class Tracer:
         Manually record a knowledge-base / vector-store retrieval that an
         auto-instrumented framework integration can't see on its own - e.g. a
         hand-rolled RAG lookup wrapped around a raw Anthropic/OpenAI call or a
-        CrewAI kickoff. Sent as a real child-span row of the active span (see ``current_span``) -
-        requires an enclosing ``with tracer.trace(...)`` block; a no-op (nothing to attach a child
-        to) if called outside one.
+        CrewAI kickoff. Sent as a real child-span row of the active span (see ``current_span``);
+        with no active span it queues and merges into the very next trace this tracer sends
+        (same behavior as ``record_tool_call``, covering the patched-client flow where the
+        retrieval runs just before a standalone ``messages.create()`` /
+        ``chat.completions.create()`` call).
         """
         active_span = self.current_span
         if active_span is None:
+            latency_ms = (
+                int(duration_ms)
+                if duration_ms is not None
+                else int((end_time - start_time) * 1000)
+                if start_time is not None and end_time is not None
+                else None
+            )
+            self._pending_retrievals.append({
+                "name": name,
+                "query": _safe_serialize(query) if query is not None else None,
+                "output": _safe_serialize(output) if output is not None else None,
+                "duration_ms": latency_ms,
+            })
             return
+        # The kind marker is what tells the engine (retrieval-context extraction for RAG
+        # judges) and the dashboard timeline that this span is a retrieval regardless of its
+        # name - the name-based "Retrieval N" heuristic remains only as a fallback for older
+        # traces, so custom names like "kb_search" work everywhere.
         active_span.child_span(
-            name, start_time=start_time, end_time=end_time, duration_ms=duration_ms, input=query, output=output
+            name,
+            start_time=start_time,
+            end_time=end_time,
+            duration_ms=duration_ms,
+            input=query,
+            output=output,
+            metadata={"kind": "retrieval"},
         )
 
     @contextmanager
@@ -1091,6 +1121,17 @@ class Tracer:
             # a projection that predates them would silently strip exactly the failure evidence.
             wire["tool_calls"] = list(wire.get("tool_calls") or []) + [dict(t) for t in pending_tool_calls]
 
+        # record_retrieval entries queued with no active span ride the root's
+        # performance_summary.retrieval_steps - the same shape older flat traces used, which the
+        # engine's retrieval-context extraction and the dashboard's references panel both read.
+        pending_retrievals, self._pending_retrievals = self._pending_retrievals, []
+        if pending_retrievals:
+            summary = dict(wire.get("performance_summary") or {})
+            summary["retrieval_steps"] = list(summary.get("retrieval_steps") or []) + [
+                dict(r) for r in pending_retrievals
+            ]
+            wire["performance_summary"] = summary
+
         return self._dispatch(wire, sync=sync)
 
     def _dispatch(self, wire: Dict[str, Any], *, sync: bool = False) -> Optional[str]:

agentx_python-0.6.27/agentx/version.py

@@ -0,0 +1 @@
+VERSION = "0.6.27"

{agentx_python-0.6.25 → agentx_python-0.6.27/agentx_python.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentx-python
-Version: 0.6.25
+Version: 0.6.27
 Summary: Official Python SDK for AgentX (https://www.agentx.so/)
 Home-page: https://github.com/AgentX-ai/AgentX-python
 Author: Robin Wang and AgentX Team

agentx_python-0.6.25/agentx/version.py

@@ -1 +0,0 @@
-VERSION = "0.6.25"