truefoundry 0.5.10rc1__py3-none-any.whl → 0.5.11rc1__py3-none-any.whl

truefoundry/ml/{autogen → _autogen}/client/models/transformers_framework.py

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -18,30 +18,26 @@ import pprint
 import re  # noqa: F401
 from typing import Optional
 
-from truefoundry.ml.autogen.client.models.library_name import LibraryName
+from truefoundry.ml._autogen.client.models.library_name import LibraryName
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 
 class TransformersFramework(BaseModel):
     """
-    +docs=Transformers framework for the model version +label=Transformers +value=transformers  # noqa: E501
+    TransformersFramework
     """
 
-    type: StrictStr = Field(
-        default=...,
-        description="+label=Type +usage=Type of the framework +value=transformers",
-    )
+    type: StrictStr = Field(default=..., description="Type of the framework")
     library_name: Optional[LibraryName] = Field(
-        default=None,
-        description="+label=Library Name +usage=Name of the library for the framework",
+        default=None, description="Name of the library for the framework"
     )
     pipeline_tag: Optional[StrictStr] = Field(
         default=None,
-        description="+label=Pipeline Tag +usage=The `pipeline()` task this model can be used with e.g. `text-generation`. See [huggingface docs](https://huggingface.co/docs/transformers/main/en/main_classes/pipelines#transformers.pipeline.task) for all possible values +docs=Pipeline tag for the framework",
+        description="The `pipeline()` task this model can be used with e.g. `text-generation`. See [huggingface docs](https://huggingface.co/docs/transformers/main/en/main_classes/pipelines#transformers.pipeline.task) for all possible values",
     )
     base_model: Optional[StrictStr] = Field(
         default=None,
-        description="+label=Base Model +usage=Base model Id. If this is a finetuned model, this points to the base model used for finetuning +docs=Base model Id. If this is a finetuned model, this points to the base model used for finetuning",
+        description="Base model Id from HuggingFace Hub. If this is a finetuned model, this points to the base model id used for finetuning.",
     )
     __properties = ["type", "library_name", "pipeline_tag", "base_model"]
 

truefoundry/ml/{autogen → _autogen}/client/models/trigger_job_run_config_request_dto.py

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -18,7 +18,7 @@ import pprint
 import re  # noqa: F401
 from typing import Dict, Optional
 
-from truefoundry.ml.autogen.client.models.command import Command
+from truefoundry.ml._autogen.client.models.command import Command
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr
 
 

truefoundry/ml/{autogen → _autogen}/client/models/trigger_job_run_config_response_dto.py

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

truefoundry/ml/{autogen → _autogen}/client/models/true_foundry_managed_source.py

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -18,22 +18,16 @@ import pprint
 import re  # noqa: F401
 from typing import Optional
 
-from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, constr, validator
 
 
 class TrueFoundryManagedSource(BaseModel):
     """
-    +usage=Source for the entity +label=Upload  # noqa: E501
+    TrueFoundryManagedSource
     """
 
-    type: StrictStr = Field(
-        default=...,
-        description="+label=Type +usage=Type of the source +value=truefoundry",
-    )
-    uri: Optional[StrictStr] = Field(
-        default=None,
-        description='+label=URI +uiType=ModelUploadInput +uiProps={"hideClear":true}',
-    )
+    type: StrictStr = Field(default=..., description="Type of the source")
+    uri: Optional[constr(strict=True)] = None
     __properties = ["type", "uri"]
 
     @validator("type")
@@ -43,6 +37,18 @@ class TrueFoundryManagedSource(BaseModel):
             raise ValueError("must be one of enum values ('truefoundry')")
         return value
 
+    @validator("uri")
+    def uri_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
+        if value is None:
+            return value
+
+        if not re.match(r"^(s3|gs|wasbs):\/\/.*", value):
+            raise ValueError(
+                r"must validate the regular expression /^(s3|gs|wasbs):\/\/.*/"
+            )
+        return value
+
     class Config:
         """Pydantic configuration"""
 

truefoundry/ml/{autogen → _autogen}/client/models/update_artifact_version_request_dto.py

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -18,9 +18,7 @@ import pprint
 import re  # noqa: F401
 from typing import Any, Dict, Optional
 
-from truefoundry.ml.autogen.client.models.artifact_version_manifest import (
-    ArtifactVersionManifest,
-)
+from truefoundry.ml._autogen.client.models.manifest2 import Manifest2
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr
 
 
@@ -32,7 +30,7 @@ class UpdateArtifactVersionRequestDto(BaseModel):
     id: StrictStr = Field(...)
     description: Optional[StrictStr] = None
     artifact_metadata: Optional[Dict[str, Any]] = None
-    manifest: Optional[ArtifactVersionManifest] = None
+    manifest: Optional[Manifest2] = None
     __properties = ["id", "description", "artifact_metadata", "manifest"]
 
     class Config:
@@ -76,7 +74,7 @@ class UpdateArtifactVersionRequestDto(BaseModel):
                 "id": obj.get("id"),
                 "description": obj.get("description"),
                 "artifact_metadata": obj.get("artifact_metadata"),
-                "manifest": ArtifactVersionManifest.from_dict(obj.get("manifest"))
+                "manifest": Manifest2.from_dict(obj.get("manifest"))
                 if obj.get("manifest") is not None
                 else None,
             }

truefoundry/ml/{autogen → _autogen}/client/models/update_dataset_request_dto.py

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -18,6 +18,9 @@ import pprint
 import re  # noqa: F401
 from typing import Any, Dict, Optional
 
+from truefoundry.ml._autogen.client.models.data_directory_manifest import (
+    DataDirectoryManifest,
+)
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr
 
 
@@ -29,7 +32,8 @@ class UpdateDatasetRequestDto(BaseModel):
     fqn: StrictStr = Field(...)
     description: Optional[StrictStr] = None
     dataset_metadata: Optional[Dict[str, Any]] = None
-    __properties = ["fqn", "description", "dataset_metadata"]
+    manifest: Optional[DataDirectoryManifest] = None
+    __properties = ["fqn", "description", "dataset_metadata", "manifest"]
 
     class Config:
         """Pydantic configuration"""
@@ -53,6 +57,9 @@ class UpdateDatasetRequestDto(BaseModel):
     def to_dict(self):
         """Returns the dictionary representation of the model using alias"""
         _dict = self.dict(by_alias=True, exclude={}, exclude_none=True)
+        # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of manifest
+        if self.manifest:
+            _dict["manifest"] = self.manifest.to_dict()
         return _dict
 
     @classmethod
@@ -69,6 +76,9 @@ class UpdateDatasetRequestDto(BaseModel):
                 "fqn": obj.get("fqn"),
                 "description": obj.get("description"),
                 "dataset_metadata": obj.get("dataset_metadata"),
+                "manifest": DataDirectoryManifest.from_dict(obj.get("manifest"))
+                if obj.get("manifest") is not None
+                else None,
             }
         )
         return _obj

truefoundry/ml/{autogen → _autogen}/client/models/update_experiment_request_dto.py

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

truefoundry/ml/{autogen → _autogen}/client/models/update_model_version_request_dto.py

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -18,9 +18,7 @@ import pprint
 import re  # noqa: F401
 from typing import Any, Dict, Optional
 
-from truefoundry.ml.autogen.client.models.model_version_manifest import (
-    ModelVersionManifest,
-)
+from truefoundry.ml._autogen.client.models.model_manifest import ModelManifest
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr
 
 
@@ -33,7 +31,7 @@ class UpdateModelVersionRequestDto(BaseModel):
     description: Optional[StrictStr] = None
     artifact_metadata: Optional[Dict[str, Any]] = None
     model_framework: Optional[StrictStr] = None
-    manifest: Optional[ModelVersionManifest] = None
+    manifest: Optional[ModelManifest] = None
     __properties = [
         "id",
         "description",
@@ -84,7 +82,7 @@ class UpdateModelVersionRequestDto(BaseModel):
                 "description": obj.get("description"),
                 "artifact_metadata": obj.get("artifact_metadata"),
                 "model_framework": obj.get("model_framework"),
-                "manifest": ModelVersionManifest.from_dict(obj.get("manifest"))
+                "manifest": ModelManifest.from_dict(obj.get("manifest"))
                 if obj.get("manifest") is not None
                 else None,
             }

truefoundry/ml/{autogen → _autogen}/client/models/update_run_request_dto.py

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

truefoundry/ml/{autogen → _autogen}/client/models/update_run_response_dto.py

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -17,7 +17,7 @@ import json
 import pprint
 import re  # noqa: F401
 
-from truefoundry.ml.autogen.client.models.run_info_dto import RunInfoDto
+from truefoundry.ml._autogen.client.models.run_info_dto import RunInfoDto
 from truefoundry.pydantic_v1 import BaseModel, Field
 
 

truefoundry/ml/{autogen → _autogen}/client/models/url.py

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -18,7 +18,7 @@ import pprint
 import re  # noqa: F401
 from typing import TYPE_CHECKING, Any, List, Optional, Union
 
-from truefoundry.ml.autogen.client.models.blob_storage_reference import (
+from truefoundry.ml._autogen.client.models.blob_storage_reference import (
     BlobStorageReference,
 )
 from truefoundry.pydantic_v1 import (
@@ -34,7 +34,7 @@ URL_ANY_OF_SCHEMAS = ["BlobStorageReference", "str"]
 
 class Url(BaseModel):
     """
-    +label=URL for the image +usage=The URL for the image, must be a non-empty valid HTTPS URL or a data URL
+    Image URL linking to the image
     """
 
     # data type: str

truefoundry/ml/{autogen → _autogen}/client/models/user_message.py

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -18,19 +18,19 @@ import pprint
 import re  # noqa: F401
 from typing import Optional
 
-from truefoundry.ml.autogen.client.models.content2 import Content2
+from truefoundry.ml._autogen.client.models.content2 import Content2
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 
 class UserMessage(BaseModel):
     """
-    +usage=User message +docs=Defines the structure of a user message, including role and content +label=User Message  # noqa: E501
+    User message turn  # noqa: E501
     """
 
-    role: StrictStr = Field(...)
+    role: StrictStr = Field(default=..., description="Role of the message")
     content: Content2 = Field(...)
     name: Optional[StrictStr] = Field(
-        default=None, description="+label=Name +usage=Name of the user message"
+        default=None, description="Name of the user this message is from"
     )
     __properties = ["role", "content", "name"]
 

truefoundry/ml/{autogen → _autogen}/client/models/validate_external_storage_root_request_dto.py

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

truefoundry/ml/{autogen → _autogen}/client/models/validate_external_storage_root_response_dto.py

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

truefoundry/ml/{autogen → _autogen}/client/models/validation_error.py

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -17,7 +17,7 @@ import json
 import pprint
 import re  # noqa: F401
 
-from truefoundry.ml.autogen.client.models.validation_error_loc_inner import (
+from truefoundry.ml._autogen.client.models.validation_error_loc_inner import (
     ValidationErrorLocInner,
 )
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, conlist

truefoundry/ml/{autogen → _autogen}/client/models/validation_error_loc_inner.py

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

truefoundry/ml/{autogen → _autogen}/client/models/xg_boost_framework.py

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -18,10 +18,10 @@ import pprint
 import re  # noqa: F401
 from typing import Optional
 
-from truefoundry.ml.autogen.client.models.xg_boost_model_schema import (
+from truefoundry.ml._autogen.client.models.xg_boost_model_schema import (
     XGBoostModelSchema,
 )
-from truefoundry.ml.autogen.client.models.xg_boost_serialization_format import (
+from truefoundry.ml._autogen.client.models.xg_boost_serialization_format import (
     XGBoostSerializationFormat,
 )
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
@@ -29,17 +29,14 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 class XGBoostFramework(BaseModel):
     """
-    +docs=XGBoost framework for the model version +label=XGBoost  # noqa: E501
+    XGBoostFramework
     """
 
-    type: StrictStr = Field(
-        default=...,
-        description="+label=Type +usage=Type of the framework +value=xgboost",
-    )
+    type: StrictStr = Field(default=..., description="Type of the framework")
     serialization_format: Optional[XGBoostSerializationFormat] = None
     model_filepath: Optional[StrictStr] = Field(
         default=None,
-        description="+label=Model file path +usage=Relative path to the model file",
+        description="Relative path to the model file in the model version contents",
     )
     model_schema: Optional[XGBoostModelSchema] = None
     __properties = ["type", "serialization_format", "model_filepath", "model_schema"]

truefoundry/ml/{autogen → _autogen}/client/models/xg_boost_model_schema.py

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -23,18 +23,17 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, conlist, valida
 
 class XGBoostModelSchema(BaseModel):
     """
-    +label=XGBoost Model Schema  # noqa: E501
+    Schema of the XGBoost model  # noqa: E501
     """
 
     infer_method_name: StrictStr = Field(
-        default=...,
-        description="+label=Inference Method Name +usage=Name of the method used for inference",
+        default=..., description="Name of the method used for inference"
     )
     inputs: conlist(Dict[str, Any]) = Field(
-        default=..., description="+label= Input Schema +usage=Schema of the input"
+        default=..., description="Schema of the input"
     )
     outputs: conlist(Dict[str, Any]) = Field(
-        default=..., description="+label= Output Schema +usage=Schema of the output"
+        default=..., description="Schema of the output"
     )
     __properties = ["infer_method_name", "inputs", "outputs"]
 

truefoundry/ml/{autogen → _autogen}/client/models/xg_boost_serialization_format.py

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -19,7 +19,7 @@ from aenum import Enum
 
 class XGBoostSerializationFormat(str, Enum):
     """
-    +label=Serialization format +usage=Serialization format used for XGBoost models
+    Serialization format used to save the xgboost model
     """
 
     """

truefoundry/ml/{autogen → _autogen}/client/rest.py

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -19,7 +19,7 @@ import ssl
 
 import urllib3
 
-from truefoundry.ml.autogen.client.exceptions import (
+from truefoundry.ml._autogen.client.exceptions import (
     ApiException,
     ApiValueError,
     BadRequestException,