truefoundry 0.5.3rc1__py3-none-any.whl → 0.5.3rc3__py3-none-any.whl

truefoundry/autodeploy/agents/base.py

@@ -17,12 +17,12 @@ from openai.types.chat.chat_completion_user_message_param import (
     ChatCompletionUserMessageParam,
 )
 from openai.types.shared_params.function_definition import FunctionDefinition
-from pydantic import BaseModel, ValidationError
 
 from truefoundry.autodeploy.constants import AUTODEPLOY_MODEL_NAME
 from truefoundry.autodeploy.logger import logger
 from truefoundry.autodeploy.tools import Event, RequestEvent, Tool
 from truefoundry.autodeploy.utils.pydantic_compat import model_dump, model_json_schema
+from truefoundry.pydantic_v1 import BaseModel, ValidationError
 
 
 def llm(

truefoundry/autodeploy/agents/developer.py

@@ -4,7 +4,6 @@ from typing import Dict, Optional
 
 import docker
 from openai import OpenAI
-from pydantic import Field
 
 from truefoundry.autodeploy.agents.base import Agent
 from truefoundry.autodeploy.agents.project_identifier import ProjectIdentifier
@@ -18,6 +17,7 @@ from truefoundry.autodeploy.tools import (
     RequestEvent,
     ResponseEvent,
 )
+from truefoundry.pydantic_v1 import Field
 
 
 class Developer(Agent):

truefoundry/autodeploy/agents/project_identifier.py

@@ -4,7 +4,6 @@ import enum
 from typing import List, Optional
 
 from openai import OpenAI
-from pydantic import Field
 from rich.console import Console
 
 from truefoundry.autodeploy.agents.base import Agent
@@ -16,6 +15,7 @@ from truefoundry.autodeploy.tools import (
     RequestEvent,
     ResponseEvent,
 )
+from truefoundry.pydantic_v1 import Field
 
 
 class ComponentType(str, enum.Enum):

truefoundry/autodeploy/agents/tester.py

@@ -4,7 +4,6 @@ from typing import Dict, Optional
 
 import docker
 from openai import OpenAI
-from pydantic import Field
 from rich.console import Console
 
 from truefoundry.autodeploy.agents.base import Agent
@@ -16,6 +15,7 @@ from truefoundry.autodeploy.tools import (
     ResponseEvent,
     SendRequest,
 )
+from truefoundry.pydantic_v1 import Field
 
 
 class Tester(Agent):

truefoundry/autodeploy/tools/base.py

@@ -2,9 +2,10 @@ from __future__ import annotations
 
 from typing import Any, ClassVar, Generator, Optional, Protocol, Type
 
-from pydantic import BaseModel
 from rich.console import Console
 
+from truefoundry.pydantic_v1 import BaseModel
+
 
 class Event(BaseModel):
     def render(self, _: Console) -> Optional[Any]: ...

truefoundry/autodeploy/tools/commit.py

@@ -4,7 +4,6 @@ import os
 import tempfile
 from typing import Any, Generator, Optional
 
-from pydantic import Field
 from rich.console import Console
 from rich.padding import Padding
 from rich.prompt import Confirm
@@ -20,6 +19,7 @@ from truefoundry.autodeploy.tools.base import (
 from truefoundry.autodeploy.utils.client import get_git_binary
 from truefoundry.autodeploy.utils.diff import LLMDiff
 from truefoundry.autodeploy.utils.pydantic_compat import model_dump
+from truefoundry.pydantic_v1 import Field
 
 
 class CommitConfirmation(Event):

truefoundry/autodeploy/tools/docker_build.py

@@ -6,7 +6,6 @@ from typing import Any, Generator, Iterable, Optional
 
 import docker
 from docker.models.images import BuildError, json_stream
-from pydantic import Field
 from rich.console import Console, ConsoleOptions, RenderResult
 from rich.padding import Padding
 from rich.text import Text
@@ -19,6 +18,7 @@ from truefoundry.autodeploy.tools.base import (
     Rule,
     Tool,
 )
+from truefoundry.pydantic_v1 import Field
 
 
 class DockerBuildLog(Event):

truefoundry/autodeploy/tools/docker_run.py

@@ -5,7 +5,6 @@ from typing import Any, Dict, Generator, List, Optional
 
 import docker
 import requests
-from pydantic import Field
 from rich.console import Console, ConsoleOptions, RenderResult
 from rich.padding import Padding
 from rich.text import Text
@@ -18,6 +17,7 @@ from truefoundry.autodeploy.tools.base import (
     Rule,
     Tool,
 )
+from truefoundry.pydantic_v1 import Field
 
 
 class DockerRunLog(Event):

truefoundry/autodeploy/tools/file_type_counts.py

@@ -5,7 +5,6 @@ from collections import Counter
 from typing import Any, Dict, Generator
 
 import gitignorefile
-from pydantic import Field
 from rich.console import Console
 from rich.table import Table
 
@@ -17,6 +16,7 @@ from truefoundry.autodeploy.tools.base import (
     Rule,
     Tool,
 )
+from truefoundry.pydantic_v1 import Field
 
 
 class ShowFileCount(Event):

truefoundry/autodeploy/tools/list_files.py

@@ -5,7 +5,6 @@ from fnmatch import fnmatch
 from typing import Any, Generator, List, Optional
 
 import gitignorefile
-from pydantic import Field
 
 from truefoundry.autodeploy.tools.base import (
     Event,
@@ -15,6 +14,7 @@ from truefoundry.autodeploy.tools.base import (
     Rule,
     Tool,
 )
+from truefoundry.pydantic_v1 import Field
 
 
 class ListFiles(Tool):

truefoundry/autodeploy/tools/read_file.py

@@ -3,8 +3,6 @@ from __future__ import annotations
 import os
 from typing import Any, Generator, List, Optional
 
-from pydantic import BaseModel, Field
-
 from truefoundry.autodeploy.constants import MAX_FILE_SIZE_READ
 from truefoundry.autodeploy.tools.base import (
     Event,
@@ -14,6 +12,7 @@ from truefoundry.autodeploy.tools.base import (
     Rule,
     Tool,
 )
+from truefoundry.pydantic_v1 import BaseModel, Field
 
 
 class Line(BaseModel):

truefoundry/autodeploy/tools/send_request.py

@@ -3,7 +3,6 @@ from __future__ import annotations
 from typing import Any, Generator, Optional
 
 import requests
-from pydantic import Field
 
 from truefoundry.autodeploy.tools.base import (
     Event,
@@ -13,6 +12,7 @@ from truefoundry.autodeploy.tools.base import (
     Rule,
     Tool,
 )
+from truefoundry.pydantic_v1 import Field
 
 
 class SendRequest(Tool):

truefoundry/autodeploy/tools/write_file.py

@@ -4,12 +4,12 @@ from __future__ import annotations
 import os
 from typing import Optional
 
-from pydantic import BaseModel, Field
 from rich.console import Console
 from rich.prompt import Confirm
 
 from truefoundry.autodeploy.tools.base import Tool
 from truefoundry.autodeploy.utils.diff import Diff
+from truefoundry.pydantic_v1 import BaseModel, Field
 
 INTERACTIVE_SESSION = True
 

truefoundry/cli/__main__.py

@@ -4,6 +4,7 @@ import sys
 import rich_click as click
 
 from truefoundry import logger
+from truefoundry.cli.commands.pat import get_generate_pat_command
 from truefoundry.cli.config import CliConfig
 from truefoundry.cli.const import GROUP_CLS
 from truefoundry.cli.util import setup_rich_click
@@ -21,6 +22,7 @@ from truefoundry.deploy.cli.commands import (
     get_terminate_command,
     get_trigger_command,
 )
+from truefoundry.gateway.cli.cli import get_gateway_cli
 from truefoundry.ml.cli.cli import get_ml_cli
 from truefoundry.version import __version__
 
@@ -88,6 +90,8 @@ def create_truefoundry_cli() -> click.MultiCommand:
     cli.add_command(get_trigger_command())
     cli.add_command(get_terminate_command())
     cli.add_command(get_ml_cli())
+    cli.add_command(get_gateway_cli())
+    cli.add_command(get_generate_pat_command())
 
     if not (sys.platform.startswith("win32") or sys.platform.startswith("cygwin")):
         cli.add_command(get_patch_command())

truefoundry/cli/commands/pat.py

@@ -0,0 +1,24 @@
+import click
+
+from truefoundry.cli.console import console
+from truefoundry.cli.const import COMMAND_CLS
+from truefoundry.deploy.lib.clients.servicefoundry_client import (
+    ServiceFoundryServiceClient,
+)
+
+
+@click.command(cls=COMMAND_CLS, name="generate-api-key")
+@click.option(
+    "--name",
+    required=True,
+    help="Name for the API key",
+)
+def generate_pat(name: str):
+    """Generate a new Personal Access Token with specified name"""
+    client = ServiceFoundryServiceClient()
+    pat = client.create_pat(name)
+    console.print(pat)
+
+
+def get_generate_pat_command():
+    return generate_pat

truefoundry/deploy/lib/clients/servicefoundry_client.py

@@ -4,7 +4,7 @@ import json
 import os
 import time
 from datetime import datetime, timezone
-from typing import TYPE_CHECKING, Any, Dict, List, Optional
+from typing import TYPE_CHECKING, Any, Dict, List, Literal, Optional
 from urllib.parse import urljoin
 
 import requests
@@ -43,6 +43,7 @@ from truefoundry.deploy.v2.lib.models import (
     BuildResponse,
     DeploymentFqnResponse,
 )
+from truefoundry.gateway.lib.entities import ProviderModels
 from truefoundry.logger import logger
 from truefoundry.pydantic_v1 import parse_obj_as
 
@@ -708,3 +709,25 @@ class ServiceFoundryServiceClient(BaseServiceFoundryServiceClient):
         )
         response_data = request_handling(response)
         return response_data
+
+    def get_gateway_models(
+        self, model_type: Optional[Literal["chat", "completion", "embedding"]] = None
+    ) -> ProviderModels:
+        url = f"{self._api_server_url}/{VERSION_PREFIX}/llm-gateway/model/enabled"
+        params = {}
+        if model_type:
+            params["modelType"] = model_type
+        response = session_with_retries().get(
+            url, headers=self._get_header(), params=params
+        )
+        response = request_handling(response)
+        return parse_obj_as(ProviderModels, response)
+
+    def create_pat(self, name: str):
+        url = f"{self._api_server_url}/{VERSION_PREFIX}/service-account/pat"
+        body = {"name": name}
+        response = session_with_retries().post(
+            url, headers=self._get_header(), json=body
+        )
+        response = request_handling(response)
+        return response

truefoundry/gateway/__init__.py

@@ -0,0 +1 @@
+"""Gateway module for Truefoundry CLI"""

truefoundry/gateway/cli/cli.py

@@ -0,0 +1,30 @@
+import click
+
+from truefoundry.cli.const import COMMAND_CLS, GROUP_CLS
+from truefoundry.cli.display_util import print_entity_list
+from truefoundry.gateway.lib.models import list_models
+
+
+def get_gateway_cli():
+    @click.group(cls=GROUP_CLS, help="Commands to interact with TrueFoundry Gateway")
+    def gateway(): ...
+
+    @gateway.group("list", cls=GROUP_CLS, help="List gateway resources")
+    def list_group():
+        """List gateway resources"""
+        pass
+
+    @list_group.command(
+        "models", cls=COMMAND_CLS, help="List available models filtered by type"
+    )
+    @click.option(
+        "--type",
+        "model_type",
+        type=click.Choice(["chat", "completion", "embedding"]),
+        help="Filter models by type",
+    )
+    def list_models_cli(model_type: str):
+        enabled_models = list_models(model_type)
+        print_entity_list("Models", enabled_models)
+
+    return gateway

truefoundry/gateway/lib/entities.py

@@ -0,0 +1,28 @@
+from typing import Any, Dict, List, Optional, Union
+
+from truefoundry.pydantic_v1 import BaseModel
+
+
+class GatewayModel(BaseModel):
+    id: str
+    name: str
+    provider: str
+    model_id: Optional[str]
+    provider_account_name: str
+    tfy_application_id: Optional[str] = None
+    enabled: bool = True
+    types: Union[str, List[str]] = ""
+    created_by: str
+    tenant_name: str
+    model_fqn: str
+
+    def list_row_data(self) -> Dict[str, Any]:
+        return {
+            "model_id": self.model_fqn,
+            "provider": self.provider,
+            "provider_model_id": self.model_id,
+        }
+
+
+class ProviderModels(BaseModel):
+    __root__: Dict[str, Dict[str, List[GatewayModel]]]

truefoundry/gateway/lib/models.py

@@ -0,0 +1,29 @@
+from typing import List, Literal, Optional
+
+from truefoundry.deploy.lib.clients.servicefoundry_client import (
+    ServiceFoundryServiceClient,
+)
+from truefoundry.gateway.lib.entities import GatewayModel
+
+
+def list_models(
+    model_type: Optional[Literal["chat", "completion", "embedding"]] = None,
+) -> List[GatewayModel]:
+    """List available models filtered by type
+
+    Args:
+        model_type (Optional[str], optional): Filter models by type ('chat' or 'completion'). Defaults to None.
+
+    Returns:
+        List: List of enabled models
+    """
+    client = ServiceFoundryServiceClient()
+    models = client.get_gateway_models(model_type)
+
+    enabled_models = []
+    for _, accounts in models.__root__.items():
+        for _, model_list in accounts.items():
+            for model in model_list:
+                enabled_models.append(model)
+
+    return enabled_models

truefoundry/ml/autogen/client/__init__.py

@@ -184,6 +184,9 @@ from truefoundry.ml.autogen.client.models.finalize_artifact_version_request_dto
     FinalizeArtifactVersionRequestDto,
 )
 from truefoundry.ml.autogen.client.models.framework import Framework
+from truefoundry.ml.autogen.client.models.get_artifact_tags_response_dto import (
+    GetArtifactTagsResponseDto,
+)
 from truefoundry.ml.autogen.client.models.get_experiment_response_dto import (
     GetExperimentResponseDto,
 )

truefoundry/ml/autogen/client/api/mlfoundry_artifacts_api.py

@@ -79,6 +79,9 @@ from truefoundry.ml.autogen.client.models.export_deployment_files_request_dto im
 from truefoundry.ml.autogen.client.models.finalize_artifact_version_request_dto import (
     FinalizeArtifactVersionRequestDto,
 )
+from truefoundry.ml.autogen.client.models.get_artifact_tags_response_dto import (
+    GetArtifactTagsResponseDto,
+)
 from truefoundry.ml.autogen.client.models.get_signed_url_for_dataset_write_request_dto import (
     GetSignedURLForDatasetWriteRequestDto,
 )
@@ -5338,6 +5341,150 @@ class MlfoundryArtifactsApi:
             _request_auth=_params.get("_request_auth"),
         )
 
+    @validate_arguments
+    def get_tags_for_artifact_get(
+        self, artifact_id: StrictStr, **kwargs
+    ) -> GetArtifactTagsResponseDto:  # noqa: E501
+        """Get Tags For Artifact  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+
+        >>> thread = api.get_tags_for_artifact_get(artifact_id, async_req=True)
+        >>> result = thread.get()
+
+        :param artifact_id: (required)
+        :type artifact_id: str
+        :param async_req: Whether to execute the request asynchronously.
+        :type async_req: bool, optional
+        :param _request_timeout: timeout setting for this request.
+               If one number provided, it will be total request
+               timeout. It can also be a pair (tuple) of
+               (connection, read) timeouts.
+        :return: Returns the result object.
+                 If the method is called asynchronously,
+                 returns the request thread.
+        :rtype: GetArtifactTagsResponseDto
+        """
+        kwargs["_return_http_data_only"] = True
+        if "_preload_content" in kwargs:
+            message = "Error! Please call the get_tags_for_artifact_get_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data"  # noqa: E501
+            raise ValueError(message)
+        return self.get_tags_for_artifact_get_with_http_info(artifact_id, **kwargs)  # noqa: E501
+
+    @validate_arguments
+    def get_tags_for_artifact_get_with_http_info(
+        self, artifact_id: StrictStr, **kwargs
+    ) -> ApiResponse:  # noqa: E501
+        """Get Tags For Artifact  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+
+        >>> thread = api.get_tags_for_artifact_get_with_http_info(artifact_id, async_req=True)
+        >>> result = thread.get()
+
+        :param artifact_id: (required)
+        :type artifact_id: str
+        :param async_req: Whether to execute the request asynchronously.
+        :type async_req: bool, optional
+        :param _preload_content: if False, the ApiResponse.data will
+                                 be set to none and raw_data will store the
+                                 HTTP response body without reading/decoding.
+                                 Default is True.
+        :type _preload_content: bool, optional
+        :param _return_http_data_only: response data instead of ApiResponse
+                                       object with status code, headers, etc
+        :type _return_http_data_only: bool, optional
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the authentication
+                              in the spec for a single request.
+        :type _request_auth: dict, optional
+        :type _content_type: string, optional: force content-type for the request
+        :return: Returns the result object.
+                 If the method is called asynchronously,
+                 returns the request thread.
+        :rtype: tuple(GetArtifactTagsResponseDto, status_code(int), headers(HTTPHeaderDict))
+        """
+
+        _params = locals()
+
+        _all_params = ["artifact_id"]
+        _all_params.extend(
+            [
+                "async_req",
+                "_return_http_data_only",
+                "_preload_content",
+                "_request_timeout",
+                "_request_auth",
+                "_content_type",
+                "_headers",
+            ]
+        )
+
+        # validate the arguments
+        for _key, _val in _params["kwargs"].items():
+            if _key not in _all_params:
+                raise ApiTypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method get_tags_for_artifact_get" % _key
+                )
+            _params[_key] = _val
+        del _params["kwargs"]
+
+        _collection_formats = {}
+
+        # process the path parameters
+        _path_params = {}
+
+        # process the query parameters
+        _query_params = []
+        if _params.get("artifact_id") is not None:  # noqa: E501
+            _query_params.append(("artifact_id", _params["artifact_id"]))
+
+        # process the header parameters
+        _header_params = dict(_params.get("_headers", {}))
+        # process the form parameters
+        _form_params = []
+        _files = {}
+        # process the body parameter
+        _body_params = None
+        # set the HTTP header `Accept`
+        _header_params["Accept"] = self.api_client.select_header_accept(
+            ["application/json"]
+        )  # noqa: E501
+
+        # authentication setting
+        _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
+
+        _response_types_map = {
+            "200": "GetArtifactTagsResponseDto",
+            "422": "HTTPValidationError",
+        }
+
+        return self.api_client.call_api(
+            "/api/2.0/mlflow/mlfoundry-artifacts/artifacts/get-tags",
+            "GET",
+            _path_params,
+            _query_params,
+            _header_params,
+            body=_body_params,
+            post_params=_form_params,
+            files=_files,
+            response_types_map=_response_types_map,
+            auth_settings=_auth_settings,
+            async_req=_params.get("async_req"),
+            _return_http_data_only=_params.get("_return_http_data_only"),  # noqa: E501
+            _preload_content=_params.get("_preload_content", True),
+            _request_timeout=_params.get("_request_timeout"),
+            collection_formats=_collection_formats,
+            _request_auth=_params.get("_request_auth"),
+        )
+
     @validate_arguments
     def list_artifact_versions_post(
         self,

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

@@ -155,6 +155,9 @@ from truefoundry.ml.autogen.client.models.finalize_artifact_version_request_dto
     FinalizeArtifactVersionRequestDto,
 )
 from truefoundry.ml.autogen.client.models.framework import Framework
+from truefoundry.ml.autogen.client.models.get_artifact_tags_response_dto import (
+    GetArtifactTagsResponseDto,
+)
 from truefoundry.ml.autogen.client.models.get_experiment_response_dto import (
     GetExperimentResponseDto,
 )

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

@@ -24,6 +24,7 @@ from truefoundry.pydantic_v1 import (
     Field,
     StrictStr,
     conint,
+    conlist,
     constr,
     validator,
 )
@@ -42,12 +43,16 @@ class ArtifactVersionManifest(BaseModel):
         default=...,
         description="+label=Metadata +docs=Metadata for the artifact or model version +usage=Metadata for the artifact or model version +uiType=JsonInput",
     )
+    tags: Optional[conlist(constr(strict=True, max_length=128))] = Field(
+        default=None,
+        description="+label=Tags +usage=Tags for the artifact or model version +docs=Tags for the artifact or model version",
+    )
     type: Optional[StrictStr] = "artifact-version"
     source: Source = Field(...)
     step: Optional[conint(strict=True, ge=0)] = Field(
         default=0, description="+label=Step"
     )
-    __properties = ["description", "metadata", "type", "source", "step"]
+    __properties = ["description", "metadata", "tags", "type", "source", "step"]
 
     @validator("type")
     def type_validate_enum(cls, value):
@@ -99,6 +104,7 @@ class ArtifactVersionManifest(BaseModel):
             {
                 "description": obj.get("description"),
                 "metadata": obj.get("metadata"),
+                "tags": obj.get("tags"),
                 "type": obj.get("type")
                 if obj.get("type") is not None
                 else "artifact-version",

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

@@ -0,0 +1,65 @@
+# coding: utf-8
+
+"""
+FastAPI
+
+No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+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 json
+import pprint
+import re  # noqa: F401
+
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, conlist
+
+
+class GetArtifactTagsResponseDto(BaseModel):
+    """
+    GetArtifactTagsResponseDto
+    """
+
+    tags: conlist(StrictStr) = Field(...)
+    __properties = ["tags"]
+
+    class Config:
+        """Pydantic configuration"""
+
+        allow_population_by_field_name = True
+        validate_assignment = True
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.dict(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) -> GetArtifactTagsResponseDto:
+        """Create an instance of GetArtifactTagsResponseDto 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.dict(by_alias=True, exclude={}, exclude_none=True)
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> GetArtifactTagsResponseDto:
+        """Create an instance of GetArtifactTagsResponseDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return GetArtifactTagsResponseDto.parse_obj(obj)
+
+        _obj = GetArtifactTagsResponseDto.parse_obj({"tags": obj.get("tags")})
+        return _obj

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

@@ -28,6 +28,7 @@ from truefoundry.pydantic_v1 import (
     Field,
     StrictStr,
     conint,
+    conlist,
     constr,
     validator,
 )
@@ -46,6 +47,10 @@ class ModelVersionManifest(BaseModel):
         default=...,
         description="+label=Metadata +docs=Metadata for the artifact or model version +usage=Metadata for the artifact or model version +uiType=JsonInput",
     )
+    tags: Optional[conlist(constr(strict=True, max_length=128))] = Field(
+        default=None,
+        description="+label=Tags +usage=Tags for the artifact or model version +docs=Tags for the artifact or model version",
+    )
     type: Optional[StrictStr] = "model-version"
     source: Source1 = Field(...)
     framework: Optional[Framework] = None
@@ -56,6 +61,7 @@ class ModelVersionManifest(BaseModel):
     __properties = [
         "description",
         "metadata",
+        "tags",
         "type",
         "source",
         "framework",
@@ -119,6 +125,7 @@ class ModelVersionManifest(BaseModel):
             {
                 "description": obj.get("description"),
                 "metadata": obj.get("metadata"),
+                "tags": obj.get("tags"),
                 "type": obj.get("type")
                 if obj.get("type") is not None
                 else "model-version",

truefoundry/ml/autogen/client_README.md

@@ -121,6 +121,7 @@ Class | Method | HTTP request | Description
 *MlfoundryArtifactsApi* | [**get_signed_urls_for_dataset_write_post**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#get_signed_urls_for_dataset_write_post) | **POST** /api/2.0/mlflow/mlfoundry-artifacts/datasets/get-signed-urls-for-write | Get Signed Urls For Dataset Write
 *MlfoundryArtifactsApi* | [**get_signed_urls_for_read_post**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#get_signed_urls_for_read_post) | **POST** /api/2.0/mlflow/mlfoundry-artifacts/artifact-versions/get-signed-urls-for-read | Get Signed Urls For Read
 *MlfoundryArtifactsApi* | [**get_signed_urls_for_write_post**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#get_signed_urls_for_write_post) | **POST** /api/2.0/mlflow/mlfoundry-artifacts/artifact-versions/get-signed-urls-for-write | Get Signed Urls For Write
+*MlfoundryArtifactsApi* | [**get_tags_for_artifact_get**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#get_tags_for_artifact_get) | **GET** /api/2.0/mlflow/mlfoundry-artifacts/artifacts/get-tags | Get Tags For Artifact
 *MlfoundryArtifactsApi* | [**list_artifact_versions_post**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#list_artifact_versions_post) | **POST** /api/2.0/mlflow/mlfoundry-artifacts/artifact-versions/list | List Artifact Versions
 *MlfoundryArtifactsApi* | [**list_artifacts_post**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#list_artifacts_post) | **POST** /api/2.0/mlflow/mlfoundry-artifacts/artifacts/list | List Artifacts
 *MlfoundryArtifactsApi* | [**list_datasets_post**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#list_datasets_post) | **POST** /api/2.0/mlflow/mlfoundry-artifacts/datasets/list | List Datasets
@@ -219,6 +220,7 @@ Class | Method | HTTP request | Description
  - [FileInfoDto](truefoundry/ml/autogen/client/docs/FileInfoDto.md)
  - [FinalizeArtifactVersionRequestDto](truefoundry/ml/autogen/client/docs/FinalizeArtifactVersionRequestDto.md)
  - [Framework](truefoundry/ml/autogen/client/docs/Framework.md)
+ - [GetArtifactTagsResponseDto](truefoundry/ml/autogen/client/docs/GetArtifactTagsResponseDto.md)
  - [GetExperimentResponseDto](truefoundry/ml/autogen/client/docs/GetExperimentResponseDto.md)
  - [GetLatestRunLogResponseDto](truefoundry/ml/autogen/client/docs/GetLatestRunLogResponseDto.md)
  - [GetMetricHistoryResponse](truefoundry/ml/autogen/client/docs/GetMetricHistoryResponse.md)

truefoundry/ml/autogen/entities/artifacts.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  artifacts.json
-#   timestamp: 2024-12-09T09:04:12+00:00
+#   timestamp: 2024-12-16T06:57:32+00:00
 
 from __future__ import annotations
 
@@ -58,6 +58,17 @@ class BaseArtifactVersion(BaseModel):
         ...,
         description="+label=Metadata\n+docs=Metadata for the artifact or model version\n+usage=Metadata for the artifact or model version\n+uiType=JsonInput",
     )
+    tags: Optional[
+        List[
+            constr(
+                regex=r"^(?!^\d+$)(?!.*[._-]{2,})[a-zA-Z0-9]+([._-][a-zA-Z0-9]+)*$",
+                max_length=128,
+            )
+        ]
+    ] = Field(
+        None,
+        description="+label=Tags\n+usage=Tags for the artifact or model version\n+docs=Tags for the artifact or model version",
+    )
 
 
 class MimeType(str, Enum):

truefoundry/ml/log_types/artifacts/artifact.py

@@ -5,11 +5,10 @@ import os
 import tempfile
 import typing
 import uuid
+import warnings
 from pathlib import Path
 from typing import TYPE_CHECKING, Any, Dict, List, NamedTuple, Optional, Union
 
-from pydantic import StrictStr
-
 from truefoundry.ml.artifact.truefoundry_artifact_repo import (
     ArtifactIdentifier,
     MlFoundryArtifactsRepository,
@@ -43,7 +42,7 @@ from truefoundry.ml.log_types.artifacts.utils import (
 )
 from truefoundry.ml.logger import logger
 from truefoundry.ml.session import _get_api_client
-from truefoundry.pydantic_v1 import BaseModel, Extra
+from truefoundry.pydantic_v1 import BaseModel, Extra, StrictStr
 
 if TYPE_CHECKING:
     from truefoundry.ml.mlfoundry_run import MlFoundryRun
@@ -84,6 +83,7 @@ class ArtifactVersion:
         self._deleted = False
         self._description: str = ""
         self._metadata: Dict[str, Any] = {}
+        self._tags: List[str] = []
         self._set_mutable_attrs()
 
     @classmethod
@@ -131,9 +131,13 @@ class ArtifactVersion:
             manifest = self._artifact_version.manifest.actual_instance
             self._description = manifest.description or ""
             self._metadata = copy.deepcopy(manifest.metadata)
+            self._tags = copy.deepcopy(manifest.tags or [])
         else:
             self._description = self._artifact_version.description or ""
-            self._metadata = copy.deepcopy(self._artifact_version.artifact_metadata)
+            self._metadata = copy.deepcopy(
+                self._artifact_version.artifact_metadata or {}
+            )
+            self._tags = []
 
     def _refetch_artifact_version(self, reset_mutable_attrs: bool = True):
         _artifact_version = (
@@ -205,6 +209,33 @@ class ArtifactVersion:
         _validate_artifact_metadata(value)
         self._metadata = copy.deepcopy(value)
 
+    @property
+    def tags(self) -> List[str]:
+        """
+        Get tags for the current model version
+        """
+        if not self._artifact_version.manifest:
+            warnings.warn(
+                message="This model version was created using an older serialization format. tags do not exist, returning empty list",
+                category=DeprecationWarning,
+                stacklevel=2,
+            )
+        return self._tags
+
+    @tags.setter
+    def tags(self, value: List[str]):
+        """
+        Set the tags for current model version
+        """
+        if not self._artifact_version.manifest:
+            warnings.warn(
+                message="This model version was created using an older serialization format. Tags will not be updated",
+                category=DeprecationWarning,
+                stacklevel=2,
+            )
+            return
+        self._tags = copy.deepcopy(value)
+
     @property
     def created_at(self) -> Optional[datetime.datetime]:
         """Get the time at which artifact was created"""
@@ -367,6 +398,7 @@ class ArtifactVersion:
             assert isinstance(manifest, ArtifactVersionManifest)
             manifest.description = self.description
             manifest.metadata = self.metadata
+            manifest.tags = self.tags
         else:
             manifest = None
         try:

truefoundry/ml/log_types/artifacts/model.py

@@ -8,7 +8,7 @@ import typing
 import uuid
 import warnings
 from pathlib import Path
-from typing import TYPE_CHECKING, Any, Dict, Optional, Union
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
 
 from truefoundry.ml.artifact.truefoundry_artifact_repo import (
     ArtifactIdentifier,
@@ -52,7 +52,7 @@ from truefoundry.ml.model_framework import (
     auto_update_model_framework_details,
 )
 from truefoundry.ml.session import _get_api_client
-from truefoundry.pydantic_v1 import BaseModel, Extra
+from truefoundry.pydantic_v1 import BaseModel, Extra, parse_obj_as
 
 if TYPE_CHECKING:
     import numpy as np
@@ -108,6 +108,7 @@ class ModelVersion:
         self._metadata: Dict[str, Any] = {}
         self._environment: Optional[ModelVersionEnvironment] = None
         self._framework: Optional[ModelFrameworkType] = None
+        self._tags: List[str] = []
         self._set_mutable_attrs()
 
     @classmethod
@@ -147,21 +148,26 @@ class ModelVersion:
 
     def _set_mutable_attrs(self):
         if self._model_version.manifest:
-            self._description = self._model_version.manifest.description or ""
-            self._metadata = copy.deepcopy(self._model_version.manifest.metadata)
-            self._environment = copy.deepcopy(self._model_version.manifest.environment)
+            manifest = self._model_version.manifest
+            self._description = manifest.description or ""
+            self._metadata = copy.deepcopy(manifest.metadata)
+            self._environment = copy.deepcopy(manifest.environment)
             self._framework = (
-                copy.deepcopy(self._model_version.manifest.framework.actual_instance)
-                if self._model_version.manifest.framework
+                parse_obj_as(
+                    ModelFrameworkType, manifest.framework.actual_instance.to_dict()
+                )
+                if manifest.framework
                 else None
             )
+            self._tags = copy.deepcopy(self._model_version.manifest.tags) or []
         else:
             self._description = self._model_version.description or ""
-            self._metadata = copy.deepcopy(self._model_version.artifact_metadata)
+            self._metadata = copy.deepcopy(self._model_version.artifact_metadata or {})
             self._environment = None
             self._framework = _ModelFramework.to_model_framework_type(
                 self._model_version.model_framework
             )
+            self._tags = []
 
     def _refetch_model_version(self, reset_mutable_attrs: bool = True):
         _model_version = self._mlfoundry_artifacts_api.get_model_version_get(
@@ -231,13 +237,46 @@ class ModelVersion:
         _validate_artifact_metadata(value)
         self._metadata = copy.deepcopy(value)
 
+    @property
+    def tags(self) -> List[str]:
+        """
+        Get tags for the current artifact version
+        """
+        if not self._model_version.manifest:
+            warnings.warn(
+                message="This model version was created using an older serialization format. tags do not exist, returning empty list",
+                category=DeprecationWarning,
+                stacklevel=2,
+            )
+        return self._tags
+
+    @tags.setter
+    def tags(self, value: List[str]):
+        """
+        Set the tags for current artifact version
+        """
+        if not self._model_version.manifest:
+            warnings.warn(
+                message="This model version was created using an older serialization format. Tags will not be updated",
+                category=DeprecationWarning,
+                stacklevel=2,
+            )
+            return
+        self._tags = copy.deepcopy(value)
+
     @property
     def environment(self) -> Optional[ModelVersionEnvironment]:
         """Get the environment details for the model"""
+        if not self._model_version.manifest:
+            warnings.warn(
+                message="This model version was created using an older serialization format. environment does not exist, returning None",
+                category=DeprecationWarning,
+                stacklevel=2,
+            )
         return self._environment
 
     @environment.setter
-    def environment(self, value: Optional[Dict[str, Any]]):
+    def environment(self, value: Optional[ModelVersionEnvironment]):
         """set the environment details for the model"""
         if not self._model_version.manifest:
             warnings.warn(
@@ -449,6 +488,7 @@ class ModelVersion:
             self._model_version.manifest.framework = (
                 Framework.from_dict(self.framework.dict()) if self.framework else None
             )
+            self._model_version.manifest.tags = self.tags
         try:
             _model_version = self._mlfoundry_artifacts_api.update_model_version_post(
                 update_model_version_request_dto=UpdateModelVersionRequestDto(

truefoundry/ml/model_framework.py

@@ -20,7 +20,8 @@ from truefoundry.common.utils import (
     get_python_version_major_minor,
     list_pip_packages_installed,
 )
-from truefoundry.ml.autogen.client import (
+from truefoundry.ml.autogen.client import (  # type: ignore[attr-defined]
+    ModelVersionEnvironment,
     SklearnSerializationFormat,
     XGBoostSerializationFormat,
 )
@@ -344,7 +345,7 @@ def _fetch_framework_specific_pip_packages(
 
 
 def auto_update_environment_details(
-    environment: autogen_artifacts.ModelVersionEnvironment,
+    environment: ModelVersionEnvironment,
     framework: Optional[ModelFrameworkType],
 ):
     """

{truefoundry-0.5.3rc1.dist-info → truefoundry-0.5.3rc3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: truefoundry
-Version: 0.5.3rc1
+Version: 0.5.3rc3
 Summary: Truefoundry CLI
 Author: Abhishek Choudhary
 Author-email: abhishek@truefoundry.com

{truefoundry-0.5.3rc1.dist-info → truefoundry-0.5.3rc3.dist-info}/RECORD

@@ -1,30 +1,31 @@
 truefoundry/__init__.py,sha256=jynqfY0pDmcaXlxewDnMQC5CYAIW9JFjptBrw7y76rY,81
 truefoundry/autodeploy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/autodeploy/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/autodeploy/agents/base.py,sha256=woCry229x7WX0IZmBopW_e_NB76aCLQRLXZBs2QBklk,6405
-truefoundry/autodeploy/agents/developer.py,sha256=Fh-EJDT3wPwCPYsrxRNtanyKMiZxhK73LHb-5Kpmh0A,4127
-truefoundry/autodeploy/agents/project_identifier.py,sha256=dm1A6MHNA-0Sc48y7dYpSA7e4uvw3Wi8AADSs-JcOZs,4505
-truefoundry/autodeploy/agents/tester.py,sha256=Hp1bul8Xd6qZPdUDUPqQwwHF7bsPQ5tK7x6gaBuX0NI,2403
+truefoundry/autodeploy/agents/base.py,sha256=lYw6bKhfCWff5BbtXXn2k4MjMbRIUkOwPBjRT70iXE8,6420
+truefoundry/autodeploy/agents/developer.py,sha256=tO9vmgGjoaqK4rX-DFCyMrViFC2Oqofd89WoXqsEm-A,4142
+truefoundry/autodeploy/agents/project_identifier.py,sha256=KlF7FwrGHDCjF3y3-YaCDxZ8kE34s6XPBagXhG2U1aM,4520
+truefoundry/autodeploy/agents/tester.py,sha256=lqvVBuVcFRvGZnSLAUF2r-1-Yti20o_cbybU19IlSXo,2418
 truefoundry/autodeploy/cli.py,sha256=1VnDAP9ylM6l00v0Ko5KL6_Dr7bL_voeAPdbEvWvU3Y,14215
 truefoundry/autodeploy/constants.py,sha256=vTh2nA7cjqghqbW2rNh3FbtcIk2scdAWZuuQCmVBO80,1273
 truefoundry/autodeploy/exception.py,sha256=fa_ZyTDUKiMKG2Uayynk1yWcEMsuVluqk2GtV4tfTPU,158
 truefoundry/autodeploy/logger.py,sha256=tkV2UKcOTFl5nz0cn4eRbzxF-2CZd8b7MK9vnhaflYw,325
 truefoundry/autodeploy/tools/__init__.py,sha256=9zJiC1d4bv9EL-p5XTCa9fAQ6ZKV--AbgeLz9bBBkyQ,875
 truefoundry/autodeploy/tools/ask.py,sha256=MxUFLP7rjpdJ85gCc3El0wUqTZDjpjAw7WOTdV4LLWE,856
-truefoundry/autodeploy/tools/base.py,sha256=eS30LQhUjQkRhFqHmdor7HgWA9kcwFVCfAWEimO0Y_0,779
-truefoundry/autodeploy/tools/commit.py,sha256=bB5kqAY5rEvfc7ZHtdueXusGznAbk58xqvzSKAXzbSE,6036
-truefoundry/autodeploy/tools/docker_build.py,sha256=TwSvmxrjBkI7PcPhjIzsc8oTA6fWzPIIpkMv2zMpb-E,4000
-truefoundry/autodeploy/tools/docker_run.py,sha256=8skvqF9zmZSIBbWkmJsmOyQRVz8ttw1SVjPhjGwDq24,5061
-truefoundry/autodeploy/tools/file_type_counts.py,sha256=RAiArWx2q1WBALJq6LKN8XMrtwecZiQrD6VtAdmTuvE,2354
-truefoundry/autodeploy/tools/list_files.py,sha256=mBhpnL0zY3Va2RM8Uuap-2G5Z-QPelSKp6O38K4DbGI,2642
-truefoundry/autodeploy/tools/read_file.py,sha256=uejLuCB9Tjki0-Ud_mk5fuo9nmKRHYHYqqzCTE1nuyo,2254
-truefoundry/autodeploy/tools/send_request.py,sha256=0Si6tdNEK152GUUkLPKKfV6K5vaS8kPw2FKIstRJKAA,1688
-truefoundry/autodeploy/tools/write_file.py,sha256=EcYetnJ4X-NWsd_CdYUKfOnAdz2XXiRukTzJfb_GzFs,3130
+truefoundry/autodeploy/tools/base.py,sha256=Ltzj2d6DO1gUMx_qSt2c2Zq91MJ5V4FJXpZwl4T3F8M,795
+truefoundry/autodeploy/tools/commit.py,sha256=MXqPnCnMHagvimSCCBGYsAbNdl7k1pf05BBBRjOP8b8,6051
+truefoundry/autodeploy/tools/docker_build.py,sha256=tAG62G1bk-kYrbRykieXxU48nkOlv3IcAv25k6zdTz8,4015
+truefoundry/autodeploy/tools/docker_run.py,sha256=PuiIvF41_L3JJN_EyoU-PAr2YkZEu0C9XX0ABcTFoJk,5076
+truefoundry/autodeploy/tools/file_type_counts.py,sha256=DUcULN0S2qIrZKTTTfz7GXeRy6RKtHSIeVkeECeEQZw,2369
+truefoundry/autodeploy/tools/list_files.py,sha256=HNZdsz-m-Jxf5yrf2bn7kRplIsNACP69Ozd4bpjlGMk,2657
+truefoundry/autodeploy/tools/read_file.py,sha256=ZpgTNef2-plWEoX4UHcVFVzHmNGoqZrvVUn1OVY_rrE,2268
+truefoundry/autodeploy/tools/send_request.py,sha256=Ea6MG0esSik7ZtWNIx9yTHlKJ593CQc11bET_YeRZgc,1703
+truefoundry/autodeploy/tools/write_file.py,sha256=tSO8LEwVJZbF17zva7FWE426_-C8pB5FVwc1vClpFjY,3145
 truefoundry/autodeploy/utils/client.py,sha256=PvbSkfgAjAogGjisinqmh4mP4svowxAC0IwzEQz3Ai0,453
 truefoundry/autodeploy/utils/diff.py,sha256=Ef8Y-VffDKel_-q-GxRam6gqiv8qTLMcqVg6iifXfcA,5358
 truefoundry/autodeploy/utils/pydantic_compat.py,sha256=hEAUy5kLjhPdzw7yGZ2iXGMXbbMVXVlGzIofmyHafXQ,412
 truefoundry/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/cli/__main__.py,sha256=0nEwE0UkFagCO5ctGcqEThdmBT2fN8MVcr0ERItnsNQ,3483
+truefoundry/cli/__main__.py,sha256=mYgMZRvI3sBP3_6DA0WwtFUZQTV3r8gJpMFvMJb-OOE,3692
+truefoundry/cli/commands/pat.py,sha256=0rm95ny1b3KfeBJSRWh1M4lR46ERnp96ovUZKu-P5Ss,602
 truefoundry/cli/config.py,sha256=tf8w4UfVzcC6eYkENvuuCPYt_V3sqVpO1bclORV9tAk,206
 truefoundry/cli/console.py,sha256=9-dMy4YPisCJQziRKTg8Qa0UJnOGl1soiUnJjsnLDvE,242
 truefoundry/cli/const.py,sha256=dVHPo1uAiDSSMXwXoT2mR5kNQjExT98QNVRz98Hz_Ts,510
@@ -80,7 +81,7 @@ truefoundry/deploy/io/rich_output_callback.py,sha256=TJLiRD-EnFVwgcepxR7WN0koKqW
 truefoundry/deploy/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/lib/auth/servicefoundry_session.py,sha256=5TCYPunAygtn5mb0mp_VcWKEalKMKPbyWMWer-Vty2g,1916
 truefoundry/deploy/lib/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/deploy/lib/clients/servicefoundry_client.py,sha256=yhHsHo9T0mTbum7X2sOuNVf-am1AdtzyK0KBQkdC0io,26212
+truefoundry/deploy/lib/clients/servicefoundry_client.py,sha256=zx9c-TjQRzMDKGCa-wDvXEX_8FcWH1QlVif2SjxgLxs,27136
 truefoundry/deploy/lib/const.py,sha256=Wg0GDnfFu-g1fJr4lU80NH2ULw0R0dYjV7LnW-PbOeM,173
 truefoundry/deploy/lib/dao/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/lib/dao/application.py,sha256=xUqF34abde71YZo9gkR-WvVJi1C6nBeGBNw4kL5FdIo,9226
@@ -103,12 +104,16 @@ truefoundry/deploy/v2/lib/deployable_patched_models.py,sha256=MROgMxhn9hDEAKwJSW
 truefoundry/deploy/v2/lib/models.py,sha256=pSolLMTArDuYpeNsmeeS5DWliloN_iCDfZSpRllMHUg,1120
 truefoundry/deploy/v2/lib/patched_models.py,sha256=NTU8J_CwdvEuF9zNXwFyN3suOp_196Wrm75DDy5qcXE,14184
 truefoundry/deploy/v2/lib/source.py,sha256=VHKuFREiixUP40D3Mrz-TA70spu1M0RbCzl--qwlFaY,9263
+truefoundry/gateway/__init__.py,sha256=HvAMuk60-fuZcMmdOtozHrUaH_JcbEHxpMSlS9ci4ls,41
+truefoundry/gateway/cli/cli.py,sha256=9raS6Nq_CNzagEPpbKICwUZ01zMoqmZINKMB9m_gJRE,908
+truefoundry/gateway/lib/entities.py,sha256=1P9YjrMoHbg58M57Pq-HlsT1ISHRyyX4JxK-o4of6cY,695
+truefoundry/gateway/lib/models.py,sha256=TXOur51tuYkq4Xil-icNVaVH-7xZbdOUyqS8N9295dg,876
 truefoundry/logger.py,sha256=u-YCNjg5HBwE70uQcpjIG64Ghos-K2ulTWaxC03BSj4,714
 truefoundry/ml/__init__.py,sha256=ssUEIs8BixPWxynKoeSh-dkRl6AtLXG0PBGYnUR5Az8,2217
 truefoundry/ml/artifact/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/ml/artifact/truefoundry_artifact_repo.py,sha256=FksxhUpRHb9pgWZmAB16DhXqkAL6UIAPA1S3RJUApQU,46201
 truefoundry/ml/autogen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/ml/autogen/client/__init__.py,sha256=Da3NzHW9aCi9zjQZteTm1aPTuyHDVqrFDtzwsdqIPbc,19981
+truefoundry/ml/autogen/client/__init__.py,sha256=42Y8v6C1oDlngKCaDODi9k3q-9MPMiqO8jGe39y5QJM,20097
 truefoundry/ml/autogen/client/api/__init__.py,sha256=NyMBxBmIzB1o5LzZZwz9LiydHB3-hPqc_sevsnY6Jrw,746
 truefoundry/ml/autogen/client/api/auth_api.py,sha256=zpWzJhUmW6HHMY_atlUf0B25k77E1kue2hmix5I5Ih0,7017
 truefoundry/ml/autogen/client/api/deprecated_api.py,sha256=mu5x_skNcwz8v1Tr6VP72-tVP7pmBV9muyKy_2NH3F0,38824
@@ -116,14 +121,14 @@ truefoundry/ml/autogen/client/api/experiments_api.py,sha256=mRKS8qGzcFJUpTWjfQoF
 truefoundry/ml/autogen/client/api/generate_code_snippet_api.py,sha256=C8iABP-pYIyIoTlL8PFi3ZBG_6B_dkQf09eErNeoj7A,22316
 truefoundry/ml/autogen/client/api/health_api.py,sha256=IAPhRAo9CLUT5ipVR1fCf-qDx57UR0wg5ekhtUl8lug,11554
 truefoundry/ml/autogen/client/api/metrics_api.py,sha256=q3L38eD-2hu4_9YvcSdnWDYXD2V8il-X9reinOAABek,14908
-truefoundry/ml/autogen/client/api/mlfoundry_artifacts_api.py,sha256=5FSOinHzGTbNh-F6KdpxFlVwPrtKVKn4jzr4Dwp9bpw,313554
+truefoundry/ml/autogen/client/api/mlfoundry_artifacts_api.py,sha256=Ze2mSkXH1IxUB6Zs3C8ISYwzpKfPXgGvdKgjmEa-XRA,319720
 truefoundry/ml/autogen/client/api/run_artifacts_api.py,sha256=x-vVnY2LEFChZxiiFauswRWwFz6Qqh30PKXjzuTvxmc,8799
 truefoundry/ml/autogen/client/api/runs_api.py,sha256=-aghrZ2VYuZOw_vBtOzWDsnK7Ji29oZQxK2CLRgyo2w,119232
 truefoundry/ml/autogen/client/api_client.py,sha256=M31IycWorn10FYS8WbO_Whaipr0tVu2pVWFCfbS7XCo,29167
 truefoundry/ml/autogen/client/api_response.py,sha256=KRyvecPMXF05PaxILHZ8JHoP4rgKBjKONMgG83aU-rM,844
 truefoundry/ml/autogen/client/configuration.py,sha256=V1oaEnxt-NfpaNmp-EZpf2glovzVhM2coWYt8HBNB4M,15723
 truefoundry/ml/autogen/client/exceptions.py,sha256=XbCbDHhYT3BVejdoGNPgEa4oS56ypkwFdxk1iOc_tFY,5355
-truefoundry/ml/autogen/client/models/__init__.py,sha256=4mkEcTiiMIGu16f2idqmFFaeT3T7E1qDIorrhhXReEk,18603
+truefoundry/ml/autogen/client/models/__init__.py,sha256=ZtmxIyFDWPGywDB8p-VZO73Cl_WkGcl4fVzEB9UHsTc,18719
 truefoundry/ml/autogen/client/models/add_custom_metrics_to_model_version_request_dto.py,sha256=_ISDspicTGjBCYYXubKfRYYSSQVyW3AvG-jFh47-Zfc,2163
 truefoundry/ml/autogen/client/models/add_features_to_model_version_request_dto.py,sha256=OT1-98DyWNfAHz_EgD2gMX2XkrGQ4Re945fhoAl8qSE,2099
 truefoundry/ml/autogen/client/models/agent.py,sha256=lUAbs092yo9hZmqzX6atKebDNf1uEwTf5jLvjOdvoeM,3872
@@ -135,7 +140,7 @@ truefoundry/ml/autogen/client/models/artifact_dto.py,sha256=JWL8_sITJJXbxywcfRj4
 truefoundry/ml/autogen/client/models/artifact_response_dto.py,sha256=T_HLjkHnqiO1bnSXiCUOlwDEHGz89mlUNhBq0SJTapo,2226
 truefoundry/ml/autogen/client/models/artifact_type.py,sha256=LBXGYUoesmL5J-0xzCR8jBST3bjb6HY86RmqSGvytRs,828
 truefoundry/ml/autogen/client/models/artifact_version_dto.py,sha256=GXjjKycpsEgJ60V2hAfL4NMxM2gZKLP2qV2yBb9WgdM,6174
-truefoundry/ml/autogen/client/models/artifact_version_manifest.py,sha256=3_r-Xsc1OwydVeh9nORAnKegkI73fGgJw4AzKzj0UPQ,3617
+truefoundry/ml/autogen/client/models/artifact_version_manifest.py,sha256=lzoIGTRplPIcXlO5Brq6DoCn1HCnzmxJmd9dPvZ3Nq4,3907
 truefoundry/ml/autogen/client/models/artifact_version_response_dto.py,sha256=JmiSYdPYG-ki6LCBeF7NtihdQ2LPiHld8cSPivY2GmQ,2421
 truefoundry/ml/autogen/client/models/artifact_version_serialization_format.py,sha256=2Qfgq1nykn0_mKNEJzOxhQwHm_95WIr9wJcHLeFWXQM,774
 truefoundry/ml/autogen/client/models/artifact_version_status.py,sha256=iIcomqhuaJkZH2rT3e6IjQutM79t9hEshW-RCIMKHEY,780
@@ -186,6 +191,7 @@ truefoundry/ml/autogen/client/models/fast_ai_framework.py,sha256=GwVoMbIzYhr4srF
 truefoundry/ml/autogen/client/models/file_info_dto.py,sha256=7oc8venicsFVk8zT9wHNhHnZGtFkFlqimFnS7ozGL9k,2156
 truefoundry/ml/autogen/client/models/finalize_artifact_version_request_dto.py,sha256=tkPcNjI0_qhX-D3m6AXqLzdmo9_K0tFBtbPDeg0Zo7U,3818
 truefoundry/ml/autogen/client/models/framework.py,sha256=W_KrqH1qKRvlrssCCWFIlmPaeAw0eiM6VqS3pbdlMXU,14720
+truefoundry/ml/autogen/client/models/get_artifact_tags_response_dto.py,sha256=zS5eBxBqzm-f5dt4GspLJvVmq_RWTtZmZTXFV8rqbX0,1883
 truefoundry/ml/autogen/client/models/get_experiment_response_dto.py,sha256=FhJpoUeRbZV1s1KR3Y07Kv6U4enLbbgdpGbguT9us0A,2923
 truefoundry/ml/autogen/client/models/get_latest_run_log_response_dto.py,sha256=g5NG252VzjGHiA-w5nhxmZxGrowJO9XCHDQTjf8y2KE,2266
 truefoundry/ml/autogen/client/models/get_metric_history_response.py,sha256=rTchO8o4sjGsNaZj8TW-VGtGGxsOBjUGzgaAWNtBZ-k,2428
@@ -244,7 +250,7 @@ truefoundry/ml/autogen/client/models/model_response_dto.py,sha256=osrTxfygkuhxWj
 truefoundry/ml/autogen/client/models/model_server.py,sha256=N5mcWmnX0KzTMtTH8A9h_FmiQcMld3B8CXD_UYeDKaA,700
 truefoundry/ml/autogen/client/models/model_version_dto.py,sha256=9Mh4d8V33gEGYr4sys5lYYDA3bHKajR3oAZLZMAy_io,7094
 truefoundry/ml/autogen/client/models/model_version_environment.py,sha256=wTayQjMen24eJCqXQugzoXLANY5NXdVsAvZSh-QIhJA,2827
-truefoundry/ml/autogen/client/models/model_version_manifest.py,sha256=gRsrZzfLm1cHk5cbHmYQtShXGLLxPKcxRMPDWCqWTZc,4657
+truefoundry/ml/autogen/client/models/model_version_manifest.py,sha256=pZysfaTZmew4MxhtH4MTnwCQWRt43JvqU3ii5gBgm9w,4955
 truefoundry/ml/autogen/client/models/model_version_response_dto.py,sha256=D6XOCyggxqTkbePuypqYSHYh1PYeDv7R_J32q61sDM0,2320
 truefoundry/ml/autogen/client/models/multi_part_upload_dto.py,sha256=Ckq405vud8RQmMyKCJQJBlW5iXO7Y2mlgo8eVkiMvxg,3738
 truefoundry/ml/autogen/client/models/multi_part_upload_response_dto.py,sha256=VjH0kvl7rMjgDHjYGHnsh7KsZ5-qn-k3ksdGLJ49nIM,2431
@@ -305,8 +311,8 @@ truefoundry/ml/autogen/client/models/xg_boost_framework.py,sha256=a0EHzncxe_hSFU
 truefoundry/ml/autogen/client/models/xg_boost_model_schema.py,sha256=QKJkiERNqF8UiolX2FjvhkUkUlUgDoo3CpTYPAdCxV8,2735
 truefoundry/ml/autogen/client/models/xg_boost_serialization_format.py,sha256=wFFSH0Z9MP8eT_fysHfUu4iqdjEZART-HxTH4e4gNI8,872
 truefoundry/ml/autogen/client/rest.py,sha256=9goba8qHjQuVx5O_yRaTKu7PvBnb7r7swfy3dwuTEgk,14281
-truefoundry/ml/autogen/client_README.md,sha256=9g4j_zKjNJleHNZu_N15KaFRsCAspWI_xMzuHkgq6iQ,36930
-truefoundry/ml/autogen/entities/artifacts.py,sha256=tH_gCB-w2A4Yq8kJck9vLVCQ1CxnIKVzBhDWMdH3Uho,20294
+truefoundry/ml/autogen/client_README.md,sha256=Dr3x2Lm10JIWsLlRBZuDsE4Y4T1XzIpsychLjZfr89Q,37262
+truefoundry/ml/autogen/entities/artifacts.py,sha256=COkzHpmqzXpEttH2UstIO8p7WN3zz_tyAG63r66i-9w,20652
 truefoundry/ml/autogen/models/__init__.py,sha256=--TGRea9SQBWFfwtcl3ekb1XGfFTdEkQGSG8a2SJ60I,187
 truefoundry/ml/autogen/models/exceptions.py,sha256=q3n7FGBrg_hUy1UyoefhMnhcXUAaqXlLIGHoOVzn_d8,1390
 truefoundry/ml/autogen/models/schema.py,sha256=IhpO9qbygLqEamP3NIt3m90SseJXCOm1ZTqNbNbW-M0,55772
@@ -329,11 +335,11 @@ truefoundry/ml/exceptions.py,sha256=8aJm2NYtAWWsRLu4MbzaoOqHsQZ6RjOFwBWQWqb6qrc,
 truefoundry/ml/git_info.py,sha256=jvAVm9ilqivnGq8qJdUvYdd8Siv0PLtqurB-PXsS5ho,2023
 truefoundry/ml/internal_namespace.py,sha256=QcqMHp6-C2im2H_02hlhi01EIcr1HhNaZprszs13EMU,1790
 truefoundry/ml/log_types/__init__.py,sha256=g4u4D4Jaj0aBK5GtrLV88-qThKZR9pSZ17vFEkN-LmM,125
-truefoundry/ml/log_types/artifacts/artifact.py,sha256=6X4lO23DwM7yTgUOY4RRoYMrf9m2AbiBBo2WftS2-dc,20457
+truefoundry/ml/log_types/artifacts/artifact.py,sha256=VC0KbcHty1fblybUvGVcfDGFcINfLgT2LByyvPcS1Ro,21565
 truefoundry/ml/log_types/artifacts/constants.py,sha256=qKxQ5mMvJE4j83BvGW3qNTKunxCiBg_EEjTdgbgJtyE,1036
 truefoundry/ml/log_types/artifacts/dataset.py,sha256=a4dxd2EN8p7Ci-cLGGiDOboN3t0395_XhWE1dmTw1Q4,13112
 truefoundry/ml/log_types/artifacts/general_artifact.py,sha256=B4XErLr-m6RmQWtxMTu3wlFRFcqSwPYp6J0OL4Ng6L0,3179
-truefoundry/ml/log_types/artifacts/model.py,sha256=sOqMptIpr2YRv7VC6AntFDvDUg5eQLsS7jQ_4TUB5Ds,22724
+truefoundry/ml/log_types/artifacts/model.py,sha256=p9cB3_0MSZkVyq8G2pyNWl1F3qICTazQlq6VCSzOV3s,24194
 truefoundry/ml/log_types/artifacts/utils.py,sha256=8mmahc1b6wvrA_E795h9ynRfhYRIdci7LRGN8U6ySm8,7516
 truefoundry/ml/log_types/image/__init__.py,sha256=fcOq8yQnNj1rkLcPeIjLXBpdA1WIeiPsXOlAAvMxx7M,76
 truefoundry/ml/log_types/image/constants.py,sha256=wLtGEOA4T5fZHSlOXPuNDLX3lpbCtwlvGKPFk_1fah0,255
@@ -346,7 +352,7 @@ truefoundry/ml/log_types/utils.py,sha256=xjJ21jdPScvFmw3TbVh5NCzbzJwaqiXJyiiT4xx
 truefoundry/ml/logger.py,sha256=VT-BF3BnBYTWVq87O58F0c8uXMu94gYzsiFlGY3_7Ao,458
 truefoundry/ml/mlfoundry_api.py,sha256=avuGgjbRnQXk774Vd3Dx7n99bbrOkL9pob9DI-vNvWE,62368
 truefoundry/ml/mlfoundry_run.py,sha256=scHjdjojRGbE4ZZlByZTFrcBTamRbm_JGidaSX9Gi9o,44601
-truefoundry/ml/model_framework.py,sha256=xHmVFI3Xcd6z6BH9TyZY8kFiw3Fp70423_M_wTYxd5Y,18841
+truefoundry/ml/model_framework.py,sha256=GAGu6bW8L4f8uvukOITB0bJOm1BSWzxC1MJxUGmpOqQ,18882
 truefoundry/ml/run_utils.py,sha256=0W208wSLUrbdfk2pjNcZlkUi9bNxG2JORqoe-5rVqHI,2423
 truefoundry/ml/session.py,sha256=F83GTC5WwGBjnJ69Ct8MqMnlutYc56JCc6YhEY1Wl-A,5394
 truefoundry/ml/validation_utils.py,sha256=J5atNhcJLvKj64ralSV9Y5Fv1Rt4SE237ICdP9-7sP4,12149
@@ -367,7 +373,7 @@ truefoundry/workflow/remote_filesystem/tfy_signed_url_client.py,sha256=5mBCIc-ON
 truefoundry/workflow/remote_filesystem/tfy_signed_url_fs.py,sha256=Hf6Dk6Fu6P7DqsK5ULgraf9DStjgigf-kjaRAMBW-RU,8680
 truefoundry/workflow/task.py,sha256=ToitYiKcNzFCtOVQwz1W8sRjbR97eVS7vQBdbgUQtKg,1779
 truefoundry/workflow/workflow.py,sha256=WaTqUjhwfAXDWu4E5ehuwAxrCbDJkoAf1oWmR2E9Qy0,4575
-truefoundry-0.5.3rc1.dist-info/METADATA,sha256=Sqr8OAIAxQccpWW5gHpOqhDxf4MWaIjlstHnzaC8GEw,2887
-truefoundry-0.5.3rc1.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-truefoundry-0.5.3rc1.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
-truefoundry-0.5.3rc1.dist-info/RECORD,,
+truefoundry-0.5.3rc3.dist-info/METADATA,sha256=zuoxLW-MzlfmWPLCP8J2NF8eydbOKPu_LNhKouhQi_0,2887
+truefoundry-0.5.3rc3.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+truefoundry-0.5.3rc3.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
+truefoundry-0.5.3rc3.dist-info/RECORD,,