mantisdk 0.1.3__py3-none-any.whl → 0.1.4__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
mantisdk/__init__.py CHANGED
@@ -1,6 +1,6 @@
1
1
  # Copyright (c) Microsoft. All rights reserved.
2
2
 
3
- __version__ = "0.1.3"
3
+ __version__ = "0.1.4"
4
4
 
5
5
  from .adapter import *
6
6
  from .algorithm import *
@@ -10,7 +10,15 @@ from .emitter import *
10
10
  from .env_var import *
11
11
  from .execution import *
12
12
  from .litagent import *
13
- from .llm_proxy import *
13
+
14
+ # LLMProxy requires litellm[proxy] which has strict dependency pins.
15
+ # Make import optional to avoid conflicts for users who don't need this feature.
16
+ # Install with: pip install 'mantisdk[proxy]'
17
+ try:
18
+ from .llm_proxy import *
19
+ except ImportError:
20
+ pass # LLMProxy not available without litellm[proxy] extras
21
+
14
22
  from .logging import configure_logger # deprecated # type: ignore
15
23
  from .logging import setup as setup_logging # type: ignore
16
24
  from .logging import setup_module as setup_module_logging # type: ignore
mantisdk/llm_proxy.py CHANGED
@@ -39,7 +39,20 @@ from fastapi import Request, Response
39
39
  from fastapi.responses import StreamingResponse
40
40
  from litellm.integrations.custom_logger import CustomLogger
41
41
  from litellm.integrations.opentelemetry import OpenTelemetry, OpenTelemetryConfig
42
- from litellm.proxy.proxy_server import app, save_worker_config # pyright: ignore[reportUnknownVariableType]
42
+
43
+ # litellm.proxy requires the [proxy] extras which have strict dependency pins
44
+ # (boto3, grpcio, uvicorn) that may conflict with other packages.
45
+ # Install with: pip install 'mantisdk[proxy]'
46
+ try:
47
+ from litellm.proxy.proxy_server import app, save_worker_config # pyright: ignore[reportUnknownVariableType]
48
+ except ImportError as e:
49
+ raise ImportError(
50
+ "LLMProxy requires litellm[proxy] extras which are not installed. "
51
+ "The litellm[proxy] extras have strict dependency pins (boto3, grpcio, uvicorn) "
52
+ "that may conflict with other packages in your environment. "
53
+ "To use LLMProxy, install with: pip install 'mantisdk[proxy]'"
54
+ ) from e
55
+
43
56
  from litellm.types.utils import CallTypes
44
57
  from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
45
58
  from opentelemetry.sdk.resources import Resource
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: mantisdk
3
- Version: 0.1.3
3
+ Version: 0.1.4
4
4
  Summary: Mantisdk - AI Agent Training and Evaluation Platform
5
5
  Project-URL: Homepage, https://github.com/withmetis/mantis
6
6
  Project-URL: Documentation, https://withmetis.github.io/mantis/mantisdk/
@@ -27,7 +27,7 @@ Requires-Dist: gepa>=0.0.24
27
27
  Requires-Dist: gpustat
28
28
  Requires-Dist: graphviz
29
29
  Requires-Dist: gunicorn
30
- Requires-Dist: litellm[proxy]>=1.74
30
+ Requires-Dist: litellm>=1.74
31
31
  Requires-Dist: openai
32
32
  Requires-Dist: opentelemetry-api>=1.35
33
33
  Requires-Dist: opentelemetry-exporter-otlp>=1.35
@@ -43,6 +43,8 @@ Provides-Extra: apo
43
43
  Requires-Dist: poml; extra == 'apo'
44
44
  Provides-Extra: mongo
45
45
  Requires-Dist: pymongo; extra == 'mongo'
46
+ Provides-Extra: proxy
47
+ Requires-Dist: litellm[proxy]>=1.74; extra == 'proxy'
46
48
  Provides-Extra: verl
47
49
  Requires-Dist: verl>=0.5.0; extra == 'verl'
48
50
  Requires-Dist: vllm>=0.8.4; extra == 'verl'
@@ -75,17 +77,20 @@ pip install mantisdk
75
77
  For optional dependencies:
76
78
 
77
79
  ```bash
80
+ # For LLMProxy feature (note: has strict dependency pins for boto3, grpcio, uvicorn)
81
+ pip install 'mantisdk[proxy]'
82
+
78
83
  # For APO (Automatic Prompt Optimization)
79
- pip install mantisdk[apo]
84
+ pip install 'mantisdk[apo]'
80
85
 
81
86
  # For VERL integration
82
- pip install mantisdk[verl]
87
+ pip install 'mantisdk[verl]'
83
88
 
84
89
  # For Weave integration
85
- pip install mantisdk[weave]
90
+ pip install 'mantisdk[weave]'
86
91
 
87
92
  # For MongoDB store
88
- pip install mantisdk[mongo]
93
+ pip install 'mantisdk[mongo]'
89
94
  ```
90
95
 
91
96
  ## Quick Start
@@ -1,8 +1,8 @@
1
- mantisdk/__init__.py,sha256=iVsMd8gsA3K-KIIWr_GlCT-WXI-N5hYF41Ql8ukjFis,718
1
+ mantisdk/__init__.py,sha256=a5HDMvI56AiLG46vFd-DZ9JfxqZHgzr96lqRmdJ6HKo,1010
2
2
  mantisdk/client.py,sha256=QTH-0LDAKWLuLH7PSaZWj7XUcnbBGrJHH-w2dJiNmKM,17059
3
3
  mantisdk/config.py,sha256=9RAdsbTkSG_qnLvTwH0xTG5srLFaj1nk5Ufz1dQHGwc,14355
4
4
  mantisdk/env_var.py,sha256=OCb5CvZo5U7-HBnqC0ZmdsrS2giAh7dV85TTjz6wLqA,4372
5
- mantisdk/llm_proxy.py,sha256=78y6LU_8gevKXcYjUm_Zu4U6CvdfwqDzWQ8Ec5gM5I4,82591
5
+ mantisdk/llm_proxy.py,sha256=6_4zOO7Q94DnK4dZcgIDjJRnz-yiEJsgGjuYanAgsTA,83160
6
6
  mantisdk/logging.py,sha256=Kj1iEcoeEmOukM_QBEEs2qu0e83JMfkgA3vDj9Lnag8,13639
7
7
  mantisdk/reward.py,sha256=lAnW_W4ITQJjlIKbU1SgAl4KfOXjc3LTbJpaVIBV3Sw,201
8
8
  mantisdk/semconv.py,sha256=aLv8-z_8a0t3YVyv6Mgo6ohl0lPngZcZK80-ZBiyPvw,5624
@@ -194,8 +194,8 @@ mantisdk/verl/daemon.py,sha256=cvIGDkotN6MPp-DubCeKBfD9hSoZCgRyIACI9X7D7uw,54493
194
194
  mantisdk/verl/dataset.py,sha256=Hr0K9oXuj7q8pf08BuvNRTbIgyNo0XY_N_JVA5jHUvI,1175
195
195
  mantisdk/verl/entrypoint.py,sha256=dnZ7TwCe9hC1kRXm_ypRTrSR9cPXz3LcxXYjtPVuOHs,8627
196
196
  mantisdk/verl/trainer.py,sha256=TCaVXo65iNcEEGujuIbx2yQ_cMfBYyU5Uudx9gOBSNI,25447
197
- mantisdk-0.1.3.dist-info/METADATA,sha256=eaJbnQsPVqpKZXwfkBwqCGFYPWf7TlSm9sUVpPglD8c,3386
198
- mantisdk-0.1.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
199
- mantisdk-0.1.3.dist-info/entry_points.txt,sha256=yDYe8wx2pXbBXBaRPtRwt6OL01VPHf6GiFsfauISW3M,42
200
- mantisdk-0.1.3.dist-info/licenses/LICENSE,sha256=j3Flk3DFJo2aHclipGIyVA6PymNGJYbY76qVqrSSogg,1046
201
- mantisdk-0.1.3.dist-info/RECORD,,
197
+ mantisdk-0.1.4.dist-info/METADATA,sha256=bCAdjn-ZHhutNSbar9NI6UHGwSudxDZfuGpfjnDN2i0,3579
198
+ mantisdk-0.1.4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
199
+ mantisdk-0.1.4.dist-info/entry_points.txt,sha256=yDYe8wx2pXbBXBaRPtRwt6OL01VPHf6GiFsfauISW3M,42
200
+ mantisdk-0.1.4.dist-info/licenses/LICENSE,sha256=j3Flk3DFJo2aHclipGIyVA6PymNGJYbY76qVqrSSogg,1046
201
+ mantisdk-0.1.4.dist-info/RECORD,,