orbitals 0.0.3__py3-none-any.whl → 0.1.1__py3-none-any.whl

orbitals/scope_guard/serving/main.py

@@ -54,19 +54,13 @@ class ScopeGuardResponse(BaseModel):
     time_taken: float
 
 
-@app.post("/api/orbitals/scope-guard/validate", response_model=ScopeGuardResponse)
-@app.post("/api/in/scope-classifier/classify", response_model=ScopeGuardResponse)
+@app.post("/orbitals/scope-guard/validate", response_model=ScopeGuardResponse)
 async def validate(
     request: Request,
     conversation: ScopeGuardInput,
     ai_service_description: Annotated[str | AIServiceDescription, Body()],
     skip_evidences: bool | None = Body(None),
 ) -> ScopeGuardResponse:
-    if request.url.path.endswith("/api/in/scope-classifier/classify"):
-        logging.warning(
-            "The /api/in/scope-classifier/classify endpoint is deprecated. Please use /api/in/scope-guard/validate instead."
-        )
-
     global scope_guard
 
     start_time = time.time()
@@ -85,10 +79,9 @@ async def validate(
 
 
 @app.post(
-    "/api/orbitals/scope-guard/batch-validate",
+    "/orbitals/scope-guard/batch-validate",
     response_model=list[ScopeGuardResponse],
 )
-@app.post("/api/in/scope-classifier/batch-classify", response_model=ScopeGuardResponse)
 async def batch_validate(
     request: Request,
     conversations: list[ScopeGuardInput],
@@ -96,11 +89,6 @@ async def batch_validate(
     ai_service_descriptions: list[str] | list[AIServiceDescription] | None = None,
     skip_evidences: bool | None = Body(None),
 ) -> list[ScopeGuardResponse]:
-    if request.url.path.endswith("/api/in/scope-classifier/batch-classify"):
-        logging.warning(
-            "The /api/in/scope-classifier/batch-classify endpoint is deprecated. Please use /api/in/scope-guard/batch-validate instead."
-        )
-
     global scope_guard
 
     start_time = time.time()

orbitals/utils.py

@@ -8,7 +8,7 @@ def maybe_configure_gpu_usage():
     optimal usage: search for the gpu with the most free memory, and
     set CUDA_VISIBLE_DEVICES to that GPU only.
 
-    Uses pynvml to avoid triggering CUDA initialization from torch.
+    Uses nvidia-ml-py (pynvml) to avoid triggering CUDA initialization from torch.
     """
     if "CUDA_VISIBLE_DEVICES" in os.environ:
         logging.info(
@@ -17,15 +17,15 @@ def maybe_configure_gpu_usage():
         return
 
     try:
-        import pynvml  # ty: ignore[unresolved-import]
+        import pynvml  # ty: ignore[unresolved-import]  # provided by nvidia-ml-py
     except ModuleNotFoundError:
-        logging.debug("pynvml not available, skipping GPU auto-configuration")
+        logging.debug("nvidia-ml-py not available, skipping GPU auto-configuration")
         return
 
     try:
         pynvml.nvmlInit()
     except pynvml.NVMLError:
-        logging.debug("NVML initialization failed, skipping GPU auto-configuration")
+        logging.error("NVML initialization failed, skipping GPU auto-configuration")
         return
 
     try:

{orbitals-0.0.3.dist-info → orbitals-0.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: orbitals
-Version: 0.0.3
+Version: 0.1.1
 Summary: LLM Guardrails tailored to your Principles
 Author-email: Luigi Procopio <luigi@principled-intelligence.com>, Edoardo Barba <edoardo@principled-intelligence.com>
 License: Apache-2.0
@@ -11,7 +11,7 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Programming Language :: Python :: 3.14
-Requires-Python: >=3.10
+Requires-Python: ==3.13
 Requires-Dist: aiohttp
 Requires-Dist: pydantic>=2.0.0
 Requires-Dist: requests
@@ -19,35 +19,38 @@ Requires-Dist: typer>=0.12.3
 Provides-Extra: all
 Requires-Dist: accelerate>=1.11.0; extra == 'all'
 Requires-Dist: fastapi[standard]>=0.119.1; extra == 'all'
-Requires-Dist: pynvml; extra == 'all'
+Requires-Dist: nvidia-ml-py; extra == 'all'
 Requires-Dist: transformers<5.0.0,>=4.47.0; extra == 'all'
 Requires-Dist: uvicorn>=0.29.0; extra == 'all'
-Requires-Dist: vllm>=0.11.0; extra == 'all'
+Requires-Dist: vllm<0.13.0,>=0.11.0; extra == 'all'
 Requires-Dist: xgrammar; extra == 'all'
 Provides-Extra: scope-guard-all
 Requires-Dist: accelerate>=1.11.0; extra == 'scope-guard-all'
 Requires-Dist: fastapi[standard]>=0.119.1; extra == 'scope-guard-all'
-Requires-Dist: pynvml; extra == 'scope-guard-all'
+Requires-Dist: nvidia-ml-py; extra == 'scope-guard-all'
 Requires-Dist: transformers<5.0.0,>=4.47.0; extra == 'scope-guard-all'
 Requires-Dist: uvicorn>=0.29.0; extra == 'scope-guard-all'
-Requires-Dist: vllm>=0.11.0; extra == 'scope-guard-all'
+Requires-Dist: vllm<0.13.0,>=0.11.0; extra == 'scope-guard-all'
 Requires-Dist: xgrammar; extra == 'scope-guard-all'
 Provides-Extra: scope-guard-hf
 Requires-Dist: accelerate>=1.11.0; extra == 'scope-guard-hf'
-Requires-Dist: pynvml; extra == 'scope-guard-hf'
+Requires-Dist: nvidia-ml-py; extra == 'scope-guard-hf'
 Requires-Dist: transformers<5.0.0,>=4.47.0; extra == 'scope-guard-hf'
 Provides-Extra: scope-guard-serve
 Requires-Dist: fastapi[standard]>=0.119.1; extra == 'scope-guard-serve'
-Requires-Dist: pynvml; extra == 'scope-guard-serve'
+Requires-Dist: nvidia-ml-py; extra == 'scope-guard-serve'
 Requires-Dist: transformers<5.0.0,>=4.47.0; extra == 'scope-guard-serve'
 Requires-Dist: uvicorn>=0.29.0; extra == 'scope-guard-serve'
-Requires-Dist: vllm>=0.11.0; extra == 'scope-guard-serve'
+Requires-Dist: vllm<0.13.0,>=0.11.0; extra == 'scope-guard-serve'
 Requires-Dist: xgrammar; extra == 'scope-guard-serve'
 Provides-Extra: scope-guard-vllm
-Requires-Dist: pynvml; extra == 'scope-guard-vllm'
+Requires-Dist: nvidia-ml-py; extra == 'scope-guard-vllm'
 Requires-Dist: transformers<5.0.0,>=4.47.0; extra == 'scope-guard-vllm'
-Requires-Dist: vllm>=0.11.0; extra == 'scope-guard-vllm'
+Requires-Dist: vllm<0.13.0,>=0.11.0; extra == 'scope-guard-vllm'
 Requires-Dist: xgrammar; extra == 'scope-guard-vllm'
+Provides-Extra: serving
+Requires-Dist: fastapi[standard]>=0.119.1; extra == 'serving'
+Requires-Dist: uvicorn>=0.29.0; extra == 'serving'
 Description-Content-Type: text/markdown
 
 <div align="center">

{orbitals-0.0.3.dist-info → orbitals-0.1.1.dist-info}/RECORD

@@ -1,6 +1,6 @@
 orbitals/__init__.py,sha256=ED6jHcYiuYpr_0vjGz0zx2lrrmJT9sDJCzIljoDfmlM,65
 orbitals/types.py,sha256=4oRinWPG6kbtW4lQ8bHrDmxEotncqMIwLCmQ2yGH7PI,1988
-orbitals/utils.py,sha256=9UFlpzTkSJeyN2n1CzUy5jIZFqnOwNYNpF4c5xIfT3E,1628
+orbitals/utils.py,sha256=FoCNGgPNqD4mey7_f3Rj5XBJKsp7-krcvvr5KvmnNG4,1677
 orbitals/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 orbitals/cli/main.py,sha256=p4MEbmtJ0L8mLiyovEq7urnVc6I0mbbCNdGEtGyY60Y,197
 orbitals/scope_guard/__init__.py,sha256=0gzzSXpfRvIcCYpu3AKQSMFYDMDJaknY9pdypt7HiuI,197
@@ -16,10 +16,10 @@ orbitals/scope_guard/guards/base.py,sha256=5DQeOY7kvTgH6GSExMpy3b_KQtNWzSlrudro6
 orbitals/scope_guard/guards/hf.py,sha256=xFNl_DtRiePB3n3V2spsSwiRRYz1qZYNw-wQTyiZsCI,3967
 orbitals/scope_guard/guards/vllm.py,sha256=3LU9DKKniQd90Ibaq2Wef20fyoZXgkeQtc58XY6kX2I,8983
 orbitals/scope_guard/serving/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-orbitals/scope_guard/serving/main.py,sha256=pjUol71h7CuU_6PiB7Zjdm5eNyDVH-F8bZVNeyEL-tE,3822
+orbitals/scope_guard/serving/main.py,sha256=ijOtsYOzsbXrgJrWKvNRjL5LmrxbHE74GbwgoXwulA8,3162
 orbitals/scope_guard/serving/vllm_logging_config.json,sha256=Bc08X8mQWJFAAHEE6ZFVUGnRc77pMVpPvji6BhFTtSE,651
-orbitals-0.0.3.dist-info/METADATA,sha256=pgtkFPmdlhU9pnorq_cfSnf0Egc36vMEbY1elxaohfU,7148
-orbitals-0.0.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-orbitals-0.0.3.dist-info/entry_points.txt,sha256=fd6lukgEvK9UBwhA1JtcB9MLTqAtntA4H2cc7-nWkeU,51
-orbitals-0.0.3.dist-info/licenses/LICENSE,sha256=Eeclrom-K-omYcKnMvijEMV-IMiQ7X-bdgxlZcXcImI,11360
-orbitals-0.0.3.dist-info/RECORD,,
+orbitals-0.1.1.dist-info/METADATA,sha256=ot2miZ_tcEGSlultyCrwqHh_Ev5mFGflWhp9KDxn-RI,7347
+orbitals-0.1.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+orbitals-0.1.1.dist-info/entry_points.txt,sha256=fd6lukgEvK9UBwhA1JtcB9MLTqAtntA4H2cc7-nWkeU,51
+orbitals-0.1.1.dist-info/licenses/LICENSE,sha256=Eeclrom-K-omYcKnMvijEMV-IMiQ7X-bdgxlZcXcImI,11360
+orbitals-0.1.1.dist-info/RECORD,,