truefoundry 0.5.3rc5__py3-none-any.whl → 0.5.5rc1__py3-none-any.whl

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, Literal, Optional
+from typing import TYPE_CHECKING, Any, Dict, List, Optional
 from urllib.parse import urljoin
 
 import requests
@@ -43,7 +43,6 @@ 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,25 +707,3 @@ 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/deploy/lib/session.py

@@ -8,6 +8,7 @@ from truefoundry.common.constants import TFY_API_KEY_ENV_KEY, TFY_HOST_ENV_KEY
 from truefoundry.common.credential_file_manager import CredentialsFileManager
 from truefoundry.common.credential_provider import EnvCredentialProvider
 from truefoundry.common.entities import CredentialsFileContent, Token
+from truefoundry.common.session import Session
 from truefoundry.common.utils import relogin_error_message, resolve_tfy_host
 from truefoundry.deploy.io.output_callback import OutputCallBack
 from truefoundry.deploy.lib.const import (
@@ -98,6 +99,12 @@ def logout(
             output_hook.print_line(PROMPT_ALREADY_LOGGED_OUT)
 
 
+def get_access_token():
+    # Create a new session which handles token refresh
+    session = Session.new()
+    return session.access_token
+
+
 def _login_with_device_code(
     base_url: str,
     auth_service: AuthServiceClient,

truefoundry/deploy/v2/lib/deploy.py

@@ -74,10 +74,12 @@ def _handle_if_local_source(component: Component, workspace_fqn: str) -> Compone
         else:
             # We'll build image on TrueFoundry servers, upload the source and update image.build_source
             logger.info("Uploading code for %s '%s'", component.type, component.name)
+            client = ServiceFoundryServiceClient()
             new_component.image.build_source = local_source_to_remote_source(
                 local_source=component.image.build_source,
                 workspace_fqn=workspace_fqn,
                 component_name=component.name,
+                upload_code_package=client.upload_code_package,
             )
             logger.debug("Uploaded code for %s '%s'", component.type, component.name)
         return new_component

truefoundry/deploy/v2/lib/source.py

@@ -136,6 +136,7 @@ def local_source_to_remote_source(
     local_source: models.LocalSource,
     workspace_fqn: str,
     component_name: str,
+    upload_code_package: Callable[[str, str, str], str],
 ) -> RemoteSource:
     with tempfile.TemporaryDirectory() as local_dir:
         package_local_path = os.path.join(local_dir, "build.tar.gz")
@@ -164,8 +165,7 @@ def local_source_to_remote_source(
             logger.exception("Failed to calculate code archive size")
 
         logger.debug("Uploading code archive.")
-        client = ServiceFoundryServiceClient()
-        remote_uri = client.upload_code_package(
+        remote_uri = upload_code_package(
             workspace_fqn=workspace_fqn,
             component_name=component_name,
             package_local_path=package_local_path,

truefoundry/ml/autogen/client/__init__.py

@@ -184,8 +184,8 @@ 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_artifact_version_aliases_response_dto import (
+    GetArtifactVersionAliasesResponseDto,
 )
 from truefoundry.ml.autogen.client.models.get_experiment_response_dto import (
     GetExperimentResponseDto,

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

@@ -79,8 +79,8 @@ 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_artifact_version_aliases_response_dto import (
+    GetArtifactVersionAliasesResponseDto,
 )
 from truefoundry.ml.autogen.client.models.get_signed_url_for_dataset_write_request_dto import (
     GetSignedURLForDatasetWriteRequestDto,
@@ -5342,15 +5342,15 @@ class MlfoundryArtifactsApi:
         )
 
     @validate_arguments
-    def get_tags_for_artifact_get(
+    def get_version_aliases_for_artifact_get(
         self, artifact_id: StrictStr, **kwargs
-    ) -> GetArtifactTagsResponseDto:  # noqa: E501
-        """Get Tags For Artifact  # noqa: E501
+    ) -> GetArtifactVersionAliasesResponseDto:  # noqa: E501
+        """Get Version Aliases 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)
+        >>> thread = api.get_version_aliases_for_artifact_get(artifact_id, async_req=True)
         >>> result = thread.get()
 
         :param artifact_id: (required)
@@ -5364,24 +5364,26 @@ class MlfoundryArtifactsApi:
         :return: Returns the result object.
                  If the method is called asynchronously,
                  returns the request thread.
-        :rtype: GetArtifactTagsResponseDto
+        :rtype: GetArtifactVersionAliasesResponseDto
         """
         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
+            message = "Error! Please call the get_version_aliases_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
+        return self.get_version_aliases_for_artifact_get_with_http_info(
+            artifact_id, **kwargs
+        )  # noqa: E501
 
     @validate_arguments
-    def get_tags_for_artifact_get_with_http_info(
+    def get_version_aliases_for_artifact_get_with_http_info(
         self, artifact_id: StrictStr, **kwargs
     ) -> ApiResponse:  # noqa: E501
-        """Get Tags For Artifact  # noqa: E501
+        """Get Version Aliases 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)
+        >>> thread = api.get_version_aliases_for_artifact_get_with_http_info(artifact_id, async_req=True)
         >>> result = thread.get()
 
         :param artifact_id: (required)
@@ -5408,7 +5410,7 @@ class MlfoundryArtifactsApi:
         :return: Returns the result object.
                  If the method is called asynchronously,
                  returns the request thread.
-        :rtype: tuple(GetArtifactTagsResponseDto, status_code(int), headers(HTTPHeaderDict))
+        :rtype: tuple(GetArtifactVersionAliasesResponseDto, status_code(int), headers(HTTPHeaderDict))
         """
 
         _params = locals()
@@ -5431,7 +5433,7 @@ class MlfoundryArtifactsApi:
             if _key not in _all_params:
                 raise ApiTypeError(
                     "Got an unexpected keyword argument '%s'"
-                    " to method get_tags_for_artifact_get" % _key
+                    " to method get_version_aliases_for_artifact_get" % _key
                 )
             _params[_key] = _val
         del _params["kwargs"]
@@ -5462,12 +5464,12 @@ class MlfoundryArtifactsApi:
         _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
 
         _response_types_map = {
-            "200": "GetArtifactTagsResponseDto",
+            "200": "GetArtifactVersionAliasesResponseDto",
             "422": "HTTPValidationError",
         }
 
         return self.api_client.call_api(
-            "/api/2.0/mlflow/mlfoundry-artifacts/artifacts/get-tags",
+            "/api/2.0/mlflow/mlfoundry-artifacts/artifacts/get-version-aliases",
             "GET",
             _path_params,
             _query_params,

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

@@ -155,8 +155,8 @@ 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_artifact_version_aliases_response_dto import (
+    GetArtifactVersionAliasesResponseDto,
 )
 from truefoundry.ml.autogen.client.models.get_experiment_response_dto import (
     GetExperimentResponseDto,

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

@@ -24,7 +24,6 @@ from truefoundry.pydantic_v1 import (
     Field,
     StrictStr,
     conint,
-    conlist,
     constr,
     validator,
 )
@@ -43,16 +42,35 @@ 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(
+    version_alias: Optional[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",
+        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)",
     )
     type: Optional[StrictStr] = "artifact-version"
     source: Source = Field(...)
     step: Optional[conint(strict=True, ge=0)] = Field(
         default=0, description="+label=Step"
     )
-    __properties = ["description", "metadata", "tags", "type", "source", "step"]
+    __properties = [
+        "description",
+        "metadata",
+        "version_alias",
+        "type",
+        "source",
+        "step",
+    ]
+
+    @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("type")
     def type_validate_enum(cls, value):
@@ -104,7 +122,7 @@ class ArtifactVersionManifest(BaseModel):
             {
                 "description": obj.get("description"),
                 "metadata": obj.get("metadata"),
-                "tags": obj.get("tags"),
+                "version_alias": obj.get("version_alias"),
                 "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 → get_artifact_version_aliases_response_dto.py}

@@ -20,13 +20,13 @@ import re  # noqa: F401
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, conlist
 
 
-class GetArtifactTagsResponseDto(BaseModel):
+class GetArtifactVersionAliasesResponseDto(BaseModel):
     """
-    GetArtifactTagsResponseDto
+    GetArtifactVersionAliasesResponseDto
     """
 
-    tags: conlist(StrictStr) = Field(...)
-    __properties = ["tags"]
+    version_aliases: conlist(StrictStr) = Field(...)
+    __properties = ["version_aliases"]
 
     class Config:
         """Pydantic configuration"""
@@ -43,8 +43,8 @@ class GetArtifactTagsResponseDto(BaseModel):
         return json.dumps(self.to_dict())
 
     @classmethod
-    def from_json(cls, json_str: str) -> GetArtifactTagsResponseDto:
-        """Create an instance of GetArtifactTagsResponseDto from a JSON string"""
+    def from_json(cls, json_str: str) -> GetArtifactVersionAliasesResponseDto:
+        """Create an instance of GetArtifactVersionAliasesResponseDto from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self):
@@ -53,13 +53,15 @@ class GetArtifactTagsResponseDto(BaseModel):
         return _dict
 
     @classmethod
-    def from_dict(cls, obj: dict) -> GetArtifactTagsResponseDto:
-        """Create an instance of GetArtifactTagsResponseDto from a dict"""
+    def from_dict(cls, obj: dict) -> GetArtifactVersionAliasesResponseDto:
+        """Create an instance of GetArtifactVersionAliasesResponseDto from a dict"""
         if obj is None:
             return None
 
         if not isinstance(obj, dict):
-            return GetArtifactTagsResponseDto.parse_obj(obj)
+            return GetArtifactVersionAliasesResponseDto.parse_obj(obj)
 
-        _obj = GetArtifactTagsResponseDto.parse_obj({"tags": obj.get("tags")})
+        _obj = GetArtifactVersionAliasesResponseDto.parse_obj(
+            {"version_aliases": obj.get("version_aliases")}
+        )
         return _obj

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

@@ -28,7 +28,6 @@ from truefoundry.pydantic_v1 import (
     Field,
     StrictStr,
     conint,
-    conlist,
     constr,
     validator,
 )
@@ -47,9 +46,9 @@ 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(
+    version_alias: Optional[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",
+        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)",
     )
     type: Optional[StrictStr] = "model-version"
     source: Source1 = Field(...)
@@ -61,7 +60,7 @@ class ModelVersionManifest(BaseModel):
     __properties = [
         "description",
         "metadata",
-        "tags",
+        "version_alias",
         "type",
         "source",
         "framework",
@@ -69,6 +68,18 @@ class ModelVersionManifest(BaseModel):
         "step",
     ]
 
+    @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("type")
     def type_validate_enum(cls, value):
         """Validates the enum"""
@@ -125,7 +136,7 @@ class ModelVersionManifest(BaseModel):
             {
                 "description": obj.get("description"),
                 "metadata": obj.get("metadata"),
-                "tags": obj.get("tags"),
+                "version_alias": obj.get("version_alias"),
                 "type": obj.get("type")
                 if obj.get("type") is not None
                 else "model-version",

truefoundry/ml/autogen/client_README.md

@@ -121,7 +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* | [**get_version_aliases_for_artifact_get**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#get_version_aliases_for_artifact_get) | **GET** /api/2.0/mlflow/mlfoundry-artifacts/artifacts/get-version-aliases | Get Version Aliases 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
@@ -220,7 +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)
+ - [GetArtifactVersionAliasesResponseDto](truefoundry/ml/autogen/client/docs/GetArtifactVersionAliasesResponseDto.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-16T06:57:32+00:00
+#   timestamp: 2025-01-03T09:11:54+00:00
 
 from __future__ import annotations
 
@@ -58,16 +58,11 @@ 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,
-            )
-        ]
+    version_alias: Optional[
+        constr(regex=r"^v[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",
+        description="+label=Version Alias\n+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)\n+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)\n+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)",
     )
 
 

truefoundry/ml/log_types/artifacts/artifact.py

@@ -84,7 +84,7 @@ class ArtifactVersion:
         self._deleted = False
         self._description: str = ""
         self._metadata: Dict[str, Any] = {}
-        self._tags: List[str] = []
+        self._version_alias: Optional[str] = None
         self._set_mutable_attrs()
 
     @classmethod
@@ -132,13 +132,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 [])
+            self._version_alias = manifest.version_alias or None
         else:
             self._description = self._artifact_version.description or ""
             self._metadata = copy.deepcopy(
                 self._artifact_version.artifact_metadata or {}
             )
-            self._tags = []
+            self._version_alias = None
 
     def _refetch_artifact_version(self, reset_mutable_attrs: bool = True):
         _artifact_version = (
@@ -211,31 +211,31 @@ class ArtifactVersion:
         self._metadata = copy.deepcopy(value)
 
     @property
-    def tags(self) -> List[str]:
+    def version_alias(self) -> Optional[str]:
         """
-        Get tags for the current model version
+        Get version alias for the current artifact 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",
+                message="This artifact version was created using an older serialization format. version alias does not exist",
                 category=TrueFoundryDeprecationWarning,
                 stacklevel=2,
             )
-        return self._tags
+        return self._version_alias
 
-    @tags.setter
-    def tags(self, value: List[str]):
+    @version_alias.setter
+    def version_alias(self, value: Optional[str]):
         """
-        Set the tags for current model version
+        Set the version alias for current artifact 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",
+                message="This artifact version was created using an older serialization format. version alias will not be updated",
                 category=TrueFoundryDeprecationWarning,
                 stacklevel=2,
             )
             return
-        self._tags = copy.deepcopy(value)
+        self._version_alias = value
 
     @property
     def created_at(self) -> Optional[datetime.datetime]:
@@ -399,7 +399,7 @@ class ArtifactVersion:
             assert isinstance(manifest, ArtifactVersionManifest)
             manifest.description = self.description
             manifest.metadata = self.metadata
-            manifest.tags = self.tags
+            manifest.version_alias = self.version_alias
         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, List, Optional, Union
+from typing import TYPE_CHECKING, Any, Dict, Optional, Union
 
 from truefoundry.common.warnings import TrueFoundryDeprecationWarning
 from truefoundry.ml.artifact.truefoundry_artifact_repo import (
@@ -109,7 +109,7 @@ class ModelVersion:
         self._metadata: Dict[str, Any] = {}
         self._environment: Optional[ModelVersionEnvironment] = None
         self._framework: Optional[ModelFrameworkType] = None
-        self._tags: List[str] = []
+        self._version_alias: Optional[str] = None
         self._set_mutable_attrs()
 
     @classmethod
@@ -160,7 +160,7 @@ class ModelVersion:
                 if manifest.framework
                 else None
             )
-            self._tags = copy.deepcopy(self._model_version.manifest.tags) or []
+            self._version_alias = self._model_version.manifest.version_alias or None
         else:
             self._description = self._model_version.description or ""
             self._metadata = copy.deepcopy(self._model_version.artifact_metadata or {})
@@ -168,7 +168,7 @@ class ModelVersion:
             self._framework = _ModelFramework.to_model_framework_type(
                 self._model_version.model_framework
             )
-            self._tags = []
+            self._version_alias = None
 
     def _refetch_model_version(self, reset_mutable_attrs: bool = True):
         _model_version = self._mlfoundry_artifacts_api.get_model_version_get(
@@ -239,31 +239,31 @@ class ModelVersion:
         self._metadata = copy.deepcopy(value)
 
     @property
-    def tags(self) -> List[str]:
+    def version_alias(self) -> Optional[str]:
         """
-        Get tags for the current artifact version
+        Get version alias for the current model 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",
+                message="This model version was created using an older serialization format. version alias does not exist",
                 category=TrueFoundryDeprecationWarning,
                 stacklevel=2,
             )
-        return self._tags
+        return self._version_alias
 
-    @tags.setter
-    def tags(self, value: List[str]):
+    @version_alias.setter
+    def version_alias(self, value: Optional[str]):
         """
-        Set the tags for current artifact version
+        Set the version alias 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",
+                message="This model version was created using an older serialization format. version alias will not be updated",
                 category=TrueFoundryDeprecationWarning,
                 stacklevel=2,
             )
             return
-        self._tags = copy.deepcopy(value)
+        self._version_alias = value
 
     @property
     def environment(self) -> Optional[ModelVersionEnvironment]:
@@ -489,7 +489,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
+            self._model_version.manifest.version_alias = self.version_alias
         try:
             _model_version = self._mlfoundry_artifacts_api.update_model_version_post(
                 update_model_version_request_dto=UpdateModelVersionRequestDto(

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

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

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

@@ -25,7 +25,6 @@ truefoundry/autodeploy/utils/diff.py,sha256=Ef8Y-VffDKel_-q-GxRam6gqiv8qTLMcqVg6
 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/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
@@ -83,7 +82,7 @@ truefoundry/deploy/io/output_callback.py,sha256=V2YwUFec4G4a67lM4r-x_64AqdOVNo_9
 truefoundry/deploy/io/rich_output_callback.py,sha256=TJLiRD-EnFVwgcepxR7WN0koKqW1X2DevETPhNPi_nU,829
 truefoundry/deploy/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/lib/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/deploy/lib/clients/servicefoundry_client.py,sha256=l4KaHJmmTC3pP5hbnHuPhDH-MPs7KQBtTtfmmfyyBDI,27020
+truefoundry/deploy/lib/clients/servicefoundry_client.py,sha256=E1AZo3KmvvXoCQV4Z7A8Iy_iPH8HethHPR-xnAImEEE,26096
 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=oMszpueXPUfTUuN_XdKwoRjQyqAgWHhZ-10cbprCVdM,9226
@@ -94,28 +93,24 @@ truefoundry/deploy/lib/logs_utils.py,sha256=SQxRv3jDDmgHdOUMhlMaAPGYskybnBUMpst7
 truefoundry/deploy/lib/messages.py,sha256=nhp0bCYf_XpUM68hTq5lBY-__vtEyV2uP7NgnJXJ_Vg,925
 truefoundry/deploy/lib/model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/lib/model/entity.py,sha256=3YRdyvMYI1aUXUu2tB5b5lZWwpxKdAhjpT79gzTCZGo,8373
-truefoundry/deploy/lib/session.py,sha256=rNMkdzZejrxq5gTYTqJ-nGm869JzdlhHzaJQvINTQ4U,4953
+truefoundry/deploy/lib/session.py,sha256=FBSq3bxf3my_613NzgtPoeeaAYLGZEoF0n7XccTFNf4,5141
 truefoundry/deploy/lib/util.py,sha256=J7r8San2wKo48A7-BlH2-OKTlBO67zlPjLEhMsL8os0,1059
 truefoundry/deploy/lib/win32.py,sha256=1RcvPTdlOAJ48rt8rCbE2Ufha2ztRqBAE9dueNXArrY,5009
 truefoundry/deploy/python_deploy_codegen.py,sha256=qJHH1BJQII9e6PhkcRFYiE_3De7_VMMm8nM4AX5Eq1o,6513
 truefoundry/deploy/v2/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/v2/lib/__init__.py,sha256=WEiVMZXOVljzEE3tpGJil14liIn_PCDoACJ6b3tZ6sI,188
-truefoundry/deploy/v2/lib/deploy.py,sha256=oK5yTlrhzKNTGZckwuBzu8Q_pxQYgBVcdA-KB-SbzCI,11990
+truefoundry/deploy/v2/lib/deploy.py,sha256=Q4CL8SKnFLQ1FRqFyIwOAf77j_rP-fv58e-F-s3GqBw,12105
 truefoundry/deploy/v2/lib/deploy_workflow.py,sha256=6Y4PNhd5PCc_4TjpcIZPkr6FCm3BNIQ_SsJTvpUu7a0,12579
 truefoundry/deploy/v2/lib/deployable_patched_models.py,sha256=MROgMxhn9hDEAKwJSWl3iz12tUVvRKzEtqF2QUT6dAk,3343
 truefoundry/deploy/v2/lib/models.py,sha256=pSolLMTArDuYpeNsmeeS5DWliloN_iCDfZSpRllMHUg,1120
 truefoundry/deploy/v2/lib/patched_models.py,sha256=YDM75pdUPdMFgH2G8ZNbztalLIiMcHVHBHt7cLWmiDg,16050
-truefoundry/deploy/v2/lib/source.py,sha256=tzMSOTrTH3dJzICjo8gI8iY6n0-ckktP3Dh7bVK-nTI,9344
-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/deploy/v2/lib/source.py,sha256=wArU8AwSy7U9EUGLEMmB4Ke7ZwgZWmh8fFU-PACogIk,9347
 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=myDpr3RZzEkd_UG9C3rZ4UKEFxXopO5HMVJNpNDiuAo,37737
 truefoundry/ml/autogen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/ml/autogen/client/__init__.py,sha256=42Y8v6C1oDlngKCaDODi9k3q-9MPMiqO8jGe39y5QJM,20097
+truefoundry/ml/autogen/client/__init__.py,sha256=28MIn6pjqHZlnjuljt-kgrR7Dw07It7khbMZm3dlQlY,20118
 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
@@ -123,14 +118,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=Ze2mSkXH1IxUB6Zs3C8ISYwzpKfPXgGvdKgjmEa-XRA,319720
+truefoundry/ml/autogen/client/api/mlfoundry_artifacts_api.py,sha256=i3OaGysy3Th-ZVqJ6HenSn7W7cQkRUDS_G5D28tk6e4,319913
 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=ZtmxIyFDWPGywDB8p-VZO73Cl_WkGcl4fVzEB9UHsTc,18719
+truefoundry/ml/autogen/client/models/__init__.py,sha256=m2SztLLPzl_aifXj8-aFFl_gqkArLijw0ZLabOFOdCM,18740
 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
@@ -142,7 +137,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=lzoIGTRplPIcXlO5Brq6DoCn1HCnzmxJmd9dPvZ3Nq4,3907
+truefoundry/ml/autogen/client/models/artifact_version_manifest.py,sha256=AOg17pPxbOuqmSLCBj_us4_heijpr_k-CZWnl87RExw,4829
 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
@@ -193,7 +188,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_artifact_version_aliases_response_dto.py,sha256=W064ced8EQCeyJLpKikEtkCzuiqUDwNjS56sh7Mc3yg,2029
 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
@@ -252,7 +247,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=pZysfaTZmew4MxhtH4MTnwCQWRt43JvqU3ii5gBgm9w,4955
+truefoundry/ml/autogen/client/models/model_version_manifest.py,sha256=UrusTkY39LBGpVfjIt0xG-zbnSIa6zVfqzxMXOQGOPI,5822
 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
@@ -313,8 +308,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=Dr3x2Lm10JIWsLlRBZuDsE4Y4T1XzIpsychLjZfr89Q,37262
-truefoundry/ml/autogen/entities/artifacts.py,sha256=COkzHpmqzXpEttH2UstIO8p7WN3zz_tyAG63r66i-9w,20652
+truefoundry/ml/autogen/client_README.md,sha256=qrAp9e6D4rKZETpvJry0U8Yn2ZkxDHMA9NUNsfmO_JY,37326
+truefoundry/ml/autogen/entities/artifacts.py,sha256=jLBQk5OYpdhXwwLEIAQLkT-ttZbg522FBWHvgwOCQGQ,20995
 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
@@ -335,11 +330,11 @@ truefoundry/ml/exceptions.py,sha256=QpDJSWmF7dIsByS0qOQbQZ_jytdNTzkHDDO3BxhTSo0,
 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=qz657N5WosBNBqqEtZ9hWatvIoa9_0MipudYH9r3Bq0,21657
+truefoundry/ml/log_types/artifacts/artifact.py,sha256=b7GXYQx6MVI5qnPIS96xGqwwWVKGY4LYrEU9pCKkU_Y,21772
 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=J-_v_ertSKQPThx6I9ivLUlCJUHPZJ71X8V0VgaIBAU,24319
+truefoundry/ml/log_types/artifacts/model.py,sha256=UfGKJA9ywDtXEIH6E-VfdjvlQNdaKO-Kv76KaTZOVkA,24413
 truefoundry/ml/log_types/artifacts/utils.py,sha256=MuPk6by8jIt7LepB1fqUC8emQBSxrTGJmIel7rbMBAc,7533
 truefoundry/ml/log_types/image/__init__.py,sha256=fcOq8yQnNj1rkLcPeIjLXBpdA1WIeiPsXOlAAvMxx7M,76
 truefoundry/ml/log_types/image/constants.py,sha256=wLtGEOA4T5fZHSlOXPuNDLX3lpbCtwlvGKPFk_1fah0,255
@@ -373,7 +368,7 @@ truefoundry/workflow/remote_filesystem/tfy_signed_url_client.py,sha256=xcT0wQmQl
 truefoundry/workflow/remote_filesystem/tfy_signed_url_fs.py,sha256=nSGPZu0Gyd_jz0KsEE-7w_BmnTD8CVF1S8cUJoxaCbc,13305
 truefoundry/workflow/task.py,sha256=ToitYiKcNzFCtOVQwz1W8sRjbR97eVS7vQBdbgUQtKg,1779
 truefoundry/workflow/workflow.py,sha256=WaTqUjhwfAXDWu4E5ehuwAxrCbDJkoAf1oWmR2E9Qy0,4575
-truefoundry-0.5.3rc5.dist-info/METADATA,sha256=pJtlnztVvgBPT53hHtnRD-mp0EzV6iTNPFKlEDvH2oc,2887
-truefoundry-0.5.3rc5.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-truefoundry-0.5.3rc5.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
-truefoundry-0.5.3rc5.dist-info/RECORD,,
+truefoundry-0.5.5rc1.dist-info/METADATA,sha256=4z9NLPPbGWUaUFxD7Os5T-eZ2gCVv5dQQKwf19qAAxI,2887
+truefoundry-0.5.5rc1.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+truefoundry-0.5.5rc1.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
+truefoundry-0.5.5rc1.dist-info/RECORD,,

{truefoundry-0.5.3rc5.dist-info → truefoundry-0.5.5rc1.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.9.1
+Generator: poetry-core 2.0.1
 Root-Is-Purelib: true
 Tag: py3-none-any

truefoundry/cli/commands/pat.py

@@ -1,24 +0,0 @@
-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/gateway/__init__.py

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

truefoundry/gateway/cli/cli.py

@@ -1,30 +0,0 @@
-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

@@ -1,28 +0,0 @@
-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

@@ -1,29 +0,0 @@
-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