agentx-python 0.8.18__py3-none-any.whl → 0.8.19__py3-none-any.whl

agentx/tracing/tracer.py

@@ -581,6 +581,14 @@ class _RetrievalRecorder:
         self.output: Any = None
 
 
+class _MemoryOpRecorder:
+    """Handle yielded by ``Tracer.trace_memory()`` - set ``output`` (what was recalled or
+    stored) inside the block."""
+
+    def __init__(self) -> None:
+        self.output: Any = None
+
+
 class _ToolCallRecorder:
     """Handle yielded by ``Tracer.trace_tool_call()`` - set ``output`` inside the block.
     ``success``/``error`` may be set manually; an exception escaping the block sets them
@@ -832,6 +840,66 @@ class Tracer:
             span_kind="retrieval",
         )
 
+    def record_memory(
+        self,
+        name: str = "Memory",
+        *,
+        operation: Optional[str] = None,
+        query: Optional[str] = None,
+        output: Any = None,
+        duration_ms: Optional[float] = None,
+        start_time: Optional[float] = None,
+        end_time: Optional[float] = None,
+    ) -> None:
+        """
+        Manually record a long-term-memory operation (a Mem0/Zep/Letta-style recall or store)
+        as a ``span_kind="memory"`` child span of the active span. ``operation`` is free text -
+        conventionally ``"read"`` or ``"write"`` - carried in the span's metadata; the kind
+        itself stays one value so dashboards and scorers can select all memory activity at once.
+        Deliberately NOT a retrieval: retrieval spans feed the RAG judges' ``{context}``
+        (knowledge grounding), while memory is recalled state - see the engine's spanKind.ts.
+        """
+        active_span = self.current_span
+        if active_span is None:
+            return
+        active_span.child_span(
+            name,
+            start_time=start_time,
+            end_time=end_time,
+            duration_ms=duration_ms,
+            input=query,
+            output=output,
+            metadata={"kind": "memory", **({"operation": operation} if operation else {})},
+            span_kind="memory",
+        )
+
+    @contextmanager
+    def trace_memory(
+        self, name: str = "Memory", *, operation: Optional[str] = None, query: Optional[str] = None
+    ) -> Iterator["_MemoryOpRecorder"]:
+        """
+        Context manager that times a memory operation and records it via
+        :meth:`record_memory` on exit::
+
+            with tracer.trace_memory("user prefs", operation="read", query=user_id) as m:
+                m.output = memory.search(user_id, question)
+        """
+        start_t = time.time()
+        recorder = _MemoryOpRecorder()
+        try:
+            yield recorder
+        finally:
+            end_t = time.time()
+            self.record_memory(
+                name,
+                operation=operation,
+                query=query,
+                output=recorder.output,
+                duration_ms=(end_t - start_t) * 1000,
+                start_time=start_t,
+                end_time=end_t,
+            )
+
     @contextmanager
     def trace_retrieval(self, name: str = "Retrieval", *, query: Optional[str] = None) -> Iterator["_RetrievalRecorder"]:
         """

agentx/version.py

@@ -1,7 +1,7 @@
-VERSION = "0.8.18"
+VERSION = "0.8.19"
 
 # The AgentX-trace-eval release this SDK version is tested against - what `agentx-trace-eval`
 # installs and converges to (see agentx/cli.py). Bump together with VERSION when releasing, so
 # every published SDK names a known-good engine+dashboard pair. Users can override with
 # AGENTX_TRACE_EVAL_VERSION=<tag|latest>.
-ENGINE_VERSION = "v0.3.18"
+ENGINE_VERSION = "v0.3.19"

{agentx_python-0.8.18.dist-info → agentx_python-0.8.19.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentx-python
-Version: 0.8.18
+Version: 0.8.19
 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.8.18.dist-info → agentx_python-0.8.19.dist-info}/RECORD

@@ -10,7 +10,7 @@ agentx/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agentx/testing.py,sha256=0shZEid_vhJgBegpO_loUq6APYDxQrUgD3jvcRdIDV8,7372
 agentx/traces.py,sha256=sz9gxutlDKNgf0fdCkyzKsGwK5eMAJuIEl3NIwGrsuo,2287
 agentx/util.py,sha256=lt2Kpg4Fj8whrSuAhLEahcPEIYRVpK9RCC5y9ShdTJs,703
-agentx/version.py,sha256=12ZEc4OJ-zU1rtsHoDBC5wqnwX7BmZ9OmiL3F4NWnMQ,366
+agentx/version.py,sha256=uZFelTcDOUCZeQsTxaCj6X0keAwmJ6whvtr9_8pafg0,366
 agentx/evaluations/__init__.py,sha256=Erv7RGFlRxGTG4rVb2uHhCqLLEX_iAWKqkZKzK6CumE,262
 agentx/evaluations/_term.py,sha256=WFpiNzdgDBeJJ-Gg-6X7TwwxllobuE4OqFUTuDQvS3Y,2529
 agentx/evaluations/client.py,sha256=f5GqsiuqknE-ENUYocBcXgV6OGMKdzmlsm76takJbwU,33112
@@ -65,10 +65,10 @@ agentx/tracing/ci_types.py,sha256=zFVcZGvc1qbVDdOlEPQ1i5R6terqJHxCzFF4ZjouxxI,14
 agentx/tracing/eval_scope.py,sha256=ElMbPxpqpQBVuaUnHNlw9RIQu71oyOpd0yR55bDH8eI,2144
 agentx/tracing/framework_detect.py,sha256=uV4O7Th-4_2UkdooAyaWCOIA0jwLJbblFQ_ucFDjeJM,2638
 agentx/tracing/ingest_client.py,sha256=spnm6bV3NXGuj2w7HwdKiJdKDJWRUFmSP3jl3rcGqkQ,18202
-agentx/tracing/tracer.py,sha256=TxY6zG6hbM81jpfrqP4imaEvoyy_hnJVUSTnoAVHMG0,56240
-agentx_python-0.8.18.dist-info/licenses/LICENSE,sha256=gZVsM-nLsE8vlaY6NXXsVoo6IlCClxkToAWmhgT3y_s,10762
-agentx_python-0.8.18.dist-info/METADATA,sha256=P0yCj9IwWgJrN6c_YcDBWj5YZeJCFaBkz_CNv1mJ2p4,22117
-agentx_python-0.8.18.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
-agentx_python-0.8.18.dist-info/entry_points.txt,sha256=rQqF1JTY3T1yfviBU1r8PU-5mBdfOk6P4bZ1L4BskIM,172
-agentx_python-0.8.18.dist-info/top_level.txt,sha256=s-q-HB9Gb_QdrZNacSeQyF_c25gQooMy7DlxzgLOHPk,7
-agentx_python-0.8.18.dist-info/RECORD,,
+agentx/tracing/tracer.py,sha256=VQbHT1qhkdOZZy5_SEBZaivl2qKjGPruAj8SqBbMcNY,58727
+agentx_python-0.8.19.dist-info/licenses/LICENSE,sha256=gZVsM-nLsE8vlaY6NXXsVoo6IlCClxkToAWmhgT3y_s,10762
+agentx_python-0.8.19.dist-info/METADATA,sha256=U2phHPdoDEXuwj0uBze7zHyq5HPr1aOE7sUL8hOg7_Y,22117
+agentx_python-0.8.19.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+agentx_python-0.8.19.dist-info/entry_points.txt,sha256=rQqF1JTY3T1yfviBU1r8PU-5mBdfOk6P4bZ1L4BskIM,172
+agentx_python-0.8.19.dist-info/top_level.txt,sha256=s-q-HB9Gb_QdrZNacSeQyF_c25gQooMy7DlxzgLOHPk,7
+agentx_python-0.8.19.dist-info/RECORD,,