truefoundry 0.5.0rc4__tar.gz → 0.5.0rc5__tar.gz

{truefoundry-0.5.0rc4 → truefoundry-0.5.0rc5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: truefoundry
-Version: 0.5.0rc4
+Version: 0.5.0rc5
 Summary: Truefoundry CLI
 Author: Abhishek Choudhary
 Author-email: abhishek@truefoundry.com

{truefoundry-0.5.0rc4 → truefoundry-0.5.0rc5}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "truefoundry"
-version = "0.5.0rc4"
+version = "0.5.0rc5"
 description = "Truefoundry CLI"
 authors = ["Abhishek Choudhary <abhishek@truefoundry.com>"]
 readme = "README.md"

{truefoundry-0.5.0rc4 → truefoundry-0.5.0rc5}/truefoundry/ml/autogen/client/models/model_version_manifest.py

@@ -49,7 +49,19 @@ class ModelVersionManifest(BaseModel):
     step: Optional[conint(strict=True, ge=0)] = Field(
         default=0, description="+label=Step"
     )
-    __properties = ["description", "metadata", "type", "source", "framework", "step"]
+    model_schema: Optional[Dict[str, Any]] = Field(
+        default=None,
+        description="+label=Model Schema +usage=Schema of the model +uiType=Hidden",
+    )
+    __properties = [
+        "description",
+        "metadata",
+        "type",
+        "source",
+        "framework",
+        "step",
+        "model_schema",
+    ]
 
     @validator("type")
     def type_validate_enum(cls, value):
@@ -114,6 +126,7 @@ class ModelVersionManifest(BaseModel):
                 if obj.get("framework") is not None
                 else None,
                 "step": obj.get("step") if obj.get("step") is not None else 0,
+                "model_schema": obj.get("model_schema"),
             }
         )
         return _obj

{truefoundry-0.5.0rc4 → truefoundry-0.5.0rc5}/truefoundry/ml/autogen/entities/artifacts.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  artifacts.json
-#   timestamp: 2024-11-15T09:16:35+00:00
+#   timestamp: 2024-11-22T19:31:48+00:00
 
 from __future__ import annotations
 
@@ -534,6 +534,10 @@ class ModelVersion(BaseArtifactVersion):
         None, description="+label=Framework\n+usage=Framework for the model version"
     )
     step: conint(ge=0) = Field(0, description="+label=Step")
+    model_schema: Optional[Dict[str, Any]] = Field(
+        None,
+        description="+label=Model Schema\n+usage=Schema of the model\n+uiType=Hidden",
+    )
 
 
 class AgentApp(BaseModel):

{truefoundry-0.5.0rc4 → truefoundry-0.5.0rc5}/truefoundry/ml/log_types/artifacts/model.py

@@ -96,6 +96,7 @@ class ModelVersion:
         self._deleted = False
         self._description: str = ""
         self._metadata: Dict[str, Any] = {}
+        self._model_schema: Optional[Dict[str, Any]] = None
         self._framework: Optional[ModelFrameworkType] = None
         self._set_mutable_attrs()
 
@@ -138,6 +139,9 @@ class ModelVersion:
         if self._model_version.manifest:
             self._description = self._model_version.manifest.description or ""
             self._metadata = copy.deepcopy(self._model_version.manifest.metadata)
+            self._model_schema = copy.deepcopy(
+                self._model_version.manifest.model_schema
+            )
             if self._model_version.manifest.framework:
                 self._framework = copy.deepcopy(
                     self._model_version.manifest.framework.actual_instance
@@ -150,6 +154,7 @@ class ModelVersion:
             self._framework = _ModelFramework.to_model_framework_type(
                 self._model_version.model_framework
             )
+            self._model_schema = None
 
     def _refetch_model_version(self, reset_mutable_attrs: bool = True):
         _model_version = self._mlfoundry_artifacts_api.get_model_version_get(
@@ -219,6 +224,23 @@ class ModelVersion:
         _validate_artifact_metadata(value)
         self._metadata = copy.deepcopy(value)
 
+    @property
+    def model_schema(self) -> Optional[Dict[str, Any]]:
+        """Get model_schema for the current model"""
+        return self._model_schema
+
+    @model_schema.setter
+    def model_schema(self, value: Optional[Dict[str, Any]]):
+        """set the model_schema for current model"""
+        if not self._model_version.manifest:
+            warnings.warn(
+                message="This model version was created using an older serialization format. model_schema will not be updated",
+                category=DeprecationWarning,
+                stacklevel=2,
+            )
+            return
+        self._model_schema = copy.deepcopy(value)
+
     @property
     def framework(self) -> Optional["ModelFrameworkType"]:
         """Get the framework of the model"""
@@ -416,6 +438,7 @@ class ModelVersion:
         if self._model_version.manifest:
             self._model_version.manifest.description = self.description
             self._model_version.manifest.metadata = self.metadata
+            self._model_version.manifest.model_schema = self.model_schema
             self._model_version.manifest.framework = (
                 Framework.from_dict(self.framework.dict()) if self.framework else None
             )
@@ -449,6 +472,7 @@ def _log_model_version(  # noqa: C901
     step: Optional[int] = 0,
     progress: Optional[bool] = None,
     framework: Optional[Union[str, ModelFramework, "ModelFrameworkType"]] = None,
+    model_schema: Optional[Dict[str, Any]] = None,
 ) -> ModelVersion:
     if (run and mlfoundry_artifacts_api) or (not run and not mlfoundry_artifacts_api):
         raise MlFoundryException(
@@ -562,6 +586,7 @@ def _log_model_version(  # noqa: C901
         source=_source_cls.from_dict(source.dict()),
         framework=Framework.from_dict(_framework.dict()) if _framework else None,
         step=step,
+        model_schema=model_schema,
     )
     artifact_version_response = mlfoundry_artifacts_api.finalize_artifact_version_post(
         finalize_artifact_version_request_dto=FinalizeArtifactVersionRequestDto(

{truefoundry-0.5.0rc4 → truefoundry-0.5.0rc5}/truefoundry/ml/mlfoundry_api.py

@@ -1232,6 +1232,7 @@ class MlFoundry:
         metadata: Optional[Dict[str, Any]] = None,
         progress: Optional[bool] = None,
         framework: Optional[Union[str, ModelFramework, "ModelFrameworkType"]] = None,
+        model_schema: Optional[Dict[str, Any]] = None,
     ) -> ModelVersion:
         """
         Serialize and log a versioned model under the current ml_repo. Each logged model generates a new version
@@ -1391,6 +1392,7 @@ class MlFoundry:
             step=None,
             progress=progress,
             framework=framework,
+            model_schema=model_schema,
         )
         logger.info(f"Logged model successfully with fqn {model_version.fqn!r}")
         return model_version

{truefoundry-0.5.0rc4 → truefoundry-0.5.0rc5}/truefoundry/ml/mlfoundry_run.py

@@ -936,6 +936,7 @@ class MlFoundryRun:
         step: int = 0,
         progress: Optional[bool] = None,
         framework: Optional[Union[str, ModelFramework, "ModelFrameworkType"]] = None,
+        model_schema: Optional[Dict[str, Any]] = None,
     ) -> ModelVersion:
         # TODO (chiragjn): Document mapping of framework to list of valid model save kwargs
         # TODO (chiragjn): Add more examples
@@ -1088,6 +1089,7 @@ class MlFoundryRun:
             step=step,
             progress=progress,
             framework=framework,
+            model_schema=model_schema,
         )
         logger.info(f"Logged model successfully with fqn {model_version.fqn!r}")
         return model_version