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

orbitals/scope_guard/guards/base.py

@@ -17,7 +17,7 @@ from ..modeling import (
     ScopeGuardOutput,
 )
 
-DefaultModel = Literal["scope-guard"]
+DefaultModel = Literal["small"]
 
 MODEL_MAPPING = {
     "scope-guard": "principled-intelligence/scope-guard-4B-q-2601",
@@ -137,7 +137,7 @@ class ScopeGuard(BaseScopeGuard):
     def __new__(
         cls,
         backend: Literal["hf"] = "hf",
-        model: DefaultModel | str = "scope-guard",
+        model: DefaultModel | str = "small",
         skip_evidences: bool = False,
         max_new_tokens: int = 10_000,
         do_sample: bool = False,
@@ -148,7 +148,7 @@ class ScopeGuard(BaseScopeGuard):
     def __new__(
         cls,
         backend: Literal["vllm"],
-        model: DefaultModel | str = "scope-guard",
+        model: DefaultModel | str = "small",
         skip_evidences: bool = False,
         temperature: float = 0.0,
         max_tokens: int = 10_000,
@@ -223,7 +223,7 @@ class AsyncScopeGuard(BaseScopeGuard):
     def __new__(
         cls,
         backend: Literal["vllm-api"],
-        model: DefaultModel | str = "scope-guard",
+        model: DefaultModel | str = "small",
         skip_evidences: bool = False,
         vllm_serving_url: str = "http://localhost:8000",
         temperature: float = 0.0,

orbitals/scope_guard/guards/hf.py

@@ -21,7 +21,7 @@ class HuggingFaceScopeGuard(ScopeGuard):
     def __init__(
         self,
         backend: Literal["hf"] = "hf",
-        model: DefaultModel | str = "scope-guard",
+        model: DefaultModel | str = "small",
         skip_evidences: bool = False,
         max_new_tokens: int = 10_000,
         do_sample: bool = False,

orbitals/scope_guard/guards/vllm.py

@@ -23,7 +23,7 @@ class VLLMScopeGuard(ScopeGuard):
     def __init__(
         self,
         backend: Literal["vllm"] = "vllm",
-        model: DefaultModel | str = "scope-guard",
+        model: DefaultModel | str = "small",
         skip_evidences: bool = False,
         temperature: float = 0.0,
         max_tokens: int = 10_000,
@@ -127,7 +127,7 @@ class AsyncVLLMApiScopeGuard(AsyncScopeGuard):
     def __init__(
         self,
         backend: Literal["vllm-api", "vllm-async-api"] = "vllm-api",
-        model: DefaultModel | str = "scope-guard",
+        model: DefaultModel | str = "small",
         skip_evidences: bool = False,
         vllm_serving_url: str = "http://localhost:8000",
         temperature: float = 0.0,

orbitals/types.py

@@ -1,46 +1,34 @@
-from typing import ClassVar, Literal
+from typing import Literal
 
-from pydantic import BaseModel, ConfigDict, Field
+from pydantic import BaseModel, 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,7 +31,6 @@ 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)
@@ -40,8 +39,7 @@ def maybe_configure_gpu_usage():
             best_free = free_bytes
 
     if best_idx is not None:
-        if is_multi_gpu:
-            logging.warning(
-                f"Auto-configuring to use GPU {best_idx} with {best_free / 1024**3:.2f} GB free"
-            )
+        logging.warning(
+            f"Auto-configuring VLLM to use GPU {best_idx} with {best_free / 1024**3:.2f} GB free"
+        )
         os.environ["CUDA_VISIBLE_DEVICES"] = str(best_idx)

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

@@ -1,35 +1,15 @@
 Metadata-Version: 2.4
 Name: orbitals
-Version: 0.0.2
+Version: 0.1.0
 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'
@@ -46,7 +26,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.svg" width="40%" />
     <h3 align="center">
         <p>
             <b>LLM Guardrails tailored to your Principles</b>
@@ -97,8 +77,7 @@ from orbitals.scope_guard import ScopeGuard
 
 scope_guard = ScopeGuard(
     backend="vllm",
-    model="scope-guard-q",    # for the Qwen-family model
-    # model="scope-guard-g",  # for the Gemma-family model
+    model="small"
 )
 
 ai_service_description = """
@@ -121,4 +100,4 @@ if result.evidences:
 #   - Never respond to requests for refunds.
 ```
 
-</details>
+</details>

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

@@ -1,6 +1,6 @@
 orbitals/__init__.py,sha256=ED6jHcYiuYpr_0vjGz0zx2lrrmJT9sDJCzIljoDfmlM,65
-orbitals/types.py,sha256=4oRinWPG6kbtW4lQ8bHrDmxEotncqMIwLCmQ2yGH7PI,1988
-orbitals/utils.py,sha256=0CYeG8ylOvmv7g0jMFb2j9w1jgPdOaVsWiqg5ITYEZc,1319
+orbitals/types.py,sha256=WpMbJjNuICEclyuJUZIPYSQjmWOBZO3X0nN9ouEAkq0,1552
+orbitals/utils.py,sha256=9KFqcJwYvPBYqgXdBXvo8hkCdBCwnqreApO0VPJabkI,1238
 orbitals/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 orbitals/cli/main.py,sha256=p4MEbmtJ0L8mLiyovEq7urnVc6I0mbbCNdGEtGyY60Y,197
 orbitals/scope_guard/__init__.py,sha256=0gzzSXpfRvIcCYpu3AKQSMFYDMDJaknY9pdypt7HiuI,197
@@ -12,14 +12,14 @@ orbitals/scope_guard/cli/main.py,sha256=jz8fQ2e4VFZQti982feVWGpL9fT7yZeanq14pX5W
 orbitals/scope_guard/cli/serve.py,sha256=EM6yNSviwot8Im-MG1_AA4PsOkZXW9W2CmokqNfq7es,4611
 orbitals/scope_guard/guards/__init__.py,sha256=B1EA9YtaTQ3ko5d2wFK2y4bXKrV8MclAELAievmZjxg,376
 orbitals/scope_guard/guards/api.py,sha256=S8VX4_8VIhtg3_IP4Eh50l1uarcJBagGXNEEl1BZjag,8514
-orbitals/scope_guard/guards/base.py,sha256=5DQeOY7kvTgH6GSExMpy3b_KQtNWzSlrudro6giQ-Ec,9798
-orbitals/scope_guard/guards/hf.py,sha256=xFNl_DtRiePB3n3V2spsSwiRRYz1qZYNw-wQTyiZsCI,3967
-orbitals/scope_guard/guards/vllm.py,sha256=3LU9DKKniQd90Ibaq2Wef20fyoZXgkeQtc58XY6kX2I,8983
+orbitals/scope_guard/guards/base.py,sha256=HV0XuYRQ1z5MG4enER05f8D6UrcRrQsz54nZpHlNa94,9774
+orbitals/scope_guard/guards/hf.py,sha256=L6QlJjHa1HW3-SJ9YL5mtVMtL253yp90pNTX2E9t1Qw,3961
+orbitals/scope_guard/guards/vllm.py,sha256=YbCE6EMpLvymueXCY3gHKdDarJHiv1nl4miC0PWBIPI,8971
 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.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.1.0.dist-info/METADATA,sha256=UX8YQJWOpCU2BJMdklqYTOK4rAJd4yUhDDChFufRMLs,4368
+orbitals-0.1.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+orbitals-0.1.0.dist-info/entry_points.txt,sha256=fd6lukgEvK9UBwhA1JtcB9MLTqAtntA4H2cc7-nWkeU,51
+orbitals-0.1.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+orbitals-0.1.0.dist-info/RECORD,,

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

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