truefoundry 0.5.5rc1__py3-none-any.whl → 0.5.6rc1__py3-none-any.whl

truefoundry/ml/autogen/client/models/chat_prompt.py

@@ -16,7 +16,7 @@ from __future__ import annotations
 import json
 import pprint
 import re  # noqa: F401
-from typing import Dict, Optional
+from typing import Any, Dict, Optional
 
 from truefoundry.ml.autogen.client.models.chat_prompt_messages_inner import (
     ChatPromptMessagesInner,
@@ -37,18 +37,24 @@ class ChatPrompt(BaseModel):
     Chat Prompt artifact.  # noqa: E501
     """
 
-    name: constr(strict=True) = Field(
+    description: Optional[constr(strict=True, max_length=512)] = Field(
+        default=None, description="+sort=2 +label=Description"
+    )
+    metadata: Dict[str, Any] = Field(
         default=...,
-        description="+label=Prompt Name +icon=fa-desktop:#326ce5 +message=Alphanumeric word, may contain '-' with a maximum length of 256 characters +sort=1000 +usage=Name of the Prompt. This uniquely identifies it in the workspace. +docs=The unique name for the prompt, consisting of alphanumeric characters and dashes, max length 100.",
+        description="+label=Metadata +docs=Key value pairs to store additional metadata +usage=Key value pairs to store additional metadata +uiType=JsonInput",
     )
-    description: Optional[constr(strict=True, max_length=512, min_length=1)] = Field(
+    version_alias: Optional[constr(strict=True, max_length=128)] = Field(
         default=None,
-        description="+label=Description +icon=fa-desktop:#326ce5 +message=Description in a maximum of 512 characters +sort=2000 +usage=Description of the prompt +docs=An optional description for the prompt, with a maximum length of 512 characters. +optional",
+        description="+label=Version Alias +usage=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc) +docs=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc) +message=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc)",
     )
-    model_configuration: ModelConfiguration = Field(...)
-    metadata: Dict[str, StrictStr] = Field(
-        default=...,
-        description="+label=Metadata +icon=fa-info-circle:#326ce5 +message=Additional metadata for the prompt +sort=7000 +usage=Define additional metadata for the prompt +docs=Key-value pairs to store additional metadata related to the prompt.",
+    name: Optional[constr(strict=True, max_length=256)] = Field(
+        default=None,
+        description="+sort=1 +label=Name +message=The name should start with lowercase alphabets  and can contain alphanumeric and can include '-' in between",
+    )
+    ml_repo: Optional[constr(strict=True)] = Field(
+        default=None,
+        description="+label=ML Repo +usage=name of the ML Repo +docs=name of the ML Repo +uiType=Hidden",
     )
     type: Optional[StrictStr] = "chat_prompt"
     messages: conlist(ChatPromptMessagesInner) = Field(
@@ -59,22 +65,52 @@ class ChatPrompt(BaseModel):
         default=None,
         description="+label=Variables +usage=Variables for the chat completion messages to be used in the prompt messages +sort=5000",
     )
+    model_configuration: ModelConfiguration = Field(...)
     __properties = [
-        "name",
         "description",
-        "model_configuration",
         "metadata",
+        "version_alias",
+        "name",
+        "ml_repo",
         "type",
         "messages",
         "variables",
+        "model_configuration",
     ]
 
+    @validator("version_alias")
+    def version_alias_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
+        if value is None:
+            return value
+
+        if not re.match(r"^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$", value):
+            raise ValueError(
+                r"must validate the regular expression /^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$/"
+            )
+        return value
+
     @validator("name")
     def name_validate_regular_expression(cls, value):
         """Validates the regular expression"""
-        if not re.match(r"^[a-zA-Z][a-zA-Z0-9-_]{0,254}[a-zA-Z0-9]$", value):
+        if value is None:
+            return value
+
+        if not re.match(r"^[A-Za-z0-9_\-]+$", value):
+            raise ValueError(
+                r"must validate the regular expression /^[A-Za-z0-9_\-]+$/"
+            )
+        return value
+
+    @validator("ml_repo")
+    def ml_repo_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
+        if value is None:
+            return value
+
+        if not re.match(r"^[a-zA-Z][a-zA-Z0-9\-]{1,98}[a-zA-Z0-9]$", value):
             raise ValueError(
-                r"must validate the regular expression /^[a-zA-Z][a-zA-Z0-9-_]{0,254}[a-zA-Z0-9]$/"
+                r"must validate the regular expression /^[a-zA-Z][a-zA-Z0-9\-]{1,98}[a-zA-Z0-9]$/"
             )
         return value
 
@@ -110,9 +146,6 @@ class ChatPrompt(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 model_configuration
-        if self.model_configuration:
-            _dict["model_configuration"] = self.model_configuration.to_dict()
         # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of each item in messages (list)
         _items = []
         if self.messages:
@@ -120,6 +153,9 @@ class ChatPrompt(BaseModel):
                 if _item:
                     _items.append(_item.to_dict())
             _dict["messages"] = _items
+        # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of model_configuration
+        if self.model_configuration:
+            _dict["model_configuration"] = self.model_configuration.to_dict()
         return _dict
 
     @classmethod
@@ -133,14 +169,11 @@ class ChatPrompt(BaseModel):
 
         _obj = ChatPrompt.parse_obj(
             {
-                "name": obj.get("name"),
                 "description": obj.get("description"),
-                "model_configuration": ModelConfiguration.from_dict(
-                    obj.get("model_configuration")
-                )
-                if obj.get("model_configuration") is not None
-                else None,
                 "metadata": obj.get("metadata"),
+                "version_alias": obj.get("version_alias"),
+                "name": obj.get("name"),
+                "ml_repo": obj.get("ml_repo"),
                 "type": obj.get("type")
                 if obj.get("type") is not None
                 else "chat_prompt",
@@ -151,6 +184,11 @@ class ChatPrompt(BaseModel):
                 if obj.get("messages") is not None
                 else None,
                 "variables": obj.get("variables"),
+                "model_configuration": ModelConfiguration.from_dict(
+                    obj.get("model_configuration")
+                )
+                if obj.get("model_configuration") is not None
+                else None,
             }
         )
         return _obj

truefoundry/ml/autogen/client/models/{external_artifact_source.py → external_blob_storage_source.py}

@@ -20,17 +20,16 @@ import re  # noqa: F401
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 
-class ExternalArtifactSource(BaseModel):
+class ExternalBlobStorageSource(BaseModel):
     """
-    +label=External  # noqa: E501
+    +label=External Blob Storage  # noqa: E501
     """
 
     type: StrictStr = Field(
-        default=...,
-        description="+label=Type +usage=Type of the source +value=external +type=External",
+        default=..., description="+label=Type +usage=Type of the source +value=external"
     )
     uri: StrictStr = Field(
-        default=..., description="+label=URI +usage=URI of the Artifact source"
+        default=..., description="+label=URI +usage=URI of the Blob Storage source"
     )
     __properties = ["type", "uri"]
 
@@ -56,8 +55,8 @@ class ExternalArtifactSource(BaseModel):
         return json.dumps(self.to_dict())
 
     @classmethod
-    def from_json(cls, json_str: str) -> ExternalArtifactSource:
-        """Create an instance of ExternalArtifactSource from a JSON string"""
+    def from_json(cls, json_str: str) -> ExternalBlobStorageSource:
+        """Create an instance of ExternalBlobStorageSource from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self):
@@ -66,15 +65,15 @@ class ExternalArtifactSource(BaseModel):
         return _dict
 
     @classmethod
-    def from_dict(cls, obj: dict) -> ExternalArtifactSource:
-        """Create an instance of ExternalArtifactSource from a dict"""
+    def from_dict(cls, obj: dict) -> ExternalBlobStorageSource:
+        """Create an instance of ExternalBlobStorageSource from a dict"""
         if obj is None:
             return None
 
         if not isinstance(obj, dict):
-            return ExternalArtifactSource.parse_obj(obj)
+            return ExternalBlobStorageSource.parse_obj(obj)
 
-        _obj = ExternalArtifactSource.parse_obj(
+        _obj = ExternalBlobStorageSource.parse_obj(
             {"type": obj.get("type"), "uri": obj.get("uri")}
         )
         return _obj

truefoundry/ml/autogen/client/models/model_version_manifest.py

@@ -39,16 +39,23 @@ class ModelVersionManifest(BaseModel):
     """
 
     description: Optional[constr(strict=True, max_length=512)] = Field(
-        default=None,
-        description="+label=Description +usage=Description of the artifact or model version +docs=Description of the artifact or model version",
+        default=None, description="+sort=2 +label=Description"
     )
     metadata: Dict[str, Any] = Field(
         default=...,
-        description="+label=Metadata +docs=Metadata for the artifact or model version +usage=Metadata for the artifact or model version +uiType=JsonInput",
+        description="+label=Metadata +docs=Key value pairs to store additional metadata +usage=Key value pairs to store additional metadata +uiType=JsonInput",
     )
     version_alias: Optional[constr(strict=True, max_length=128)] = Field(
         default=None,
-        description="+label=Version Alias +usage=The version alias for artifact or model version which should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v-prod, v-dev, etc) +docs=The version alias for artifact or model version which should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v-prod, v-dev, etc) +message=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v-prod, v-dev, etc)",
+        description="+label=Version Alias +usage=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc) +docs=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc) +message=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc)",
+    )
+    name: Optional[constr(strict=True, max_length=256)] = Field(
+        default=None,
+        description="+sort=1 +label=Name +message=The name should start with lowercase alphabets  and can contain alphanumeric and can include '-' in between",
+    )
+    ml_repo: Optional[constr(strict=True)] = Field(
+        default=None,
+        description="+label=ML Repo +usage=name of the ML Repo +docs=name of the ML Repo +uiType=Hidden",
     )
     type: Optional[StrictStr] = "model-version"
     source: Source1 = Field(...)
@@ -61,6 +68,8 @@ class ModelVersionManifest(BaseModel):
         "description",
         "metadata",
         "version_alias",
+        "name",
+        "ml_repo",
         "type",
         "source",
         "framework",
@@ -80,6 +89,30 @@ class ModelVersionManifest(BaseModel):
             )
         return value
 
+    @validator("name")
+    def name_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
+        if value is None:
+            return value
+
+        if not re.match(r"^[A-Za-z0-9_\-]+$", value):
+            raise ValueError(
+                r"must validate the regular expression /^[A-Za-z0-9_\-]+$/"
+            )
+        return value
+
+    @validator("ml_repo")
+    def ml_repo_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
+        if value is None:
+            return value
+
+        if not re.match(r"^[a-zA-Z][a-zA-Z0-9\-]{1,98}[a-zA-Z0-9]$", value):
+            raise ValueError(
+                r"must validate the regular expression /^[a-zA-Z][a-zA-Z0-9\-]{1,98}[a-zA-Z0-9]$/"
+            )
+        return value
+
     @validator("type")
     def type_validate_enum(cls, value):
         """Validates the enum"""
@@ -137,6 +170,8 @@ class ModelVersionManifest(BaseModel):
                 "description": obj.get("description"),
                 "metadata": obj.get("metadata"),
                 "version_alias": obj.get("version_alias"),
+                "name": obj.get("name"),
+                "ml_repo": obj.get("ml_repo"),
                 "type": obj.get("type")
                 if obj.get("type") is not None
                 else "model-version",

truefoundry/ml/autogen/client/models/source.py

@@ -18,11 +18,11 @@ import pprint
 import re  # noqa: F401
 from typing import TYPE_CHECKING, Any, List, Optional, Union
 
-from truefoundry.ml.autogen.client.models.external_artifact_source import (
-    ExternalArtifactSource,
+from truefoundry.ml.autogen.client.models.external_blob_storage_source import (
+    ExternalBlobStorageSource,
 )
-from truefoundry.ml.autogen.client.models.true_foundry_artifact_source import (
-    TrueFoundryArtifactSource,
+from truefoundry.ml.autogen.client.models.true_foundry_managed_source import (
+    TrueFoundryManagedSource,
 )
 from truefoundry.pydantic_v1 import (
     BaseModel,
@@ -31,7 +31,7 @@ from truefoundry.pydantic_v1 import (
     validator,
 )
 
-SOURCE_ANY_OF_SCHEMAS = ["ExternalArtifactSource", "TrueFoundryArtifactSource"]
+SOURCE_ANY_OF_SCHEMAS = ["ExternalBlobStorageSource", "TrueFoundryManagedSource"]
 
 
 class Source(BaseModel):
@@ -39,12 +39,12 @@ class Source(BaseModel):
     +label=Artifact Source +usage=Source for the Artifact version +uiType=Group
     """
 
-    # data type: TrueFoundryArtifactSource
-    anyof_schema_1_validator: Optional[TrueFoundryArtifactSource] = None
-    # data type: ExternalArtifactSource
-    anyof_schema_2_validator: Optional[ExternalArtifactSource] = None
+    # data type: TrueFoundryManagedSource
+    anyof_schema_1_validator: Optional[TrueFoundryManagedSource] = None
+    # data type: ExternalBlobStorageSource
+    anyof_schema_2_validator: Optional[ExternalBlobStorageSource] = None
     if TYPE_CHECKING:
-        actual_instance: Union[ExternalArtifactSource, TrueFoundryArtifactSource]
+        actual_instance: Union[ExternalBlobStorageSource, TrueFoundryManagedSource]
     else:
         actual_instance: Any
     any_of_schemas: List[str] = Field(SOURCE_ANY_OF_SCHEMAS, const=True)
@@ -70,18 +70,18 @@ class Source(BaseModel):
     def actual_instance_must_validate_anyof(cls, v):
         instance = Source.construct()
         error_messages = []
-        # validate data type: TrueFoundryArtifactSource
-        if not isinstance(v, TrueFoundryArtifactSource):
+        # validate data type: TrueFoundryManagedSource
+        if not isinstance(v, TrueFoundryManagedSource):
             error_messages.append(
-                f"Error! Input type `{type(v)}` is not `TrueFoundryArtifactSource`"
+                f"Error! Input type `{type(v)}` is not `TrueFoundryManagedSource`"
             )
         else:
             return v
 
-        # validate data type: ExternalArtifactSource
-        if not isinstance(v, ExternalArtifactSource):
+        # validate data type: ExternalBlobStorageSource
+        if not isinstance(v, ExternalBlobStorageSource):
             error_messages.append(
-                f"Error! Input type `{type(v)}` is not `ExternalArtifactSource`"
+                f"Error! Input type `{type(v)}` is not `ExternalBlobStorageSource`"
             )
         else:
             return v
@@ -89,7 +89,7 @@ class Source(BaseModel):
         if error_messages:
             # no match
             raise ValueError(
-                "No match found when setting the actual_instance in Source with anyOf schemas: ExternalArtifactSource, TrueFoundryArtifactSource. Details: "
+                "No match found when setting the actual_instance in Source with anyOf schemas: ExternalBlobStorageSource, TrueFoundryManagedSource. Details: "
                 + ", ".join(error_messages)
             )
         else:
@@ -104,15 +104,15 @@ class Source(BaseModel):
         """Returns the object represented by the json string"""
         instance = Source.construct()
         error_messages = []
-        # anyof_schema_1_validator: Optional[TrueFoundryArtifactSource] = None
+        # anyof_schema_1_validator: Optional[TrueFoundryManagedSource] = None
         try:
-            instance.actual_instance = TrueFoundryArtifactSource.from_json(json_str)
+            instance.actual_instance = TrueFoundryManagedSource.from_json(json_str)
             return instance
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
-        # anyof_schema_2_validator: Optional[ExternalArtifactSource] = None
+        # anyof_schema_2_validator: Optional[ExternalBlobStorageSource] = None
         try:
-            instance.actual_instance = ExternalArtifactSource.from_json(json_str)
+            instance.actual_instance = ExternalBlobStorageSource.from_json(json_str)
             return instance
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
@@ -120,7 +120,7 @@ class Source(BaseModel):
         if error_messages:
             # no match
             raise ValueError(
-                "No match found when deserializing the JSON string into Source with anyOf schemas: ExternalArtifactSource, TrueFoundryArtifactSource. Details: "
+                "No match found when deserializing the JSON string into Source with anyOf schemas: ExternalBlobStorageSource, TrueFoundryManagedSource. Details: "
                 + ", ".join(error_messages)
             )
         else:

truefoundry/ml/autogen/client/models/source1.py

@@ -18,11 +18,11 @@ import pprint
 import re  # noqa: F401
 from typing import TYPE_CHECKING, Any, List, Optional, Union
 
-from truefoundry.ml.autogen.client.models.external_artifact_source import (
-    ExternalArtifactSource,
+from truefoundry.ml.autogen.client.models.external_blob_storage_source import (
+    ExternalBlobStorageSource,
 )
-from truefoundry.ml.autogen.client.models.true_foundry_artifact_source import (
-    TrueFoundryArtifactSource,
+from truefoundry.ml.autogen.client.models.true_foundry_managed_source import (
+    TrueFoundryManagedSource,
 )
 from truefoundry.pydantic_v1 import (
     BaseModel,
@@ -31,7 +31,7 @@ from truefoundry.pydantic_v1 import (
     validator,
 )
 
-SOURCE1_ANY_OF_SCHEMAS = ["ExternalArtifactSource", "TrueFoundryArtifactSource"]
+SOURCE1_ANY_OF_SCHEMAS = ["ExternalBlobStorageSource", "TrueFoundryManagedSource"]
 
 
 class Source1(BaseModel):
@@ -39,12 +39,12 @@ class Source1(BaseModel):
     +label=Model Source +usage=Source for the model version +uiType=Group
     """
 
-    # data type: TrueFoundryArtifactSource
-    anyof_schema_1_validator: Optional[TrueFoundryArtifactSource] = None
-    # data type: ExternalArtifactSource
-    anyof_schema_2_validator: Optional[ExternalArtifactSource] = None
+    # data type: TrueFoundryManagedSource
+    anyof_schema_1_validator: Optional[TrueFoundryManagedSource] = None
+    # data type: ExternalBlobStorageSource
+    anyof_schema_2_validator: Optional[ExternalBlobStorageSource] = None
     if TYPE_CHECKING:
-        actual_instance: Union[ExternalArtifactSource, TrueFoundryArtifactSource]
+        actual_instance: Union[ExternalBlobStorageSource, TrueFoundryManagedSource]
     else:
         actual_instance: Any
     any_of_schemas: List[str] = Field(SOURCE1_ANY_OF_SCHEMAS, const=True)
@@ -70,18 +70,18 @@ class Source1(BaseModel):
     def actual_instance_must_validate_anyof(cls, v):
         instance = Source1.construct()
         error_messages = []
-        # validate data type: TrueFoundryArtifactSource
-        if not isinstance(v, TrueFoundryArtifactSource):
+        # validate data type: TrueFoundryManagedSource
+        if not isinstance(v, TrueFoundryManagedSource):
             error_messages.append(
-                f"Error! Input type `{type(v)}` is not `TrueFoundryArtifactSource`"
+                f"Error! Input type `{type(v)}` is not `TrueFoundryManagedSource`"
             )
         else:
             return v
 
-        # validate data type: ExternalArtifactSource
-        if not isinstance(v, ExternalArtifactSource):
+        # validate data type: ExternalBlobStorageSource
+        if not isinstance(v, ExternalBlobStorageSource):
             error_messages.append(
-                f"Error! Input type `{type(v)}` is not `ExternalArtifactSource`"
+                f"Error! Input type `{type(v)}` is not `ExternalBlobStorageSource`"
             )
         else:
             return v
@@ -89,7 +89,7 @@ class Source1(BaseModel):
         if error_messages:
             # no match
             raise ValueError(
-                "No match found when setting the actual_instance in Source1 with anyOf schemas: ExternalArtifactSource, TrueFoundryArtifactSource. Details: "
+                "No match found when setting the actual_instance in Source1 with anyOf schemas: ExternalBlobStorageSource, TrueFoundryManagedSource. Details: "
                 + ", ".join(error_messages)
             )
         else:
@@ -104,15 +104,15 @@ class Source1(BaseModel):
         """Returns the object represented by the json string"""
         instance = Source1.construct()
         error_messages = []
-        # anyof_schema_1_validator: Optional[TrueFoundryArtifactSource] = None
+        # anyof_schema_1_validator: Optional[TrueFoundryManagedSource] = None
         try:
-            instance.actual_instance = TrueFoundryArtifactSource.from_json(json_str)
+            instance.actual_instance = TrueFoundryManagedSource.from_json(json_str)
             return instance
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
-        # anyof_schema_2_validator: Optional[ExternalArtifactSource] = None
+        # anyof_schema_2_validator: Optional[ExternalBlobStorageSource] = None
         try:
-            instance.actual_instance = ExternalArtifactSource.from_json(json_str)
+            instance.actual_instance = ExternalBlobStorageSource.from_json(json_str)
             return instance
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
@@ -120,7 +120,7 @@ class Source1(BaseModel):
         if error_messages:
             # no match
             raise ValueError(
-                "No match found when deserializing the JSON string into Source1 with anyOf schemas: ExternalArtifactSource, TrueFoundryArtifactSource. Details: "
+                "No match found when deserializing the JSON string into Source1 with anyOf schemas: ExternalBlobStorageSource, TrueFoundryManagedSource. Details: "
                 + ", ".join(error_messages)
             )
         else:

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

@@ -16,21 +16,22 @@ from __future__ import annotations
 import json
 import pprint
 import re  # noqa: F401
+from typing import Optional
 
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 
-class TrueFoundryArtifactSource(BaseModel):
+class TrueFoundryManagedSource(BaseModel):
     """
-    +usage=Source for the Artifact or Model +label=Upload  # noqa: E501
+    +usage=Source for the entity +label=Upload  # noqa: E501
     """
 
     type: StrictStr = Field(
         default=...,
-        description="+usage=Type of the source +type=Upload +value=truefoundry +label=Upload",
+        description="+label=Type +usage=Type of the source +value=truefoundry",
     )
-    uri: StrictStr = Field(
-        default=...,
+    uri: Optional[StrictStr] = Field(
+        default=None,
         description='+label=URI +uiType=ModelUploadInput +uiProps={"hideClear":true}',
     )
     __properties = ["type", "uri"]
@@ -57,8 +58,8 @@ class TrueFoundryArtifactSource(BaseModel):
         return json.dumps(self.to_dict())
 
     @classmethod
-    def from_json(cls, json_str: str) -> TrueFoundryArtifactSource:
-        """Create an instance of TrueFoundryArtifactSource from a JSON string"""
+    def from_json(cls, json_str: str) -> TrueFoundryManagedSource:
+        """Create an instance of TrueFoundryManagedSource from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self):
@@ -67,15 +68,15 @@ class TrueFoundryArtifactSource(BaseModel):
         return _dict
 
     @classmethod
-    def from_dict(cls, obj: dict) -> TrueFoundryArtifactSource:
-        """Create an instance of TrueFoundryArtifactSource from a dict"""
+    def from_dict(cls, obj: dict) -> TrueFoundryManagedSource:
+        """Create an instance of TrueFoundryManagedSource from a dict"""
         if obj is None:
             return None
 
         if not isinstance(obj, dict):
-            return TrueFoundryArtifactSource.parse_obj(obj)
+            return TrueFoundryManagedSource.parse_obj(obj)
 
-        _obj = TrueFoundryArtifactSource.parse_obj(
+        _obj = TrueFoundryManagedSource.parse_obj(
             {"type": obj.get("type"), "uri": obj.get("uri")}
         )
         return _obj

truefoundry/ml/autogen/client_README.md

@@ -78,7 +78,6 @@ Class | Method | HTTP request | Description
 *ExperimentsApi* | [**hard_delete_experiment_post**](truefoundry/ml/autogen/client/docs/ExperimentsApi.md#hard_delete_experiment_post) | **POST** /api/2.0/mlflow/experiments/hard-delete | Hard Delete Experiment
 *ExperimentsApi* | [**list_experiment_columns_get**](truefoundry/ml/autogen/client/docs/ExperimentsApi.md#list_experiment_columns_get) | **GET** /api/2.0/mlflow/experiments/columns | List Experiment Columns
 *ExperimentsApi* | [**list_experiments_get**](truefoundry/ml/autogen/client/docs/ExperimentsApi.md#list_experiments_get) | **GET** /api/2.0/mlflow/experiments/list | List Experiments
-*ExperimentsApi* | [**put_privacy_type_put**](truefoundry/ml/autogen/client/docs/ExperimentsApi.md#put_privacy_type_put) | **PUT** /api/2.0/mlflow/experiments/privacy-type | Put Privacy Type
 *ExperimentsApi* | [**restore_experiment_post**](truefoundry/ml/autogen/client/docs/ExperimentsApi.md#restore_experiment_post) | **POST** /api/2.0/mlflow/experiments/restore | Restore Experiment
 *ExperimentsApi* | [**set_experiment_tag_post**](truefoundry/ml/autogen/client/docs/ExperimentsApi.md#set_experiment_tag_post) | **POST** /api/2.0/mlflow/experiments/set-experiment-tag | Set Experiment Tag
 *ExperimentsApi* | [**update_experiment_post**](truefoundry/ml/autogen/client/docs/ExperimentsApi.md#update_experiment_post) | **POST** /api/2.0/mlflow/experiments/update | Update Experiment
@@ -215,7 +214,7 @@ Class | Method | HTTP request | Description
  - [ExperimentResponseDto](truefoundry/ml/autogen/client/docs/ExperimentResponseDto.md)
  - [ExperimentTagDto](truefoundry/ml/autogen/client/docs/ExperimentTagDto.md)
  - [ExportDeploymentFilesRequestDto](truefoundry/ml/autogen/client/docs/ExportDeploymentFilesRequestDto.md)
- - [ExternalArtifactSource](truefoundry/ml/autogen/client/docs/ExternalArtifactSource.md)
+ - [ExternalBlobStorageSource](truefoundry/ml/autogen/client/docs/ExternalBlobStorageSource.md)
  - [FastAIFramework](truefoundry/ml/autogen/client/docs/FastAIFramework.md)
  - [FileInfoDto](truefoundry/ml/autogen/client/docs/FileInfoDto.md)
  - [FinalizeArtifactVersionRequestDto](truefoundry/ml/autogen/client/docs/FinalizeArtifactVersionRequestDto.md)
@@ -323,7 +322,7 @@ Class | Method | HTTP request | Description
  - [TransformersFramework](truefoundry/ml/autogen/client/docs/TransformersFramework.md)
  - [TriggerJobRunConfigRequestDto](truefoundry/ml/autogen/client/docs/TriggerJobRunConfigRequestDto.md)
  - [TriggerJobRunConfigResponseDto](truefoundry/ml/autogen/client/docs/TriggerJobRunConfigResponseDto.md)
- - [TrueFoundryArtifactSource](truefoundry/ml/autogen/client/docs/TrueFoundryArtifactSource.md)
+ - [TrueFoundryManagedSource](truefoundry/ml/autogen/client/docs/TrueFoundryManagedSource.md)
  - [UpdateArtifactVersionRequestDto](truefoundry/ml/autogen/client/docs/UpdateArtifactVersionRequestDto.md)
  - [UpdateDatasetRequestDto](truefoundry/ml/autogen/client/docs/UpdateDatasetRequestDto.md)
  - [UpdateExperimentRequestDto](truefoundry/ml/autogen/client/docs/UpdateExperimentRequestDto.md)