mantisdk 0.1.2__tar.gz → 0.1.4__tar.gz

{mantisdk-0.1.2 → mantisdk-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mantisdk
-Version: 0.1.2
+Version: 0.1.4
 Summary: Mantisdk - AI Agent Training and Evaluation Platform
 Project-URL: Homepage, https://github.com/withmetis/mantis
 Project-URL: Documentation, https://withmetis.github.io/mantis/mantisdk/
@@ -19,7 +19,6 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Python: >=3.10
-Requires-Dist: agentops>=0.4.13
 Requires-Dist: aiohttp
 Requires-Dist: aiologic
 Requires-Dist: fastapi
@@ -28,7 +27,7 @@ Requires-Dist: gepa>=0.0.24
 Requires-Dist: gpustat
 Requires-Dist: graphviz
 Requires-Dist: gunicorn
-Requires-Dist: litellm[proxy]>=1.74
+Requires-Dist: litellm>=1.74
 Requires-Dist: openai
 Requires-Dist: opentelemetry-api>=1.35
 Requires-Dist: opentelemetry-exporter-otlp>=1.35
@@ -44,6 +43,8 @@ Provides-Extra: apo
 Requires-Dist: poml; extra == 'apo'
 Provides-Extra: mongo
 Requires-Dist: pymongo; extra == 'mongo'
+Provides-Extra: proxy
+Requires-Dist: litellm[proxy]>=1.74; extra == 'proxy'
 Provides-Extra: verl
 Requires-Dist: verl>=0.5.0; extra == 'verl'
 Requires-Dist: vllm>=0.8.4; extra == 'verl'
@@ -76,17 +77,20 @@ pip install mantisdk
 For optional dependencies:
 
 ```bash
+# For LLMProxy feature (note: has strict dependency pins for boto3, grpcio, uvicorn)
+pip install 'mantisdk[proxy]'
+
 # For APO (Automatic Prompt Optimization)
-pip install mantisdk[apo]
+pip install 'mantisdk[apo]'
 
 # For VERL integration
-pip install mantisdk[verl]
+pip install 'mantisdk[verl]'
 
 # For Weave integration
-pip install mantisdk[weave]
+pip install 'mantisdk[weave]'
 
 # For MongoDB store
-pip install mantisdk[mongo]
+pip install 'mantisdk[mongo]'
 ```
 
 ## Quick Start

{mantisdk-0.1.2 → mantisdk-0.1.4}/README.md

@@ -23,17 +23,20 @@ pip install mantisdk
 For optional dependencies:
 
 ```bash
+# For LLMProxy feature (note: has strict dependency pins for boto3, grpcio, uvicorn)
+pip install 'mantisdk[proxy]'
+
 # For APO (Automatic Prompt Optimization)
-pip install mantisdk[apo]
+pip install 'mantisdk[apo]'
 
 # For VERL integration
-pip install mantisdk[verl]
+pip install 'mantisdk[verl]'
 
 # For Weave integration
-pip install mantisdk[weave]
+pip install 'mantisdk[weave]'
 
 # For MongoDB store
-pip install mantisdk[mongo]
+pip install 'mantisdk[mongo]'
 ```
 
 ## Quick Start

{mantisdk-0.1.2 → mantisdk-0.1.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mantisdk"
-version = "0.1.2"
+version = "0.1.4"
 description = "Mantisdk - AI Agent Training and Evaluation Platform"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -28,14 +28,13 @@ dependencies = [
   "gpustat",
   "setproctitle",
   "flask",
-  "agentops>=0.4.13",
   "uvicorn",
   "fastapi",
   "aiohttp",
   "opentelemetry-api>=1.35",
   "opentelemetry-sdk>=1.35",
   "opentelemetry-exporter-otlp>=1.35",
-  "litellm[proxy]>=1.74",
+  "litellm>=1.74",
   "pydantic>=2.11",
   "openai",
   "rich",
@@ -56,6 +55,12 @@ Issues = "https://github.com/withmetis/mantis/issues"
 msk = "mantisdk.cli:main"
 
 [project.optional-dependencies]
+# LLMProxy feature requires litellm[proxy] which has strict dependency pins
+# that may conflict with other packages (boto3, grpcio, uvicorn).
+# Install with: pip install 'mantisdk[proxy]'
+proxy = [
+  "litellm[proxy]>=1.74",
+]
 apo = [
   "poml",
 ]
@@ -100,11 +105,9 @@ experiment = [
 ]
 
 core-legacy = [
-  "agentops<=0.4.18",
   "openai<2.0.0",
 ]
 core-stable = [
-  "agentops>=0.4.21",
   "openai>=2.0.0",
 ]
 
@@ -334,7 +337,6 @@ testpaths = ["tests"]
 markers = [
   "openai: tests that require OpenAI API",
   "gpu: tests that require GPU",
-  "agentops: tests that require AgentOps",
   "weave: tests that require Weave",
   "llmproxy: tests that require LiteLLM",
   "mongo: tests that require MongoDB",

{mantisdk-0.1.2 → mantisdk-0.1.4}/src/mantisdk/__init__.py

@@ -1,6 +1,6 @@
 # Copyright (c) Microsoft. All rights reserved.
 
-__version__ = "0.1.2"
+__version__ = "0.1.4"
 
 from .adapter import *
 from .algorithm import *
@@ -10,7 +10,15 @@ from .emitter import *
 from .env_var import *
 from .execution import *
 from .litagent import *
-from .llm_proxy import *
+
+# LLMProxy requires litellm[proxy] which has strict dependency pins.
+# Make import optional to avoid conflicts for users who don't need this feature.
+# Install with: pip install 'mantisdk[proxy]'
+try:
+    from .llm_proxy import *
+except ImportError:
+    pass  # LLMProxy not available without litellm[proxy] extras
+
 from .logging import configure_logger  # deprecated  # type: ignore
 from .logging import setup as setup_logging  # type: ignore
 from .logging import setup_module as setup_module_logging  # type: ignore

{mantisdk-0.1.2 → mantisdk-0.1.4}/src/mantisdk/llm_proxy.py

@@ -39,7 +39,20 @@ from fastapi import Request, Response
 from fastapi.responses import StreamingResponse
 from litellm.integrations.custom_logger import CustomLogger
 from litellm.integrations.opentelemetry import OpenTelemetry, OpenTelemetryConfig
-from litellm.proxy.proxy_server import app, save_worker_config  # pyright: ignore[reportUnknownVariableType]
+
+# litellm.proxy requires the [proxy] extras which have strict dependency pins
+# (boto3, grpcio, uvicorn) that may conflict with other packages.
+# Install with: pip install 'mantisdk[proxy]'
+try:
+    from litellm.proxy.proxy_server import app, save_worker_config  # pyright: ignore[reportUnknownVariableType]
+except ImportError as e:
+    raise ImportError(
+        "LLMProxy requires litellm[proxy] extras which are not installed. "
+        "The litellm[proxy] extras have strict dependency pins (boto3, grpcio, uvicorn) "
+        "that may conflict with other packages in your environment. "
+        "To use LLMProxy, install with: pip install 'mantisdk[proxy]'"
+    ) from e
+
 from litellm.types.utils import CallTypes
 from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
 from opentelemetry.sdk.resources import Resource

{mantisdk-0.1.2 → mantisdk-0.1.4}/src/mantisdk/tracing/__init__.py

@@ -3,25 +3,29 @@
 """Standalone tracing module for MantisDK.
 
 This module provides a "one-liner" entry point for instrumenting applications
-with OpenTelemetry, compatible with OpenInference and AgentOps ecosystems,
+with OpenTelemetry, compatible with OpenInference ecosystem,
 while natively supporting export to Mantis Insight.
 
 Example usage::
 
-    import mantisdk.tracing_claude as tracing
+    import mantisdk.tracing as tracing
 
     # Auto-detect Insight from environment variables
     tracing.init()
 
-    # Use context managers for manual spans
-    with tracing.trace("my-workflow"):
-        with tracing.span("step-1"):
-            do_work()
-
-    # Use decorators
+    # Use as a decorator (auto-captures input/output)
     @tracing.trace
-    def my_function():
-        pass
+    def my_function(query: str) -> dict:
+        return {"result": query.upper()}
+
+    # Use as a context manager (explicit input/output)
+    with tracing.trace("my-workflow", input=query) as span:
+        result = do_work()
+        span.set_attribute(tracing.semconv.TRACE_OUTPUT, result)
+
+    # Use semantic conventions for rich metadata
+    span.set_attribute(tracing.semconv.USER_ID, "user-123")
+    span.set_attribute(tracing.semconv.SESSION_ID, "session-456")
 
     # Ensure spans are flushed on shutdown
     tracing.shutdown()
@@ -34,7 +38,8 @@ Environment variables for Insight auto-detect:
 """
 
 from .init import init, instrument, shutdown, flush
-from .api import trace, span, tool, atrace, aspan
+from .api import trace, span, tool, atrace, aspan, get_current_span
+from . import semconv
 
 # Re-export exporter factory for explicit configuration
 from .exporters.insight import insight as insight_exporter
@@ -45,13 +50,17 @@ __all__ = [
     "instrument",
     "shutdown",
     "flush",
-    # Context managers (sync)
+    # Context managers / decorators (sync)
     "trace",
     "span",
     "tool",
     # Context managers (async)
     "atrace",
     "aspan",
+    # Span utilities
+    "get_current_span",
+    # Semantic conventions
+    "semconv",
     # Exporter factory
     "insight_exporter",
 ]