orbitals 0.0.1__py3-none-any.whl → 0.0.2__py3-none-any.whl

orbitals/types.py

@@ -1,34 +1,46 @@
-from typing import Literal
+from typing import ClassVar, Literal
 
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, ConfigDict, Field
 
 
 class LLMUsage(BaseModel):
+    model_config: ClassVar[ConfigDict] = ConfigDict(extra="forbid")
+
     prompt_tokens: int
     completion_tokens: int
     total_tokens: int
 
 
 class ConversationMessage(BaseModel):
+    model_config: ClassVar[ConfigDict] = ConfigDict(extra="forbid")
+
     role: Literal["user", "assistant"]
     content: str
 
 
 class Conversation(BaseModel):
+    model_config: ClassVar[ConfigDict] = ConfigDict(extra="forbid")
+
     messages: list[ConversationMessage]
 
 
 class SupportingMaterial(BaseModel):
+    model_config: ClassVar[ConfigDict] = ConfigDict(extra="forbid")
+
     pass
 
 
 class Principle(BaseModel):
+    model_config: ClassVar[ConfigDict] = ConfigDict(extra="forbid")
+
     title: str = Field(description="Title of the principle")
     description: str = Field(description="Description of the principle")
     supporting_materials: list[SupportingMaterial] | None
 
 
 class AIServiceDescription(BaseModel):
+    model_config: ClassVar[ConfigDict] = ConfigDict(extra="forbid")
+
     identity_role: str = Field(
         description="Identity, role and objectives of the AI Service. Gives a general idea of what the service is about."
     )

orbitals/utils.py

@@ -31,6 +31,7 @@ def maybe_configure_gpu_usage():
 
     best_idx = None
     best_free = -1
+    is_multi_gpu = torch.cuda.device_count() > 1
 
     for i in range(torch.cuda.device_count()):
         free_bytes, _ = torch.cuda.mem_get_info(i)  # (free, total)
@@ -39,7 +40,8 @@ def maybe_configure_gpu_usage():
             best_free = free_bytes
 
     if best_idx is not None:
-        logging.warning(
-            f"Auto-configuring VLLM to use GPU {best_idx} with {best_free / 1024**3:.2f} GB free"
-        )
+        if is_multi_gpu:
+            logging.warning(
+                f"Auto-configuring to use GPU {best_idx} with {best_free / 1024**3:.2f} GB free"
+            )
         os.environ["CUDA_VISIBLE_DEVICES"] = str(best_idx)

{orbitals-0.0.1.dist-info → orbitals-0.0.2.dist-info}/METADATA

@@ -1,15 +1,35 @@
 Metadata-Version: 2.4
 Name: orbitals
-Version: 0.0.1
+Version: 0.0.2
 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
 License-File: LICENSE
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.10
+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-Dist: aiohttp
 Requires-Dist: pydantic>=2.0.0
 Requires-Dist: requests
 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: 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: 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: 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: xgrammar; extra == 'scope-guard-all'
 Provides-Extra: scope-guard-hf
 Requires-Dist: accelerate>=1.11.0; extra == 'scope-guard-hf'
 Requires-Dist: transformers<5.0.0,>=4.47.0; extra == 'scope-guard-hf'
@@ -26,7 +46,7 @@ Requires-Dist: xgrammar; extra == 'scope-guard-vllm'
 Description-Content-Type: text/markdown
 
 <div align="center">
-    <img src="assets/Orbitals Banner.png" width="70%" />
+    <img src="assets/orbitals-banner.png" width="70%" />
     <h3 align="center">
         <p>
             <b>LLM Guardrails tailored to your Principles</b>
@@ -77,7 +97,8 @@ from orbitals.scope_guard import ScopeGuard
 
 scope_guard = ScopeGuard(
     backend="vllm",
-    model="scope-guard"
+    model="scope-guard-q",    # for the Qwen-family model
+    # model="scope-guard-g",  # for the Gemma-family model
 )
 
 ai_service_description = """

{orbitals-0.0.1.dist-info → orbitals-0.0.2.dist-info}/RECORD

@@ -1,6 +1,6 @@
 orbitals/__init__.py,sha256=ED6jHcYiuYpr_0vjGz0zx2lrrmJT9sDJCzIljoDfmlM,65
-orbitals/types.py,sha256=WpMbJjNuICEclyuJUZIPYSQjmWOBZO3X0nN9ouEAkq0,1552
-orbitals/utils.py,sha256=9KFqcJwYvPBYqgXdBXvo8hkCdBCwnqreApO0VPJabkI,1238
+orbitals/types.py,sha256=4oRinWPG6kbtW4lQ8bHrDmxEotncqMIwLCmQ2yGH7PI,1988
+orbitals/utils.py,sha256=0CYeG8ylOvmv7g0jMFb2j9w1jgPdOaVsWiqg5ITYEZc,1319
 orbitals/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 orbitals/cli/main.py,sha256=p4MEbmtJ0L8mLiyovEq7urnVc6I0mbbCNdGEtGyY60Y,197
 orbitals/scope_guard/__init__.py,sha256=0gzzSXpfRvIcCYpu3AKQSMFYDMDJaknY9pdypt7HiuI,197
@@ -18,8 +18,8 @@ orbitals/scope_guard/guards/vllm.py,sha256=3LU9DKKniQd90Ibaq2Wef20fyoZXgkeQtc58X
 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/vllm_logging_config.json,sha256=Bc08X8mQWJFAAHEE6ZFVUGnRc77pMVpPvji6BhFTtSE,651
-orbitals-0.0.1.dist-info/METADATA,sha256=p0OEEBeFIf3pU1eoJSuXuf0OgWKhg3rSuwDkPuzokYo,4382
-orbitals-0.0.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-orbitals-0.0.1.dist-info/entry_points.txt,sha256=fd6lukgEvK9UBwhA1JtcB9MLTqAtntA4H2cc7-nWkeU,51
-orbitals-0.0.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-orbitals-0.0.1.dist-info/RECORD,,
+orbitals-0.0.2.dist-info/METADATA,sha256=QH7qs5k6izQj5pAnxgBQ2iUNN1S4ydTKpY8iVl13nRw,5506
+orbitals-0.0.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+orbitals-0.0.2.dist-info/entry_points.txt,sha256=fd6lukgEvK9UBwhA1JtcB9MLTqAtntA4H2cc7-nWkeU,51
+orbitals-0.0.2.dist-info/licenses/LICENSE,sha256=Eeclrom-K-omYcKnMvijEMV-IMiQ7X-bdgxlZcXcImI,11360
+orbitals-0.0.2.dist-info/RECORD,,

{orbitals-0.0.1.dist-info → orbitals-0.0.2.dist-info}/licenses/LICENSE

@@ -186,7 +186,7 @@
       same "printed page" as the copyright notice for easier
       identification within third-party archives.
 
-   Copyright [yyyy] [name of copyright owner]
+   Copyright 2026 Principled Intelligence s.r.l.
 
    Licensed under the Apache License, Version 2.0 (the "License");
    you may not use this file except in compliance with the License.