lumen-resources 0.3.2__py3-none-any.whl → 0.4.1__py3-none-any.whl

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/result_schemas/README.md

@@ -9,6 +9,21 @@ datamodel-codegen \
   --use-union-operator \
   --output-model-type pydantic_v2.BaseModel \
   --field-constraints \
-  --input-file-type jsonschema
+  --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/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.dist-info → lumen_resources-0.4.1.dist-info}/METADATA

@@ -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.dist-info → lumen_resources-0.4.1.dist-info}/RECORD

@@ -2,27 +2,27 @@ lumen_resources/__init__.py,sha256=EwI0cDEF9eIlWT4eYWwL8MUR8-vaHEzpZ0Kf94Kxn60,2
 lumen_resources/cli.py,sha256=gGGfe0BmqjZX-2fIEEQ-MgElMn21OrabPs7Fcu6aY6A,14204
 lumen_resources/downloader.py,sha256=Gn0or7qvk3BfmKu7VtKklXxVKuBj2lk9b5Nq7Cll-kY,21589
 lumen_resources/exceptions.py,sha256=cUS76Revda0Vt6rPeUNLI2QhluNyLon1GiF5ZM4Kx30,3148
-lumen_resources/lumen_config.py,sha256=9lrsdVcqlwJ1lwrKRfD449FKBKtQ4ut5D26tmwuObdA,5176
+lumen_resources/lumen_config.py,sha256=YqJohnwSjL1rUb9UzFv3fwav4PcU6etIJ96y0zi1XqM,6042
 lumen_resources/lumen_config_validator.py,sha256=6GIBwresO1k7yALaNt5EgQZpcxFVq7k1obRQjl4AEEI,9734
 lumen_resources/model_info.py,sha256=O7LRJnC2OJ5uAuic074hzWThruHhVyMg5Kt7eNAdZQ4,2355
 lumen_resources/model_info_validator.py,sha256=91QsUCRuxZT3wCt7xI3wsLVBmEiVXxJUk8d2MOHwNb0,9294
 lumen_resources/platform.py,sha256=EwqjQMoQY2jK0VP1jstgIW_Lu82Cxn9HDxwdCj1EN8w,9824
-lumen_resources/result_schemas/README.md,sha256=nPbihM4RCJxFbBHyUsRYLljyaRjULuMQgW6K1nWYGy0,384
+lumen_resources/result_schemas/README.md,sha256=jQ_J3WBcZJZbD9kVkbEj6wngDBN_HQ5S-wRiUy_ociI,845
 lumen_resources/result_schemas/__init__.py,sha256=iOxu2V8w5tXrFF5jH-Qn-En7nr2b3ZlcaiswyJCtxCQ,357
 lumen_resources/result_schemas/embedding_v1.py,sha256=XlruWBMKefG7p_f06KS-_iLs8fSwO_wxkfFuBUWgSTk,671
 lumen_resources/result_schemas/face_v1.py,sha256=Vqd-VxzmmPC0tqNtRV9qYx7Xaax5CXxjJhjJ5whiT9w,1314
 lumen_resources/result_schemas/labels_v1.py,sha256=alrU2XKiLSVXGTAfcOZlPbGGKX2s-UrGQ6uYluM-BBY,829
 lumen_resources/result_schemas/ocr_v1.py,sha256=5YFAwBkRJ_dMlyENxosXRYEIyZnDOAYx7JRDVnY7ah4,1259
 lumen_resources/result_schemas/text_generation_v1.py,sha256=DQ6vb5e9tHYzd1Qf-JkDTflzbnyUWOCgleeJnTImt50,2147
-lumen_resources/schemas/config-schema.yaml,sha256=TXw3Gn1q0DcqVHM4jyvSA_sWfmX_GNXe_nXLvRU7_2I,8388
+lumen_resources/schemas/config-schema.yaml,sha256=ts_X2ANVcWpDDeCdHYiQA-7cvEIWxPxWIjMVQFB3sHY,8671
 lumen_resources/schemas/model_info-schema.json,sha256=VkARnNvzdFKee_AROQG9AUU5Q9s2P_dt7Nvm9Mj39aU,5837
 lumen_resources/schemas/result_schemas/embedding_v1.json,sha256=6iZaXbmkP0J5fXGD4Gkrt6AZPvpK6FZaQ754sOXxFrc,841
 lumen_resources/schemas/result_schemas/face_v1.json,sha256=XcnHxwP_KR8lAv6s6npjWQxwyYAQTuBLEKrvlOqH84g,1771
 lumen_resources/schemas/result_schemas/labels_v1.json,sha256=AnOiM0VCuIKrLdrbA73pmu4QD4QaHIDNsANeU3BIOeg,1222
 lumen_resources/schemas/result_schemas/ocr_v1.json,sha256=btVUZjR_RW4CLJjIyEGCITPJFJxMqGSM7QX8OyfVguo,1653
 lumen_resources/schemas/result_schemas/text_generation_v1.json,sha256=f_qwzlfKR_zZMZsn9qQ3U1U3t7HkSk9j7XED-q_s2hc,2876
-lumen_resources-0.3.2.dist-info/METADATA,sha256=16Os1V15jiQlyhm8FucFSQ55YAx682maKWK-BhEUwGw,4002
-lumen_resources-0.3.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-lumen_resources-0.3.2.dist-info/entry_points.txt,sha256=fLCrIB9BxyIDAbJVqDGW4QyvLPlvL53WI-6lkTZ3h2M,61
-lumen_resources-0.3.2.dist-info/top_level.txt,sha256=XgLNoNrF2RIpI2sYIpjLuUREYRVHW13ElHoCnFYHjAQ,16
-lumen_resources-0.3.2.dist-info/RECORD,,
+lumen_resources-0.4.1.dist-info/METADATA,sha256=bR2AusEUZ-YGSMpEazIe79PFhXgIXhuK5UZI9MOOKxE,4002
+lumen_resources-0.4.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+lumen_resources-0.4.1.dist-info/entry_points.txt,sha256=fLCrIB9BxyIDAbJVqDGW4QyvLPlvL53WI-6lkTZ3h2M,61
+lumen_resources-0.4.1.dist-info/top_level.txt,sha256=XgLNoNrF2RIpI2sYIpjLuUREYRVHW13ElHoCnFYHjAQ,16
+lumen_resources-0.4.1.dist-info/RECORD,,