mycelium-ai 0.5.0__py3-none-any.whl

mycelium/api/generated_sources/worker_schemas/models/save_config_response.py

@@ -0,0 +1,73 @@
+# coding: utf-8
+
+"""
+    Mycelium Worker (Client) API
+
+    Configuration API for Mycelium client workers
+
+    The version of the OpenAPI document: 0.1.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+
+from typing import Optional
+from pydantic import BaseModel, Field, StrictBool, StrictStr, ConfigDict
+
+class SaveConfigResponse(BaseModel):
+    """
+    SaveConfigResponse
+    """
+    message: StrictStr = Field(...)
+    status: StrictStr = Field(...)
+    reloaded: StrictBool = Field(...)
+    reload_error: Optional[StrictStr] = None
+    __properties = ["message", "status", "reloaded", "reload_error"]
+
+    model_config = {"populate_by_name": True, "validate_assignment": True}
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> SaveConfigResponse:
+        """Create an instance of SaveConfigResponse from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self):
+        """Returns the dictionary representation of the model using alias"""
+        _dict = self.model_dump(by_alias=True,
+                          exclude={
+                          },
+                          exclude_none=True)
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> SaveConfigResponse:
+        """Create an instance of SaveConfigResponse from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return SaveConfigResponse.model_validate(obj)
+
+        _obj = SaveConfigResponse.model_validate({
+            "message": obj.get("message"),
+            "status": obj.get("status"),
+            "reloaded": obj.get("reloaded"),
+            "reload_error": obj.get("reload_error")
+        })
+        return _obj
+
+

mycelium/api/generated_sources/worker_schemas/models/worker_clap_section.py

@@ -0,0 +1,75 @@
+# coding: utf-8
+
+"""
+    Mycelium Worker (Client) API
+
+    Configuration API for Mycelium client workers
+
+    The version of the OpenAPI document: 0.1.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+
+from typing import Optional
+from pydantic import BaseModel, Field, StrictInt, StrictStr, ConfigDict
+
+class WorkerClapSection(BaseModel):
+    """
+    WorkerClapSection
+    """
+    model_id: StrictStr = Field(...)
+    target_sr: StrictInt = Field(...)
+    chunk_duration_s: StrictInt = Field(...)
+    num_chunks: StrictInt = Field(...)
+    max_load_duration_s: Optional[StrictInt] = None
+    __properties = ["model_id", "target_sr", "chunk_duration_s", "num_chunks", "max_load_duration_s"]
+
+    model_config = {"populate_by_name": True, "validate_assignment": True}
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> WorkerClapSection:
+        """Create an instance of WorkerClapSection from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self):
+        """Returns the dictionary representation of the model using alias"""
+        _dict = self.model_dump(by_alias=True,
+                          exclude={
+                          },
+                          exclude_none=True)
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> WorkerClapSection:
+        """Create an instance of WorkerClapSection from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return WorkerClapSection.model_validate(obj)
+
+        _obj = WorkerClapSection.model_validate({
+            "model_id": obj.get("model_id"),
+            "target_sr": obj.get("target_sr"),
+            "chunk_duration_s": obj.get("chunk_duration_s"),
+            "num_chunks": obj.get("num_chunks"),
+            "max_load_duration_s": obj.get("max_load_duration_s")
+        })
+        return _obj
+
+

mycelium/api/generated_sources/worker_schemas/models/worker_client_api_section.py

@@ -0,0 +1,69 @@
+# coding: utf-8
+
+"""
+    Mycelium Worker (Client) API
+
+    Configuration API for Mycelium client workers
+
+    The version of the OpenAPI document: 0.1.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+
+
+from pydantic import BaseModel, Field, StrictInt, StrictStr, ConfigDict
+
+class WorkerClientAPISection(BaseModel):
+    """
+    WorkerClientAPISection
+    """
+    host: StrictStr = Field(...)
+    port: StrictInt = Field(...)
+    __properties = ["host", "port"]
+
+    model_config = {"populate_by_name": True, "validate_assignment": True}
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> WorkerClientAPISection:
+        """Create an instance of WorkerClientAPISection from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self):
+        """Returns the dictionary representation of the model using alias"""
+        _dict = self.model_dump(by_alias=True,
+                          exclude={
+                          },
+                          exclude_none=True)
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> WorkerClientAPISection:
+        """Create an instance of WorkerClientAPISection from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return WorkerClientAPISection.model_validate(obj)
+
+        _obj = WorkerClientAPISection.model_validate({
+            "host": obj.get("host"),
+            "port": obj.get("port")
+        })
+        return _obj
+
+

mycelium/api/generated_sources/worker_schemas/models/worker_client_section.py

@@ -0,0 +1,79 @@
+# coding: utf-8
+
+"""
+    Mycelium Worker (Client) API
+
+    Configuration API for Mycelium client workers
+
+    The version of the OpenAPI document: 0.1.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+
+from typing import Union
+from pydantic import BaseModel, Field, StrictFloat, StrictInt, StrictStr, ConfigDict
+
+class WorkerClientSection(BaseModel):
+    """
+    WorkerClientSection
+    """
+    server_host: StrictStr = Field(...)
+    server_port: StrictInt = Field(...)
+    download_queue_size: StrictInt = Field(...)
+    job_queue_size: StrictInt = Field(...)
+    poll_interval: Union[StrictFloat, StrictInt] = Field(...)
+    download_workers: StrictInt = Field(...)
+    gpu_batch_size: StrictInt = Field(...)
+    __properties = ["server_host", "server_port", "download_queue_size", "job_queue_size", "poll_interval", "download_workers", "gpu_batch_size"]
+
+    model_config = {"populate_by_name": True, "validate_assignment": True}
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> WorkerClientSection:
+        """Create an instance of WorkerClientSection from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self):
+        """Returns the dictionary representation of the model using alias"""
+        _dict = self.model_dump(by_alias=True,
+                          exclude={
+                          },
+                          exclude_none=True)
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> WorkerClientSection:
+        """Create an instance of WorkerClientSection from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return WorkerClientSection.model_validate(obj)
+
+        _obj = WorkerClientSection.model_validate({
+            "server_host": obj.get("server_host"),
+            "server_port": obj.get("server_port"),
+            "download_queue_size": obj.get("download_queue_size"),
+            "job_queue_size": obj.get("job_queue_size"),
+            "poll_interval": obj.get("poll_interval"),
+            "download_workers": obj.get("download_workers"),
+            "gpu_batch_size": obj.get("gpu_batch_size")
+        })
+        return _obj
+
+

mycelium/api/generated_sources/worker_schemas/models/worker_config_request.py

@@ -0,0 +1,73 @@
+# coding: utf-8
+
+"""
+    Mycelium Worker (Client) API
+
+    Configuration API for Mycelium client workers
+
+    The version of the OpenAPI document: 0.1.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+
+from typing import Any, Dict
+from pydantic import BaseModel, Field, ConfigDict
+
+class WorkerConfigRequest(BaseModel):
+    """
+    WorkerConfigRequest
+    """
+    client: Dict[str, Any] = Field(...)
+    client_api: Dict[str, Any] = Field(...)
+    clap: Dict[str, Any] = Field(...)
+    logging: Dict[str, Any] = Field(...)
+    __properties = ["client", "client_api", "clap", "logging"]
+
+    model_config = {"populate_by_name": True, "validate_assignment": True}
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> WorkerConfigRequest:
+        """Create an instance of WorkerConfigRequest from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self):
+        """Returns the dictionary representation of the model using alias"""
+        _dict = self.model_dump(by_alias=True,
+                          exclude={
+                          },
+                          exclude_none=True)
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> WorkerConfigRequest:
+        """Create an instance of WorkerConfigRequest from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return WorkerConfigRequest.model_validate(obj)
+
+        _obj = WorkerConfigRequest.model_validate({
+            "client": obj.get("client"),
+            "client_api": obj.get("client_api"),
+            "clap": obj.get("clap"),
+            "logging": obj.get("logging")
+        })
+        return _obj
+
+

mycelium/api/generated_sources/worker_schemas/models/worker_config_response.py

@@ -0,0 +1,89 @@
+# coding: utf-8
+
+"""
+    Mycelium Worker (Client) API
+
+    Configuration API for Mycelium client workers
+
+    The version of the OpenAPI document: 0.1.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+
+
+from pydantic import BaseModel, Field, ConfigDict
+from mycelium.api.generated_sources.worker_schemas.models.worker_clap_section import WorkerClapSection
+from mycelium.api.generated_sources.worker_schemas.models.worker_client_api_section import WorkerClientAPISection
+from mycelium.api.generated_sources.worker_schemas.models.worker_client_section import WorkerClientSection
+from mycelium.api.generated_sources.worker_schemas.models.worker_logging_section import WorkerLoggingSection
+
+class WorkerConfigResponse(BaseModel):
+    """
+    WorkerConfigResponse
+    """
+    client: WorkerClientSection = Field(...)
+    client_api: WorkerClientAPISection = Field(...)
+    clap: WorkerClapSection = Field(...)
+    logging: WorkerLoggingSection = Field(...)
+    __properties = ["client", "client_api", "clap", "logging"]
+
+    model_config = {"populate_by_name": True, "validate_assignment": True}
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> WorkerConfigResponse:
+        """Create an instance of WorkerConfigResponse from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self):
+        """Returns the dictionary representation of the model using alias"""
+        _dict = self.model_dump(by_alias=True,
+                          exclude={
+                          },
+                          exclude_none=True)
+        # override the default output from pydantic by calling `to_dict()` of client
+        if self.client:
+            _dict['client'] = self.client.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of client_api
+        if self.client_api:
+            _dict['client_api'] = self.client_api.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of clap
+        if self.clap:
+            _dict['clap'] = self.clap.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of logging
+        if self.logging:
+            _dict['logging'] = self.logging.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> WorkerConfigResponse:
+        """Create an instance of WorkerConfigResponse from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return WorkerConfigResponse.model_validate(obj)
+
+        _obj = WorkerConfigResponse.model_validate({
+            "client": WorkerClientSection.from_dict(obj.get("client")) if obj.get("client") is not None else None,
+            "client_api": WorkerClientAPISection.from_dict(obj.get("client_api")) if obj.get("client_api") is not None else None,
+            "clap": WorkerClapSection.from_dict(obj.get("clap")) if obj.get("clap") is not None else None,
+            "logging": WorkerLoggingSection.from_dict(obj.get("logging")) if obj.get("logging") is not None else None
+        })
+        return _obj
+
+

mycelium/api/generated_sources/worker_schemas/models/worker_logging_section.py

@@ -0,0 +1,67 @@
+# coding: utf-8
+
+"""
+    Mycelium Worker (Client) API
+
+    Configuration API for Mycelium client workers
+
+    The version of the OpenAPI document: 0.1.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+
+
+from pydantic import BaseModel, Field, StrictStr, ConfigDict
+
+class WorkerLoggingSection(BaseModel):
+    """
+    WorkerLoggingSection
+    """
+    level: StrictStr = Field(...)
+    __properties = ["level"]
+
+    model_config = {"populate_by_name": True, "validate_assignment": True}
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> WorkerLoggingSection:
+        """Create an instance of WorkerLoggingSection from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self):
+        """Returns the dictionary representation of the model using alias"""
+        _dict = self.model_dump(by_alias=True,
+                          exclude={
+                          },
+                          exclude_none=True)
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> WorkerLoggingSection:
+        """Create an instance of WorkerLoggingSection from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return WorkerLoggingSection.model_validate(obj)
+
+        _obj = WorkerLoggingSection.model_validate({
+            "level": obj.get("level")
+        })
+        return _obj
+
+