lumen-resources 0.3.2__tar.gz → 0.4.1__tar.gz

{lumen_resources-0.3.2 → lumen_resources-0.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lumen-resources
-Version: 0.3.2
+Version: 0.4.1
 Summary: Unified model resource management for Lumen ML services
 Author-email: EdwinZhanCN <support@lumilio.org>
 License: MIT

{lumen_resources-0.3.2 → lumen_resources-0.4.1}/src/lumen_resources/lumen_config.py

@@ -1,11 +1,11 @@
 # generated by datamodel-codegen:
 #   filename:  config-schema.yaml
-#   timestamp: 2025-12-12T07:57:50+00:00
+#   timestamp: 2025-12-23T18:16:59+00:00
 
 from __future__ import annotations
 
 from enum import Enum
-from typing import Literal
+from typing import Any, Literal
 
 from pydantic import BaseModel, ConfigDict, Field, RootModel
 
@@ -15,13 +15,16 @@ class Region(Enum):
     Platform region selection (cn=ModelScope, other=HuggingFace)
     """
 
-    cn = "cn"
-    other = "other"
+    cn = 'cn'
+    other = 'other'
 
 
 class Metadata(BaseModel):
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
     version: str = Field(
-        ..., examples=["1.0.0", "2.1.3"], pattern="^\\d+\\.\\d+\\.\\d+$"
+        ..., examples=['1.0.0', '2.1.3'], pattern='^\\d+\\.\\d+\\.\\d+$'
     )
     """
     Configuration version (semantic versioning)
@@ -30,7 +33,7 @@ class Metadata(BaseModel):
     """
     Platform region selection (cn=ModelScope, other=HuggingFace)
     """
-    cache_dir: str = Field(..., examples=["~/.lumen/models", "/opt/lumen/models"])
+    cache_dir: str = Field(..., examples=['~/.lumen/models', '/opt/lumen/models'])
     """
     Model cache directory path (supports ~ expansion)
     """
@@ -41,20 +44,26 @@ class Mode(Enum):
     Deployment mode
     """
 
-    single = "single"
-    hub = "hub"
+    single = 'single'
+    hub = 'hub'
 
 
 class Service(RootModel[str]):
-    root: str = Field(..., pattern="^[a-z][a-z0-9_]*$")
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
+    root: str = Field(..., pattern='^[a-z][a-z0-9_]*$')
 
 
 class Deployment(BaseModel):
-    mode: Literal["single"]
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
+    mode: Literal['single']
     """
     Deployment mode
     """
-    service: str = Field(..., pattern="^[a-z][a-z0-9_]*$")
+    service: str = Field(..., pattern='^[a-z][a-z0-9_]*$')
     """
     Service name for single mode (required if mode=single)
     """
@@ -65,11 +74,14 @@ class Deployment(BaseModel):
 
 
 class Deployment1(BaseModel):
-    mode: Literal["hub"]
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
+    mode: Literal['hub']
     """
     Deployment mode
     """
-    service: str | None = Field(None, pattern="^[a-z][a-z0-9_]*$")
+    service: str | None = Field(None, pattern='^[a-z][a-z0-9_]*$')
     """
     Service name for single mode (required if mode=single)
     """
@@ -80,12 +92,15 @@ class Deployment1(BaseModel):
 
 
 class Mdns(BaseModel):
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
     enabled: bool | None = False
     """
     Enable mDNS service discovery
     """
     service_name: str | None = Field(
-        None, examples=["lumen-clip", "lumen-hub"], pattern="^[a-z][a-z0-9-]*$"
+        None, examples=['lumen-clip', 'lumen-hub'], pattern='^[a-z][a-z0-9-]*$'
     )
     """
     mDNS service name (required if enabled=true)
@@ -93,25 +108,31 @@ class Mdns(BaseModel):
 
 
 class Server(BaseModel):
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
     port: int = Field(..., ge=1024, le=65535)
     """
     gRPC server port
     """
-    host: str | None = Field("0.0.0.0", examples=["0.0.0.0", "127.0.0.1", "[::]"])
+    host: str | None = Field('0.0.0.0', examples=['0.0.0.0', '127.0.0.1', '[::]'])
     """
     Server bind address
     """
     mdns: Mdns | None = None
 
 
-class Import(BaseModel):
+class ImportInfo(BaseModel):
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
     registry_class: str = Field(
         ...,
         examples=[
-            "lumen_clip.service_registry.ClipService",
-            "lumen_face.service_registry.FaceService",
+            'lumen_clip.service_registry.ClipService',
+            'lumen_face.service_registry.FaceService',
         ],
-        pattern="^[a-z_][a-z0-9_.]*\\.[A-Z][a-zA-Z0-9]*$",
+        pattern='^[a-z_][a-z0-9_.]*\\.[A-Z][a-zA-Z0-9]*$',
     )
     """
     Full dotted path to service registry class
@@ -119,10 +140,10 @@ class Import(BaseModel):
     add_to_server: str = Field(
         ...,
         examples=[
-            "lumen_clip.proto.ml_service_pb2_grpc.add_InferenceServicer_to_server",
-            "lumen_face.proto.ml_service_pb2_grpc.add_FaceServicer_to_server",
+            'lumen_clip.proto.ml_service_pb2_grpc.add_InferenceServicer_to_server',
+            'lumen_face.proto.ml_service_pb2_grpc.add_FaceServicer_to_server',
         ],
-        pattern="^[a-z_][a-z0-9_.]*\\.add_[A-Za-z0-9_]+_to_server$",
+        pattern='^[a-z_][a-z0-9_.]*\\.add_[A-Za-z0-9_]+_to_server$',
     )
     """
     Full dotted path to gRPC add_to_server function
@@ -135,7 +156,8 @@ class BackendSettings(BaseModel):
     """
 
     model_config = ConfigDict(
-        extra="forbid",
+        extra='forbid',
+        populate_by_name=True,
     )
     device: str | None = None
     """
@@ -145,14 +167,9 @@ class BackendSettings(BaseModel):
     """
     Maximum batch size for inference.
     """
-    onnx_providers: list[str] | None = None
+    onnx_providers: list[Any] | None = None
     """
-    List of ONNX execution providers. If null, uses ONNX Runtime defaults.
-    """
-    prefer_fp16: bool | None = True
-    """
-    Whether to prefer FP16 model files over FP32 when available.
-    Improves performance on supported hardware.
+    List of ONNX execution providers. Each item can be a string or a tuple of (name, config_dict).
     """
 
 
@@ -161,13 +178,16 @@ class Runtime(Enum):
     Model runtime type
     """
 
-    torch = "torch"
-    onnx = "onnx"
-    rknn = "rknn"
+    torch = 'torch'
+    onnx = 'onnx'
+    rknn = 'rknn'
 
 
 class ModelConfig(BaseModel):
-    model: str = Field(..., examples=["ViT-B-32", "CN-CLIP-ViT-B-16", "MobileCLIP-S2"])
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
+    model: str = Field(..., examples=['ViT-B-32', 'CN-CLIP-ViT-B-16', 'MobileCLIP-S2'])
     """
     Model repository name
     """
@@ -176,29 +196,36 @@ class ModelConfig(BaseModel):
     Model runtime type
     """
     rknn_device: str | None = Field(
-        None, examples=["rk3566", "rk3588"], pattern="^rk\\d+$"
+        None, examples=['rk3566', 'rk3588'], pattern='^rk\\d+$'
     )
     """
     RKNN device identifier (required if runtime=rknn)
     """
-    dataset: str | None = Field(None, examples=["ImageNet_1k", "TreeOfLife-10M"])
+    dataset: str | None = Field(None, examples=['ImageNet_1k', 'TreeOfLife-10M'])
     """
     Dataset name for zero-shot classification (optional)
     """
+    precision: str | None = Field(None, examples=['fp32', 'int8', 'q4fp16'])
+    """
+    Preferred precision for running the model, valid only when runtime is 'onnx' or 'rknn'. The download validator will check the precision field in model_info.json to verify if the preferred precision is available for the current model.
+    """
 
 
 class Services(BaseModel):
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
     enabled: bool
     """
     Whether to load this service
     """
     package: str = Field(
-        ..., examples=["lumen_clip", "lumen_face"], pattern="^[a-z][a-z0-9_]*$"
+        ..., examples=['lumen_clip', 'lumen_face'], pattern='^[a-z][a-z0-9_]*$'
     )
     """
     Python package name
     """
-    import_: Import = Field(..., alias="import")
+    import_info: ImportInfo | None = None
     backend_settings: BackendSettings | None = None
     models: dict[str, ModelConfig]
     """
@@ -212,7 +239,8 @@ class LumenConfig(BaseModel):
     """
 
     model_config = ConfigDict(
-        extra="forbid",
+        extra='forbid',
+        populate_by_name=True,
     )
     metadata: Metadata
     deployment: Deployment | Deployment1

lumen_resources-0.4.1/src/lumen_resources/result_schemas/README.md

@@ -0,0 +1,29 @@
+```bash
+datamodel-codegen \
+  --input src/lumen_resources/schemas/result_schemas \
+  --output src/lumen_resources/result_schemas/ \
+  --use-schema-description \
+  --use-field-description \
+  --target-python-version 3.10 \
+  --use-standard-collections \
+  --use-union-operator \
+  --output-model-type pydantic_v2.BaseModel \
+  --field-constraints \
+  --input-file-type jsonschema \
+  --allow-population-by-field-name
+```
+
+```bash
+datamodel-codegen \
+  --input src/lumen_resources/schemas/config-schema.yaml \
+  --output src/lumen_resources/lumen_config.py \
+  --use-schema-description \
+  --use-field-description \
+  --target-python-version 3.10 \
+  --use-standard-collections \
+  --use-union-operator \
+  --output-model-type pydantic_v2.BaseModel \
+  --field-constraints \
+  --input-file-type jsonschema \
+  --allow-population-by-field-name
+```

{lumen_resources-0.3.2 → lumen_resources-0.4.1}/src/lumen_resources/schemas/config-schema.yaml

@@ -131,6 +131,9 @@ properties:
         patternProperties:
             "^[a-z][a-z0-9_]*$":
                 type: object
+                x-pydantic-config:
+                    validate_by_name: true
+                    validate_by_alias: true
                 required:
                     - enabled
                     - package
@@ -150,7 +153,7 @@ properties:
                             - "lumen_clip"
                             - "lumen_face"
 
-                    import:
+                    import_info:
                         type: object
                         required:
                             - registry_class
@@ -197,15 +200,10 @@ definitions:
                 minimum: 1
                 default: 8
             onnx_providers:
-                type: [array, "null"] # Allow array or null
-                items:
-                    type: string
-                description: "List of ONNX execution providers. If null, uses ONNX Runtime defaults."
+                type: [array, "null"]
+                description: "List of ONNX execution providers. Each item can be a string or a tuple of (name, config_dict)."
                 default: null
-            prefer_fp16:
-                type: boolean
-                description: "Whether to prefer FP16 model files over FP32 when available. Improves performance on supported hardware."
-                default: true
+
         additionalProperties: false
     ModelConfig:
         type: object
@@ -240,6 +238,13 @@ definitions:
                 examples:
                     - "ImageNet_1k"
                     - "TreeOfLife-10M"
+            precision:
+                type: string
+                description: Preferred precision for running the model, valid only when runtime is 'onnx' or 'rknn'. The download validator will check the precision field in model_info.json to verify if the preferred precision is available for the current model.
+                examples:
+                    - "fp32"
+                    - "int8"
+                    - "q4fp16"
 
         # If runtime=rknn, rknn_device is required
         if:

{lumen_resources-0.3.2 → lumen_resources-0.4.1}/src/lumen_resources.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lumen-resources
-Version: 0.3.2
+Version: 0.4.1
 Summary: Unified model resource management for Lumen ML services
 Author-email: EdwinZhanCN <support@lumilio.org>
 License: MIT

{lumen_resources-0.3.2 → lumen_resources-0.4.1}/src/lumen_resources.egg-info/SOURCES.txt

@@ -1,7 +1,6 @@
 .gitignore
 README.md
 pyproject.toml
-uv.lock
 docs/examples/clip_torch_cn.yaml
 docs/examples/hub-service.yaml
 docs/examples/model_info_template.json

lumen_resources-0.3.2/src/lumen_resources/result_schemas/README.md

@@ -1,14 +0,0 @@
-```bash
-datamodel-codegen \
-  --input src/lumen_resources/schemas/result_schemas \
-  --output src/lumen_resources/result_schemas/ \
-  --use-schema-description \
-  --use-field-description \
-  --target-python-version 3.10 \
-  --use-standard-collections \
-  --use-union-operator \
-  --output-model-type pydantic_v2.BaseModel \
-  --field-constraints \
-  --input-file-type jsonschema
-```
-