mantisdk 0.1.3__tar.gz → 0.1.5__tar.gz

{mantisdk-0.1.3 → mantisdk-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mantisdk
-Version: 0.1.3
+Version: 0.1.5
 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,6 +19,7 @@ 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
@@ -27,7 +28,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
@@ -43,6 +44,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'
@@ -75,17 +78,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.3 → mantisdk-0.1.5}/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.3 → mantisdk-0.1.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mantisdk"
-version = "0.1.3"
+version = "0.1.5"
 description = "Mantisdk - AI Agent Training and Evaluation Platform"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -28,13 +28,14 @@ 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",
@@ -55,6 +56,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",
 ]

{mantisdk-0.1.3 → mantisdk-0.1.5}/src/mantisdk/__init__.py

@@ -1,6 +1,6 @@
 # Copyright (c) Microsoft. All rights reserved.
 
-__version__ = "0.1.3"
+__version__ = "0.1.5"
 
 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.3 → mantisdk-0.1.5}/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