agentx-python 0.8.25__py3-none-any.whl → 0.8.26__py3-none-any.whl

agentx/integrations/__init__.py

@@ -7,4 +7,5 @@
 #   from agentx.integrations.anthropic import patch_anthropic_client
 #   from agentx.integrations.google_adk import AgentXADKPlugin
 #   from agentx.integrations.google_genai import patch_genai_client
+#   from agentx.integrations.nvidia_nim import patch_nim_client
 #   from agentx.integrations.moveworks import MoveworksImporter  # Data API pull sync, not in-process

agentx/integrations/nvidia_nim.py

@@ -0,0 +1,70 @@
+"""
+NVIDIA NIM integration for AgentX production tracing.
+
+NIM (NVIDIA Inference Microservices) serves models behind an OpenAI-compatible
+``/v1/chat/completions`` API, so the client you patch is the ordinary ``openai``
+Python client pointed at a NIM endpoint - a local NIM container
+(``http://localhost:8000/v1``) or NVIDIA's hosted API
+(``https://integrate.api.nvidia.com/v1``). This module reuses the OpenAI patch
+machinery verbatim and differs in exactly one way: traces are stamped
+``framework="nvidia-nim"``, so NIM traffic gets its own row in Monitor's
+Platforms chart and the framework filters instead of blending into "openai".
+
+Usage::
+
+    from agentx.integrations.nvidia_nim import patch_nim_client
+    import openai
+
+    nim = openai.OpenAI(
+        base_url="http://localhost:8000/v1",  # or https://integrate.api.nvidia.com/v1
+        api_key=os.environ.get("NVIDIA_API_KEY", "not-needed-for-local-nim"),
+    )
+    patch_nim_client(nim, agentx.tracer, name="nim-agent")
+
+    # All subsequent nim.chat.completions.create() calls are now traced.
+
+Works with both ``openai.OpenAI`` and ``openai.AsyncOpenAI`` clients. Token
+usage comes straight off the response's OpenAI-shaped ``usage`` block; NIM
+reports no prompt-cache fields, so cache token counts stay unset.
+
+Streaming calls (``stream=True``) are passed through untouched and are not
+currently traced - same posture as ``patch_openai_client``, see its docstring.
+
+Requires: ``pip install "agentx-python[nvidia-nim]"`` (installs the ``openai``
+client package; there is no separate NIM SDK dependency).
+"""
+from __future__ import annotations
+
+from typing import Any, Dict, Optional
+
+from agentx.tracing.tracer import Tracer
+from agentx.integrations.openai import _patch_chat_completions_create
+
+NIM_FRAMEWORK = "nvidia-nim"
+
+
+def patch_nim_client(
+    client: Any,
+    tracer: Tracer,
+    name: str = "nim-agent",
+    metadata: Optional[Dict[str, Any]] = None,
+    session_id: Optional[str] = None,
+) -> None:
+    """
+    Monkey-patch ``client.chat.completions.create`` on an OpenAI-compatible
+    client pointed at a NIM endpoint, sending a trace for every non-streaming
+    call with ``framework="nvidia-nim"``.
+
+    The original method is still called and its return value passed through
+    unchanged. Sync and async clients both work; ``stream=True`` calls pass
+    through untraced. Patching is idempotent - and because it shares the guard
+    with ``patch_openai_client``, whichever of the two patched a given client
+    first wins (patch each client with the integration that matches where its
+    ``base_url`` actually points).
+    """
+    chat = getattr(client, "chat", None)
+    completions = getattr(chat, "completions", None) if chat is not None else None
+    if completions is None:
+        raise ValueError("Provided client does not have a .chat.completions attribute")
+
+    _patch_chat_completions_create(completions, tracer, name, metadata, session_id, framework=NIM_FRAMEWORK)

agentx/integrations/openai.py

@@ -113,7 +113,11 @@ def _patch_chat_completions_create(
     name: str,
     metadata: Optional[Dict[str, Any]],
     session_id: Optional[str],
+    framework: str = "openai",
 ) -> None:
+    # `framework` exists for OpenAI-compatible endpoints served by other vendors
+    # (agentx.integrations.nvidia_nim stamps "nvidia-nim" through here) - the request/response
+    # shapes are identical, so they share this machinery instead of duplicating it.
     original = completions_resource.create
     if getattr(original, "_agentx_patched", False):
         return  # already patched
@@ -148,7 +152,7 @@ def _patch_chat_completions_create(
             finish_llm_call(
                 tracer,
                 name=name,
-                framework="openai",
+                framework=framework,
                 metadata=metadata,
                 session_id=session_id,
                 start_t=start_t,

agentx/version.py

@@ -1,7 +1,7 @@
-VERSION = "0.8.25"
+VERSION = "0.8.26"
 
 # 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.28"
+ENGINE_VERSION = "v0.3.29"

{agentx_python-0.8.25.dist-info → agentx_python-0.8.26.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentx-python
-Version: 0.8.25
+Version: 0.8.26
 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
@@ -23,6 +23,8 @@ Provides-Extra: openai-agents
 Requires-Dist: openai-agents>=0.0.3; extra == "openai-agents"
 Provides-Extra: openai
 Requires-Dist: openai>=1.0.0; extra == "openai"
+Provides-Extra: nvidia-nim
+Requires-Dist: openai>=1.0.0; extra == "nvidia-nim"
 Provides-Extra: anthropic
 Requires-Dist: anthropic>=0.25.0; extra == "anthropic"
 Provides-Extra: google-adk
@@ -250,6 +252,7 @@ extra:
 | CrewAI                | `pip install "agentx-python[crewai]"`        | `AgentXCrewObserver`      |
 | OpenAI Agents SDK     | `pip install "agentx-python[openai-agents]"` | `AgentXTracingProcessor`  |
 | OpenAI (raw client)   | `pip install "agentx-python[openai]"`        | `patch_openai_client`     |
+| NVIDIA NIM            | `pip install "agentx-python[nvidia-nim]"`    | `patch_nim_client`        |
 | Anthropic             | `pip install "agentx-python[anthropic]"`     | `patch_anthropic_client`  |
 | Google ADK            | `pip install "agentx-python[google-adk]"`    | `AgentXADKPlugin`         |
 | Google GenAI (Gemini) | `pip install "agentx-python[google-genai]"`  | `patch_genai_client`      |

{agentx_python-0.8.25.dist-info → agentx_python-0.8.26.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=Jh07n3GLS1SGRsDoLDTUaAbuMEdvjqxCCMA9gUcSZ0M,2331
 agentx/util.py,sha256=ivCuFQ5AGUw9fR9fJy5HrSp3lLwwngsDwppnvyZpq9E,1471
-agentx/version.py,sha256=5NwobACaQAFSlFPHs5YxWk1ukEuHv1RHNtfpjXDLuHA,366
+agentx/version.py,sha256=mHuhKZSPSp8Xp5Pq7q93wgdcMtBUM_OBt-nWG4Qoh6g,366
 agentx/evaluations/__init__.py,sha256=Erv7RGFlRxGTG4rVb2uHhCqLLEX_iAWKqkZKzK6CumE,262
 agentx/evaluations/_term.py,sha256=WFpiNzdgDBeJJ-Gg-6X7TwwxllobuE4OqFUTuDQvS3Y,2529
 agentx/evaluations/client.py,sha256=7c35tS8zM3fuvFfVDQk5etTTEFiIykmPjuJxkuFyFTY,38013
@@ -27,7 +27,7 @@ agentx/evaluations/adapters/__init__.py,sha256=fK8Hx75usbiY03XUSmnLnSrwBXip2CQs2
 agentx/evaluations/adapters/http_endpoint.py,sha256=-Gika9lBcXjtbnyB-uNR6nYEVJhWEPtDazi9ixLvJtg,2059
 agentx/evaluations/adapters/precomputed.py,sha256=vxnQELmpiU2NkQATfiVjG7v1ZmmkrEA70Y11ijfYtHw,1208
 agentx/evaluations/adapters/raw.py,sha256=FkDq_mdf21-xt5EHnGyIGcS6VZxkEkFcmM9VTd-T5sA,1130
-agentx/integrations/__init__.py,sha256=W-EcQhHE01-SGPeKTB2AgfvUyRzNP4BqUMMtP2qQ4Q0,647
+agentx/integrations/__init__.py,sha256=p-YHLIudYxQj1bBj77NiVPsP0VQHemX5E59phkI3Y-Q,711
 agentx/integrations/_traced_call.py,sha256=OuFmSLw5vjI54gpdte7-zOt7gld7faQmvhuoh_3OPlo,6861
 agentx/integrations/anthropic.py,sha256=z6o9cC9rBzxxbF1UBiMr-kjEgDUO-BCuPOFoqGN3qF4,10451
 agentx/integrations/autogen.py,sha256=V1kvTjxQqOG9KsYd4dztaRYH7zTknW0_fqzKjIQG5dk,9507
@@ -39,7 +39,8 @@ agentx/integrations/langchain.py,sha256=YDeEFT2irhoEvydwt3QjbmTpk_AZZbYPS9euOR18
 agentx/integrations/litellm.py,sha256=eW8iCaNq0feRbZma8iOvmJ28WiAO7ZaRmlpPjHAtcvo,7072
 agentx/integrations/llamaindex.py,sha256=ruzMiC2TC3Xhub4xoXKPHjsvzV1sQ3QqhjkAdMxUUi4,18001
 agentx/integrations/moveworks.py,sha256=IyBswLE5LwMSp1VbIvG5izYV5BvXmv4atZelnrDn8SE,25670
-agentx/integrations/openai.py,sha256=1KLs-hJaeW2emHPwNkmC3zcGksL3zd87uWbcLGRW704,6644
+agentx/integrations/nvidia_nim.py,sha256=bvS6ebEnQZT1DVlIjQQ2RKLI4uPPTdlMKINgDwj0Z-o,2961
+agentx/integrations/openai.py,sha256=edJL-HXTO7w-vvIZGQChMu0AnPCY5z0duplIi8z2gXE,6936
 agentx/integrations/openai_agents.py,sha256=KSOQ55eRPFxxkqm7p4Dsz6HcJ_9uUEwlyywPI8dwbvY,14852
 agentx/monitor/__init__.py,sha256=Rkk-Xv1ymXwDFbx8fetmcOzXfSU1bvsazI_-x5Ati5g,1715
 agentx/monitor/_transport.py,sha256=ge0pmdvFdUtCJRkTLvAgjuTQVA8YOCYGQNoyRuselRk,2077
@@ -67,9 +68,9 @@ agentx/tracing/eval_scope.py,sha256=ElMbPxpqpQBVuaUnHNlw9RIQu71oyOpd0yR55bDH8eI,
 agentx/tracing/framework_detect.py,sha256=uV4O7Th-4_2UkdooAyaWCOIA0jwLJbblFQ_ucFDjeJM,2638
 agentx/tracing/ingest_client.py,sha256=teukTPckFWPg5RhpbGbVDBejBzPIPXUO3_DJcuVh2uc,22660
 agentx/tracing/tracer.py,sha256=NS0bs5HbDDK3l0Z95KVKV3xCNJT7LINS0EoPFM-wzIk,68543
-agentx_python-0.8.25.dist-info/licenses/LICENSE,sha256=gZVsM-nLsE8vlaY6NXXsVoo6IlCClxkToAWmhgT3y_s,10762
-agentx_python-0.8.25.dist-info/METADATA,sha256=8N9cFzwTIJc93T1hKF1XbrjrBb8Ho_kcrgnFTWlZVJU,22806
-agentx_python-0.8.25.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
-agentx_python-0.8.25.dist-info/entry_points.txt,sha256=rQqF1JTY3T1yfviBU1r8PU-5mBdfOk6P4bZ1L4BskIM,172
-agentx_python-0.8.25.dist-info/top_level.txt,sha256=s-q-HB9Gb_QdrZNacSeQyF_c25gQooMy7DlxzgLOHPk,7
-agentx_python-0.8.25.dist-info/RECORD,,
+agentx_python-0.8.26.dist-info/licenses/LICENSE,sha256=gZVsM-nLsE8vlaY6NXXsVoo6IlCClxkToAWmhgT3y_s,10762
+agentx_python-0.8.26.dist-info/METADATA,sha256=JH0SY8e_dBGzmK92xd_Cas62qpPweH0BoBLbDmE7xPA,22986
+agentx_python-0.8.26.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+agentx_python-0.8.26.dist-info/entry_points.txt,sha256=rQqF1JTY3T1yfviBU1r8PU-5mBdfOk6P4bZ1L4BskIM,172
+agentx_python-0.8.26.dist-info/top_level.txt,sha256=s-q-HB9Gb_QdrZNacSeQyF_c25gQooMy7DlxzgLOHPk,7
+agentx_python-0.8.26.dist-info/RECORD,,