mantisdk 0.1.4__tar.gz → 0.1.6__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mantisdk
-Version: 0.1.4
+Version: 0.1.6
 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

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

@@ -1,6 +1,6 @@
 [project]
 name = "mantisdk"
-version = "0.1.4"
+version = "0.1.6"
 description = "Mantisdk - AI Agent Training and Evaluation Platform"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -28,6 +28,7 @@ dependencies = [
   "gpustat",
   "setproctitle",
   "flask",
+  "agentops>=0.4.13",
   "uvicorn",
   "fastapi",
   "aiohttp",

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

@@ -1,6 +1,6 @@
 # Copyright (c) Microsoft. All rights reserved.
 
-__version__ = "0.1.4"
+__version__ = "0.1.6"
 
 from .adapter import *
 from .algorithm import *
@@ -26,5 +26,12 @@ from .runner import *
 from .server import MantisdkServer  # deprecated  # type: ignore
 from .store import *
 from .tracer import *
-from .trainer import *
+
+# Trainer requires LLMProxy which requires litellm[proxy]
+# Make import optional to avoid conflicts for users who don't need training features.
+try:
+    from .trainer import *
+except ImportError:
+    pass  # Trainer not available without litellm[proxy] extras
+
 from .types import *