truefoundry 0.4.1__py3-none-any.whl → 0.4.2rc1__py3-none-any.whl

truefoundry/common/auth_service_client.py

@@ -8,7 +8,7 @@ from truefoundry.common.constants import VERSION_PREFIX
 from truefoundry.common.entities import DeviceCode, Token
 from truefoundry.common.exceptions import BadRequestException
 from truefoundry.common.request_utils import request_handling, requests_retry_session
-from truefoundry.common.utils import poll_for_function
+from truefoundry.common.utils import poll_for_function, relogin_error_message
 from truefoundry.logger import logger
 
 
@@ -16,6 +16,7 @@ class AuthServiceClient(ABC):
     def __init__(self, tenant_name: str):
         self._tenant_name = tenant_name
 
+    # TODO (chiragjn): Rename base_url to tfy_host
     @classmethod
     def from_base_url(cls, base_url: str) -> "AuthServiceClient":
         from truefoundry.common.servicefoundry_client import (
@@ -54,7 +55,9 @@ class ServiceFoundryServerAuthServiceClient(AuthServiceClient):
         if not token.refresh_token:
             # TODO: Add a way to propagate error messages without traceback to the output interface side
             raise Exception(
-                f"Unable to resume login session. Please log in again using `tfy login {host_arg_str} --relogin`"
+                relogin_error_message(
+                    "Unable to resume login session.", host=host_arg_str
+                )
             )
         url = f"{self._api_server_url}/{VERSION_PREFIX}/oauth2/token"
         data = {
@@ -70,7 +73,9 @@ class ServiceFoundryServerAuthServiceClient(AuthServiceClient):
             return Token.parse_obj(response_data)
         except BadRequestException as ex:
             raise Exception(
-                f"Unable to resume login session. Please log in again using `tfy login {host_arg_str} --relogin`"
+                relogin_error_message(
+                    "Unable to resume login session.", host=host_arg_str
+                )
             ) from ex
 
     def get_device_code(self) -> DeviceCode:
@@ -127,7 +132,9 @@ class AuthServerServiceClient(AuthServiceClient):
         if not token.refresh_token:
             # TODO: Add a way to propagate error messages without traceback to the output interface side
             raise Exception(
-                f"Unable to resume login session. Please log in again using `tfy login {host_arg_str} --relogin`"
+                relogin_error_message(
+                    "Unable to resume login session.", host=host_arg_str
+                )
             )
         url = f"{self._auth_server_url}/api/{VERSION_PREFIX}/oauth/token/refresh"
         data = {
@@ -141,7 +148,9 @@ class AuthServerServiceClient(AuthServiceClient):
             return Token.parse_obj(response_data)
         except BadRequestException as ex:
             raise Exception(
-                f"Unable to resume login session. Please log in again using `tfy login {host_arg_str} --relogin`"
+                relogin_error_message(
+                    "Unable to resume login session.", host=host_arg_str
+                )
             ) from ex
 
     def get_device_code(self) -> DeviceCode:

truefoundry/common/credential_file_manager.py

@@ -10,6 +10,7 @@ from filelock import FileLock, Timeout
 
 from truefoundry.common.constants import CREDENTIAL_FILEPATH
 from truefoundry.common.entities import CredentialsFileContent
+from truefoundry.common.utils import relogin_error_message
 from truefoundry.logger import logger
 
 
@@ -87,9 +88,9 @@ class CredentialsFileManager:
             return CredentialsFileContent.parse_file(self._credentials_file_path)
         except Exception as ex:
             raise Exception(
-                "Error while reading the credentials file "
-                f"{self._credentials_file_path}. Please login again "
-                "using `tfy login --relogin` or `tfy.login(relogin=True)` function"
+                relogin_error_message(
+                    f"Error while reading the credentials file {self._credentials_file_path}"
+                )
             ) from ex
 
     @_ensure_lock_taken

truefoundry/common/credential_provider.py

@@ -6,7 +6,7 @@ from truefoundry.common.auth_service_client import AuthServiceClient
 from truefoundry.common.constants import TFY_API_KEY_ENV_KEY
 from truefoundry.common.credential_file_manager import CredentialsFileManager
 from truefoundry.common.entities import CredentialsFileContent, Token
-from truefoundry.common.utils import resolve_base_url
+from truefoundry.common.utils import resolve_tfy_host
 from truefoundry.logger import logger
 
 TOKEN_REFRESH_LOCK = threading.RLock()
@@ -17,6 +17,7 @@ class CredentialProvider(ABC):
     @abstractmethod
     def token(self) -> Token: ...
 
+    # TODO (chiragjn): Rename base_url to tfy_host
     @property
     @abstractmethod
     def base_url(self) -> str: ...
@@ -34,10 +35,8 @@ class EnvCredentialProvider(CredentialProvider):
             raise Exception(
                 f"Value of {TFY_API_KEY_ENV_KEY} env var should be non-empty string"
             )
-        # TODO: Read host from cred file as well.
-        base_url = resolve_base_url().strip("/")
-        self._host = base_url
-        self._auth_service = AuthServiceClient.from_base_url(base_url=base_url)
+        self._host = resolve_tfy_host()
+        self._auth_service = AuthServiceClient.from_base_url(base_url=self._host)
         self._token: Token = Token(access_token=api_key, refresh_token=None)  # type: ignore[call-arg]
 
     @staticmethod

truefoundry/common/servicefoundry_client.py

@@ -68,6 +68,7 @@ def _cached_get_python_sdk_config(api_server_url: str) -> PythonSDKConfig:
 
 
 class ServiceFoundryServiceClient:
+    # TODO (chiragjn): Rename base_url to tfy_host
     def __init__(self, base_url: str):
         self._base_url = base_url.strip("/")
         self._api_server_url = append_servicefoundry_path_to_base_url(self._base_url)

truefoundry/common/utils.py

@@ -14,6 +14,17 @@ from truefoundry.common.constants import (
 T = TypeVar("T")
 
 
+def relogin_error_message(message: str, host: str = "HOST") -> str:
+    suffix = ""
+    if host == "HOST":
+        suffix = " where HOST is TrueFoundry platform URL"
+    return (
+        f"{message}\n"
+        f"Please login again using `tfy login --host {host} --relogin` "
+        f"or `truefoundry.login(host={host!r}, relogin=True)` function" + suffix
+    )
+
+
 def timed_lru_cache(
     seconds: int = 300, maxsize: Optional[int] = None
 ) -> Callable[[Callable[..., T]], Callable[..., T]]:
@@ -42,14 +53,25 @@ def poll_for_function(
         time.sleep(poll_after_secs)
 
 
-def resolve_base_url(host: Optional[str] = None) -> str:
-    if not host and not os.getenv(TFY_HOST_ENV_KEY):
+def validate_tfy_host(tfy_host: str) -> str:
+    if not (tfy_host.startswith("https://") or tfy_host.startswith("http://")):
+        raise ValueError(
+            f"Invalid host {tfy_host!r}. It should start with https:// or http://"
+        )
+
+
+def resolve_tfy_host(tfy_host: Optional[str] = None) -> str:
+    if not tfy_host and not os.getenv(TFY_HOST_ENV_KEY):
         raise ValueError(
-            f"Either `host` should be provided by --host <value>, or `{TFY_HOST_ENV_KEY}` env must be set"
+            f"Either `host` should be provided using `--host <value>`, or `{TFY_HOST_ENV_KEY}` env must be set"
         )
-    return host or os.getenv(TFY_HOST_ENV_KEY)
+    tfy_host = tfy_host or os.getenv(TFY_HOST_ENV_KEY)
+    tfy_host = tfy_host.strip("/")
+    validate_tfy_host(tfy_host)
+    return tfy_host
 
 
+# TODO (chiragjn): Rename base_url to tfy_host
 def append_servicefoundry_path_to_base_url(base_url: str):
     if urlsplit(base_url).netloc.startswith("localhost"):
         return os.getenv(SERVICEFOUNDRY_SERVER_URL_ENV_KEY)

truefoundry/deploy/lib/auth/servicefoundry_session.py

@@ -8,6 +8,7 @@ from truefoundry.common.credential_provider import (
     FileCredentialProvider,
 )
 from truefoundry.common.entities import UserInfo
+from truefoundry.common.utils import relogin_error_message
 from truefoundry.logger import logger
 
 ACTIVE_SESSION: Optional[ServiceFoundrySession] = None
@@ -41,8 +42,9 @@ class ServiceFoundrySession:
                 break
         if final_cred_provider is None:
             raise Exception(
-                "Please login again using `tfy login --relogin`"
-                "or `tfy.login(relogin=True)` function"
+                relogin_error_message(
+                    "No active session found. Perhaps you are not logged in?",
+                )
             )
         return final_cred_provider
 

truefoundry/deploy/lib/clients/servicefoundry_client.py

@@ -76,6 +76,7 @@ def _upload_packaged_code(metadata, package_file):
 
 
 class ServiceFoundryServiceClient(BaseServiceFoundryServiceClient):
+    # TODO (chiragjn): Rename base_url to tfy_host
     def __init__(self, init_session: bool = True, base_url: Optional[str] = None):
         self._session: Optional[ServiceFoundrySession] = None
         if init_session:

truefoundry/deploy/lib/session.py

@@ -8,7 +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.utils import resolve_base_url
+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 (
     RICH_OUTPUT_CALLBACK,
@@ -43,7 +43,7 @@ def login(
             "Login will just save the credentials on disk."
         )
 
-    host = resolve_base_url(host).strip("/")
+    host = resolve_tfy_host(host)
 
     with CredentialsFileManager() as cred_file:
         if not relogin and cred_file.exists():
@@ -58,10 +58,10 @@ def login(
             user_info = cred_file_content.to_token().to_user_info()
             user_name_display_info = user_info.email or user_info.user_type.value
             output_hook.print_line(
-                f"Already logged in to {cred_file_content.host!r} as "
-                f"{user_info.user_id!r} ({user_name_display_info})\n"
-                "Please use `tfy login --relogin` or `tfy.login(relogin=True)` "
-                "to force relogin"
+                relogin_error_message(
+                    f"Already logged in to {cred_file_content.host!r} as {user_info.user_id!r} ({user_name_display_info})",
+                    host=host,
+                )
             )
             return False
 

truefoundry/ml/autogen/client/__init__.py

@@ -144,6 +144,9 @@ from truefoundry.ml.autogen.client.models.delete_artifact_versions_request_dto i
 from truefoundry.ml.autogen.client.models.delete_dataset_request_dto import (
     DeleteDatasetRequestDto,
 )
+from truefoundry.ml.autogen.client.models.delete_files_for_dataset_request_dto import (
+    DeleteFilesForDatasetRequestDto,
+)
 from truefoundry.ml.autogen.client.models.delete_model_version_request_dto import (
     DeleteModelVersionRequestDto,
 )

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

@@ -73,6 +73,9 @@ from truefoundry.ml.autogen.client.models.delete_artifact_versions_request_dto i
 from truefoundry.ml.autogen.client.models.delete_dataset_request_dto import (
     DeleteDatasetRequestDto,
 )
+from truefoundry.ml.autogen.client.models.delete_files_for_dataset_request_dto import (
+    DeleteFilesForDatasetRequestDto,
+)
 from truefoundry.ml.autogen.client.models.delete_model_version_request_dto import (
     DeleteModelVersionRequestDto,
 )
@@ -2280,6 +2283,166 @@ class MlfoundryArtifactsApi:
             _request_auth=_params.get("_request_auth"),
         )
 
+    @validate_arguments
+    def delete_files_for_dataset_delete(
+        self,
+        delete_files_for_dataset_request_dto: DeleteFilesForDatasetRequestDto,
+        **kwargs,
+    ) -> object:  # noqa: E501
+        """Delete Files For Dataset  # noqa: E501
+
+        Delete files from the dataset.  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+
+        >>> thread = api.delete_files_for_dataset_delete(delete_files_for_dataset_request_dto, async_req=True)
+        >>> result = thread.get()
+
+        :param delete_files_for_dataset_request_dto: (required)
+        :type delete_files_for_dataset_request_dto: DeleteFilesForDatasetRequestDto
+        :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: object
+        """
+        kwargs["_return_http_data_only"] = True
+        if "_preload_content" in kwargs:
+            message = "Error! Please call the delete_files_for_dataset_delete_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data"  # noqa: E501
+            raise ValueError(message)
+        return self.delete_files_for_dataset_delete_with_http_info(
+            delete_files_for_dataset_request_dto, **kwargs
+        )  # noqa: E501
+
+    @validate_arguments
+    def delete_files_for_dataset_delete_with_http_info(
+        self,
+        delete_files_for_dataset_request_dto: DeleteFilesForDatasetRequestDto,
+        **kwargs,
+    ) -> ApiResponse:  # noqa: E501
+        """Delete Files For Dataset  # noqa: E501
+
+        Delete files from the dataset.  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+
+        >>> thread = api.delete_files_for_dataset_delete_with_http_info(delete_files_for_dataset_request_dto, async_req=True)
+        >>> result = thread.get()
+
+        :param delete_files_for_dataset_request_dto: (required)
+        :type delete_files_for_dataset_request_dto: DeleteFilesForDatasetRequestDto
+        :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(object, status_code(int), headers(HTTPHeaderDict))
+        """
+
+        _params = locals()
+
+        _all_params = ["delete_files_for_dataset_request_dto"]
+        _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 delete_files_for_dataset_delete" % _key
+                )
+            _params[_key] = _val
+        del _params["kwargs"]
+
+        _collection_formats = {}
+
+        # process the path parameters
+        _path_params = {}
+
+        # process the query parameters
+        _query_params = []
+        # process the header parameters
+        _header_params = dict(_params.get("_headers", {}))
+        # process the form parameters
+        _form_params = []
+        _files = {}
+        # process the body parameter
+        _body_params = None
+        if _params["delete_files_for_dataset_request_dto"] is not None:
+            _body_params = _params["delete_files_for_dataset_request_dto"]
+
+        # set the HTTP header `Accept`
+        _header_params["Accept"] = self.api_client.select_header_accept(
+            ["application/json"]
+        )  # noqa: E501
+
+        # set the HTTP header `Content-Type`
+        _content_types_list = _params.get(
+            "_content_type",
+            self.api_client.select_header_content_type(["application/json"]),
+        )
+        if _content_types_list:
+            _header_params["Content-Type"] = _content_types_list
+
+        # authentication setting
+        _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
+
+        _response_types_map = {
+            "200": "object",
+            "422": "HTTPValidationError",
+        }
+
+        return self.api_client.call_api(
+            "/api/2.0/mlflow/mlfoundry-artifacts/datasets/files/",
+            "DELETE",
+            _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 delete_model_version_post(
         self, delete_model_version_request_dto: DeleteModelVersionRequestDto, **kwargs

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

@@ -115,6 +115,9 @@ from truefoundry.ml.autogen.client.models.delete_artifact_versions_request_dto i
 from truefoundry.ml.autogen.client.models.delete_dataset_request_dto import (
     DeleteDatasetRequestDto,
 )
+from truefoundry.ml.autogen.client.models.delete_files_for_dataset_request_dto import (
+    DeleteFilesForDatasetRequestDto,
+)
 from truefoundry.ml.autogen.client.models.delete_model_version_request_dto import (
     DeleteModelVersionRequestDto,
 )

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

@@ -0,0 +1,68 @@
+# 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 DeleteFilesForDatasetRequestDto(BaseModel):
+    """
+    DeleteFilesForDatasetRequestDto
+    """
+
+    dataset_fqn: StrictStr = Field(...)
+    paths: conlist(StrictStr, max_items=100) = Field(...)
+    __properties = ["dataset_fqn", "paths"]
+
+    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) -> DeleteFilesForDatasetRequestDto:
+        """Create an instance of DeleteFilesForDatasetRequestDto 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) -> DeleteFilesForDatasetRequestDto:
+        """Create an instance of DeleteFilesForDatasetRequestDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return DeleteFilesForDatasetRequestDto.parse_obj(obj)
+
+        _obj = DeleteFilesForDatasetRequestDto.parse_obj(
+            {"dataset_fqn": obj.get("dataset_fqn"), "paths": obj.get("paths")}
+        )
+        return _obj

truefoundry/ml/autogen/client_README.md

@@ -98,6 +98,7 @@ Class | Method | HTTP request | Description
 *MlfoundryArtifactsApi* | [**delete_artifact_post**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#delete_artifact_post) | **POST** /api/2.0/mlflow/mlfoundry-artifacts/artifact/delete | Delete Artifact
 *MlfoundryArtifactsApi* | [**delete_artifact_version_post**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#delete_artifact_version_post) | **POST** /api/2.0/mlflow/mlfoundry-artifacts/artifact-versions/delete | Delete Artifact Version
 *MlfoundryArtifactsApi* | [**delete_dataset_post**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#delete_dataset_post) | **POST** /api/2.0/mlflow/mlfoundry-artifacts/datasets/delete | Delete Dataset
+*MlfoundryArtifactsApi* | [**delete_files_for_dataset_delete**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#delete_files_for_dataset_delete) | **DELETE** /api/2.0/mlflow/mlfoundry-artifacts/datasets/files/ | Delete Files For Dataset
 *MlfoundryArtifactsApi* | [**delete_model_version_post**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#delete_model_version_post) | **POST** /api/2.0/mlflow/mlfoundry-artifacts/model-versions/delete | Delete Model Version
 *MlfoundryArtifactsApi* | [**finalize_artifact_version_post**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#finalize_artifact_version_post) | **POST** /api/2.0/mlflow/mlfoundry-artifacts/artifact-versions/finalize | Finalize Artifact Version
 *MlfoundryArtifactsApi* | [**get_artifact_by_fqn_get**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#get_artifact_by_fqn_get) | **GET** /api/2.0/mlflow/mlfoundry-artifacts/artifacts/get-by-fqn | Get Artifact By Fqn
@@ -197,6 +198,7 @@ Class | Method | HTTP request | Description
  - [DatasetResponseDto](truefoundry/ml/autogen/client/docs/DatasetResponseDto.md)
  - [DeleteArtifactVersionsRequestDto](truefoundry/ml/autogen/client/docs/DeleteArtifactVersionsRequestDto.md)
  - [DeleteDatasetRequestDto](truefoundry/ml/autogen/client/docs/DeleteDatasetRequestDto.md)
+ - [DeleteFilesForDatasetRequestDto](truefoundry/ml/autogen/client/docs/DeleteFilesForDatasetRequestDto.md)
  - [DeleteModelVersionRequestDto](truefoundry/ml/autogen/client/docs/DeleteModelVersionRequestDto.md)
  - [DeleteRunRequest](truefoundry/ml/autogen/client/docs/DeleteRunRequest.md)
  - [DeleteTagRequestDto](truefoundry/ml/autogen/client/docs/DeleteTagRequestDto.md)

truefoundry/ml/autogen/entities/artifacts.py

@@ -0,0 +1,286 @@
+# generated by datamodel-codegen:
+#   filename:  artifacts.json
+#   timestamp: 2024-09-12T13:40:36+00:00
+
+from __future__ import annotations
+
+from enum import Enum
+from typing import Any, Dict, List, Literal, Optional, Union
+
+from truefoundry.pydantic_v1 import BaseModel, Field, constr
+
+
+class Agent(BaseModel):
+    type: Literal["agent"] = Field(..., description="+value=agent")
+    name: constr(regex=r"^[a-zA-Z][a-zA-Z0-9\-]{1,30}[a-zA-Z0-9]$") = Field(
+        ..., description="Names cannot repeat accross agent\n+uiType=Hidden"
+    )
+    available_tools: List[constr(min_length=1)] = Field(
+        ..., description="+sort=30\n+uiType=Hidden"
+    )
+    goal: constr(min_length=1, max_length=128) = Field(
+        ..., description="+sort=10\n+uiType=TextArea"
+    )
+    instruction: constr(min_length=1, max_length=2620) = Field(
+        ...,
+        description='`instruction` is the system prompt for now. (2.5 * 1024)\n+sort=20\n+uiType=AgentInstructions\n+uiProps={"helpText":"Use the syntax ${Tool FQN} to reference a tool, and ${AGENT FQN} to reference another agent"}',
+    )
+    model_id: constr(min_length=1) = Field(
+        ...,
+        description='+sort=40\n+uiType=EnabledModelSelector\n+uiProps={"searchable":true,"modelType":"chat","providerType":"openai"}',
+    )
+
+
+class Method(str, Enum):
+    """
+    +sort=50
+    +uiType=Hidden
+    """
+
+    get = "get"
+    post = "post"
+    put = "put"
+    delete = "delete"
+    patch = "patch"
+
+
+class AgentWithFQN(Agent):
+    id: str
+    fqn: str
+
+
+class MimeType(str, Enum):
+    """
+    +label=MIME Type
+    +usage=MIME type of the content
+    """
+
+    text_plain = "text/plain"
+    application_json = "application/json"
+    image_png = "image/png"
+    image_jpeg = "image/jpeg"
+    application_x_directory = "application/x-directory"
+
+
+class BlobStorageReference(BaseModel):
+    """
+    +usage=Blob Storage Location
+    +label=Blob Storage Location
+    +docs=Defines the structure for blob storage content references, including type, path, and mime_type
+    """
+
+    type: Literal["blob-storage"]
+    path: constr(regex=r"^.{1,}$") = Field(
+        ..., description="+label=Path\n+usage=Path to the content in blob storage"
+    )
+    mime_type: MimeType = Field(
+        ..., description="+label=MIME Type\n+usage=MIME type of the content"
+    )
+
+
+class ImageUrl(BaseModel):
+    """
+    +label=URL for the image
+    +usage=The URL for the image, must be a non-empty valid HTTPS URL or a data URL
+    """
+
+    url: Union[
+        constr(
+            regex=r"^\b((https?://)?(?:www\.)?[-a-zA-Z0-9@:%._\+~#=]{1,256}\.[a-zA-Z0-9()]{1,6}(?:[-a-zA-Z0-9()@:%_\+.~#?&/=]*)|data:image/[a-zA-Z]+;base64,[a-zA-Z0-9+/=\s]+)$"
+        ),
+        BlobStorageReference,
+    ] = Field(
+        ...,
+        description="+label=URL for the image\n+usage=The URL for the image, must be a non-empty valid HTTPS URL or a data URL",
+    )
+
+
+class ImageContentPart(BaseModel):
+    """
+    +usage=Image URL
+    +label=Image URL
+    """
+
+    type: Literal["image_url"]
+    image_url: ImageUrl = Field(
+        ...,
+        description="+label=URL for the image\n+usage=The URL for the image, must be a non-empty valid HTTPS URL or a data URL",
+    )
+
+
+class Parameters(BaseModel):
+    """
+    +usage=Parameters for the provider
+    +label=Parameters
+    +docs=Key-value pairs to store additional parameters for the provider
+    """
+
+    max_tokens: Optional[int] = None
+    temperature: Optional[float] = None
+    top_k: Optional[float] = None
+    top_p: Optional[float] = None
+    stop: Optional[Union[List[str], str]] = None
+
+
+class ModelConfiguration(BaseModel):
+    """
+    +label=Model Configuration
+    +icon=fa-cogs:#326ce5
+    +message=Configuration details for the provider and corresponding config
+    +usage=Define the provider and its configuration
+    +docs=Configuration settings specific to the provider, including model settings and other parameters.
+    """
+
+    provider: str = Field(
+        ...,
+        description='+label=Provider Name\n+usage=Name of the provider, must be non-empty, e.g., "openai", "google_gemini"',
+    )
+    model: constr(regex=r"^[a-zA-Z][a-zA-Z0-9\-]{1,30}[a-zA-Z0-9]$") = Field(
+        ..., description="+label=Model name\n+usage=Model name, must be non-empty"
+    )
+    parameters: Optional[Parameters] = Field(
+        None,
+        description="+usage=Parameters for the provider\n+label=Parameters\n+docs=Key-value pairs to store additional parameters for the provider",
+    )
+    extra_parameters: Optional[Dict[str, Any]] = Field(
+        None,
+        description="+usage=Extra parameters for the provider\n+label=Extra Parameters\n+docs=Additional parameters for the provider",
+    )
+
+
+class SystemMessage(BaseModel):
+    """
+    +usage=System message
+    +docs=Defines the structure of a system message, including role and content
+    +label=System Message
+    """
+
+    role: Literal["system"]
+    content: Union[constr(regex=r"^.[\s\S]*$"), BlobStorageReference] = Field(
+        ..., description="+label=Content\n+usage=Text content for the system message"
+    )
+    name: Optional[str] = Field(
+        None, description="+label=Name\n+usage=Name of the system"
+    )
+
+
+class TextContentPart(BaseModel):
+    """
+    +usage=Text content
+    +label=Text content
+    """
+
+    type: Literal["text"]
+    text: Union[constr(regex=r"^.[\s\S]*$"), BlobStorageReference]
+
+
+class UserMessage(BaseModel):
+    """
+    +usage=User message
+    +docs=Defines the structure of a user message, including role and content
+    +label=User Message
+    """
+
+    role: Literal["user"]
+    content: Union[
+        constr(regex=r"^.[\s\S]*$"),
+        BlobStorageReference,
+        List[Union[TextContentPart, ImageContentPart]],
+    ] = Field(
+        ..., description="+label=Content\n+usage=Text content for the user message"
+    )
+    name: Optional[str] = Field(
+        None, description="+label=Name\n+usage=Name of the user message"
+    )
+
+
+class AgentOpenAPITool(BaseModel):
+    type: Literal["openapi-tool"] = Field(..., description="+value=openapi-tool")
+    name: constr(regex=r"^[a-zA-Z][a-zA-Z0-9\-]{1,30}[a-zA-Z0-9]$") = Field(
+        ..., description="Names cannot repeat accross tools\n+uiType=Hidden"
+    )
+    description: constr(min_length=1, max_length=128) = Field(
+        ..., description="+sort=10\n+uiType=TextArea"
+    )
+    openapi_spec: Union[BlobStorageReference, Dict[str, Any]] = Field(
+        ..., description="+sort=20\n+uiType=OpenapiSchema"
+    )
+    base_url: constr(
+        regex=r"^(https?://)?(?:www\.)?[-a-zA-Z0-9@:%._\+~#=]{1,256}\.[a-zA-Z0-9()]{1,6}(?:[-a-zA-Z0-9()@:%_\+.~#?&/=]*)$"
+    ) = Field(..., description="+sort=30")
+    path: constr(min_length=1) = Field(
+        ...,
+        description="+sort=40\n+uiType=MethodPathSelector\n+label=Method + Path\n+usage=Select one Method and Path for this tool",
+    )
+    method: Method = Field(..., description="+sort=50\n+uiType=Hidden")
+    headers: Optional[Dict[str, str]] = Field(
+        None,
+        description='+sort=60\n+uiType=KV\n+uiProps={"allowSecrets":true,"secretConfig":{"enableNew":true,"hideOptions":true}}',
+    )
+
+
+class AgentOpenAPIToolWithFQN(AgentOpenAPITool):
+    id: str
+    fqn: str
+
+
+class AssistantMessage(BaseModel):
+    """
+    +usage=Assistant message
+    +docs=Defines the structure of an assistant message, including role and content
+    +label=Assistant Message
+    """
+
+    role: Literal["assistant"]
+    content: Union[constr(regex=r"^.[\s\S]*$"), BlobStorageReference] = Field(
+        ..., description="+label=Content\n+usage=Text content for the assistant message"
+    )
+    name: Optional[str] = Field(
+        None, description="+label=Name\n+usage=Name of the assistant message"
+    )
+
+
+class BasePrompt(BaseModel):
+    """
+    Main prompt structure
+    """
+
+    name: constr(regex=r"^[a-zA-Z][a-zA-Z0-9-_]{0,254}[a-zA-Z0-9]$") = Field(
+        ...,
+        description="+label=Prompt Name\n+icon=fa-desktop:#326ce5\n+message=Alphanumeric word, may contain '-' with a maximum length of 256 characters\n+sort=1000\n+usage=Name of the Prompt. This uniquely identifies it in the workspace.\n+docs=The unique name for the prompt, consisting of alphanumeric characters and dashes, max length 100.",
+    )
+    description: Optional[constr(min_length=1, max_length=512)] = Field(
+        None,
+        description="+label=Description\n+icon=fa-desktop:#326ce5\n+message=Description in a maximum of 512 characters\n+sort=2000\n+usage=Description of the prompt\n+docs=An optional description for the prompt, with a maximum length of 512 characters.\n+optional",
+    )
+    model_configuration: ModelConfiguration
+    metadata: Dict[str, str] = Field(
+        ...,
+        description="+label=Metadata\n+icon=fa-info-circle:#326ce5\n+message=Additional metadata for the prompt\n+sort=7000\n+usage=Define additional metadata for the prompt\n+docs=Key-value pairs to store additional metadata related to the prompt.",
+    )
+
+
+class ChatPrompt(BasePrompt):
+    type: Literal["chat_prompt"] = Field(
+        ...,
+        description='+label=Type\n+icon=fa-desktop:#326ce5\n+message=Type of the prompt\n+sort=3000\n+usage=Type of the prompt\n+docs=Type of the prompt, set to "chat_prompt"\n+default="chat_prompt"',
+    )
+    messages: List[Union[SystemMessage, AssistantMessage, UserMessage]] = Field(
+        ...,
+        description="+sort=4000\n+usage=Chat completion messages\n+label=Messages in the chat conversation\n+message=Chat completion messages\n+usage=List of messages in the chat conversation, must be non-empty\n+docs=Messages that define the chat conversation, including system, assistant, and user messages.",
+    )
+    variables: Optional[Dict[str, Optional[str]]] = Field(
+        None,
+        description="+label=Variables\n+usage=Variables for the chat completion messages to be used in the prompt messages\n+sort=5000",
+    )
+
+
+class AgentApp(BaseModel):
+    type: Literal["agent-app"] = Field(..., description="+value=agent-app")
+    tools: List[AgentOpenAPIToolWithFQN]
+    agents: List[AgentWithFQN]
+    root_agent: constr(min_length=1)
+
+
+class Artifacts(BaseModel):
+    __root__: Union[ChatPrompt, AgentOpenAPITool, Agent, AgentApp]

truefoundry/ml/clients/servicefoundry_client.py

@@ -16,6 +16,7 @@ from truefoundry.ml.run_utils import append_servicefoundry_path_to_tracking_uri
 
 
 class ServiceFoundryServiceClient(BaseServiceFoundryServiceClient):
+    # TODO (chiragjn): Rename tracking_uri to tfy_host
     def __init__(self, tracking_uri: str, token: Optional[str] = None):
         super().__init__(base_url=tracking_uri)
         self.host_creds = HostCreds(

truefoundry/ml/log_types/artifacts/artifact.py

@@ -5,7 +5,7 @@ import os
 import tempfile
 import uuid
 from pathlib import Path
-from typing import TYPE_CHECKING, Any, Dict, NamedTuple, Optional, Tuple, Union
+from typing import TYPE_CHECKING, Any, Dict, List, NamedTuple, Optional, Tuple, Union
 
 from truefoundry.ml.artifact.truefoundry_artifact_repo import (
     ArtifactIdentifier,
@@ -25,6 +25,7 @@ from truefoundry.ml.autogen.client import (  # type: ignore[attr-defined]
 from truefoundry.ml.autogen.client import (  # type: ignore[attr-defined]
     InternalMetadata as InternalMetadataDto,
 )
+from truefoundry.ml.autogen.entities.artifacts import ChatPrompt
 from truefoundry.ml.exceptions import MlFoundryException
 from truefoundry.ml.log_types.artifacts.constants import INTERNAL_METADATA_PATH
 from truefoundry.ml.log_types.artifacts.utils import (
@@ -351,6 +352,63 @@ class ArtifactVersion:
         self._set_mutable_attrs()
 
 
+class ChatPromptVersion(ArtifactVersion):
+    def __init__(self, artifact_version: ArtifactVersionDto, artifact: ArtifactDto):
+        if artifact.type != ArtifactType.CHAT_PROMPT:
+            raise ValueError(
+                f"{artifact_version.fqn!r} is not a chat prompt type artifact"
+            )
+        super().__init__(
+            artifact_version=artifact_version,
+            artifact=artifact,
+        )
+        self._chat_prompt = ChatPrompt.parse_obj(
+            artifact_version.internal_metadata.to_dict()
+        )
+
+    @classmethod
+    def from_fqn(cls, fqn: str) -> "ChatPromptVersion":
+        api_client = _get_api_client()
+        mlfoundry_artifacts_api = MlfoundryArtifactsApi(api_client=api_client)
+        _artifact_version = mlfoundry_artifacts_api.get_artifact_version_by_fqn_get(
+            fqn=fqn
+        )
+        artifact_version = _artifact_version.artifact_version
+        _artifact = mlfoundry_artifacts_api.get_artifact_by_id_get(
+            id=artifact_version.artifact_id
+        )
+        return cls(
+            artifact_version=_artifact_version.artifact_version,
+            artifact=_artifact.artifact,
+        )
+
+    @property
+    def model(self) -> str:
+        return self._chat_prompt.model_configuration.model
+
+    @property
+    def provider(self) -> str:
+        return self._chat_prompt.model_configuration.provider
+
+    @property
+    def messages(self) -> List[Dict[str, Any]]:
+        return [message.dict() for message in self._chat_prompt.messages]
+
+    @property
+    def parameters(self) -> Dict[str, Any]:
+        _parameters = self._chat_prompt.model_configuration.parameters
+        return _parameters.dict(exclude_unset=True) if _parameters else {}
+
+    @property
+    def extra_parameters(self) -> Dict[str, Any]:
+        _extra_parameters = self._chat_prompt.model_configuration.extra_parameters
+        return _extra_parameters.dict(exclude_unset=True) if _extra_parameters else {}
+
+    @property
+    def variables(self) -> Dict[str, Any]:
+        return self._chat_prompt.variables or {}
+
+
 def _log_artifact_version_helper(
     run: "MlFoundryRun",
     name: str,

truefoundry/ml/mlfoundry_api.py

@@ -7,6 +7,7 @@ from typing import Any, Dict, Iterator, List, Optional, Sequence, Tuple, Union
 import coolname
 import pandas as pd
 
+from truefoundry.common.utils import relogin_error_message
 from truefoundry.ml import constants, env_vars
 from truefoundry.ml.autogen.client import (  # type: ignore[attr-defined]
     ArtifactDto,
@@ -34,7 +35,11 @@ from truefoundry.ml.clients.servicefoundry_client import ServiceFoundryServiceCl
 from truefoundry.ml.enums import ModelFramework, ViewType
 from truefoundry.ml.exceptions import MlFoundryException
 from truefoundry.ml.internal_namespace import NAMESPACE
-from truefoundry.ml.log_types.artifacts.artifact import ArtifactPath, ArtifactVersion
+from truefoundry.ml.log_types.artifacts.artifact import (
+    ArtifactPath,
+    ArtifactVersion,
+    ChatPromptVersion,
+)
 from truefoundry.ml.log_types.artifacts.dataset import DataDirectory
 from truefoundry.ml.log_types.artifacts.general_artifact import _log_artifact_version
 from truefoundry.ml.log_types.artifacts.model import ModelVersion, _log_model_version
@@ -209,8 +214,9 @@ class MlFoundry:
         session = get_active_session()
         if session is None:
             raise MlFoundryException(
-                "No active session found. Perhaps you are not logged in?\n"
-                "Please log in using `tfy login [--host HOST] --relogin"
+                relogin_error_message(
+                    "No active session found. Perhaps you are not logged in?",
+                )
             )
         servicefoundry_client = ServiceFoundryServiceClient(
             tracking_uri=self.get_tracking_uri(),
@@ -1554,6 +1560,45 @@ class MlFoundry:
             if not datasets or not page_token:
                 done = True
 
+    def get_chat_prompt_version_by_fqn(self, fqn: str) -> ChatPromptVersion:
+        """
+        Get the prompt by prompt version FQN
+        Args:
+            fqn (str): Fully qualified name of the chat prompt version.
+        Returns:
+            ChatPromptVersion: An instance of the ChatPromptVersion class with the fetched prompt details.
+
+        Examples:
+            ```python
+            from truefoundry.ml import get_client
+            client = get_client()
+
+            # Get the chat prompt by FQN
+            fqn = 'chat_prompt:truefoundry/prompt-demo/demo-prompt-1:2'
+            chat_prompt = client.get_chat_prompt_version_by_fqn(fqn=fqn)
+
+            # Check the chat prompt values
+            print(chat_prompt)
+
+            # Use with OpenAI client
+            from openai import OpenAI
+            openai_client = OpenAI(api_key="api_key")
+
+            # Make the OpenAI call
+            response = openai_client.chat.completions.create(
+                model=chat_prompt.model,
+                messages=chat_prompt.messages,
+                **chat_prompt.parameters,
+                **chat_prompt.extra_parameters
+            )
+
+            # Extract the content from the response
+            response_content = response.choices[0].message.content
+            print(response_content)
+            ```
+        """
+        return ChatPromptVersion.from_fqn(fqn=fqn)
+
 
 def get_client() -> MlFoundry:
     """Initializes and returns the mlfoundry client.

truefoundry/ml/mlfoundry_run.py

@@ -19,6 +19,7 @@ from typing import (
 from urllib.parse import urljoin, urlsplit
 
 from truefoundry import version
+from truefoundry.common.utils import relogin_error_message
 from truefoundry.ml import constants, enums
 from truefoundry.ml.autogen.client import (  # type: ignore[attr-defined]
     ArtifactType,
@@ -222,8 +223,9 @@ class MlFoundryRun:
         session = get_active_session()
         if session is None:
             raise MlFoundryException(
-                "No active session found. Perhaps you are not logged in?\n"
-                "Please log in using `tfy login [--host HOST] --relogin"
+                relogin_error_message(
+                    "No active session found. Perhaps you are not logged in?",
+                )
             )
         base_url = "{uri.scheme}://{uri.netloc}/".format(
             uri=urlsplit(session.tracking_uri)

truefoundry/ml/session.py

@@ -10,6 +10,7 @@ from truefoundry.common.credential_provider import (
 )
 from truefoundry.common.entities import Token, UserInfo
 from truefoundry.common.request_utils import urllib3_retry
+from truefoundry.common.utils import relogin_error_message
 from truefoundry.ml.autogen.client import (  # type: ignore[attr-defined]
     ApiClient,
     Configuration,
@@ -83,6 +84,7 @@ class Session:
         self._assert_not_closed()
         return self._user_info
 
+    # TODO (chiragjn): Rename tracking_uri to tfy_host
     @property
     def tracking_uri(self) -> str:
         return self._cred_provider.base_url
@@ -120,8 +122,9 @@ def _get_api_client(
             return ApiClient()
         else:
             raise MlFoundryException(
-                "No active session found. Perhaps you are not logged in?\n"
-                "Please log in using `tfy login [--host HOST] --relogin"
+                relogin_error_message(
+                    "No active session found. Perhaps you are not logged in?",
+                )
             )
 
     creds = session.get_host_creds()
@@ -144,8 +147,9 @@ def init_session() -> Session:
                 break
         if final_cred_provider is None:
             raise MlFoundryException(
-                "Please login using `tfy login` command "
-                "or `truefoundry.login()` function call"
+                relogin_error_message(
+                    "No active session found. Perhaps you are not logged in?",
+                )
             )
         new_session = Session(cred_provider=final_cred_provider)
 

{truefoundry-0.4.1.dist-info → truefoundry-0.4.2rc1.dist-info}/METADATA

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

{truefoundry-0.4.1.dist-info → truefoundry-0.4.2rc1.dist-info}/RECORD

@@ -26,15 +26,15 @@ truefoundry/autodeploy/utils/pydantic_compat.py,sha256=hEAUy5kLjhPdzw7yGZ2iXGMXb
 truefoundry/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/cli/__main__.py,sha256=-NkhYlT3mC5MhtekueKAvCw-sWvguj0LJRpXWzvvFjc,727
 truefoundry/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/common/auth_service_client.py,sha256=upUTo6VODAaOKtszCQZNWnPpRT8CEhpCq9RLBJ0RrRY,7840
+truefoundry/common/auth_service_client.py,sha256=tZOa0NdATnItsMeTnEnUeTZQIgUJtpU-nvLdWtB4Px8,7978
 truefoundry/common/constants.py,sha256=kHP5ebp7EMfEYaOC-K8T_XPfYi7mXttYgrXYS8bOgkg,359
-truefoundry/common/credential_file_manager.py,sha256=MRn5h9Gm4pxZEF_XT-_b2agMBiqyw0o2SLpq9ty_1lY,4255
-truefoundry/common/credential_provider.py,sha256=UHr3tH39wGjhC3XdRUdgW7IZeh0cHSIEYJMVO8B4eyw,4107
+truefoundry/common/credential_file_manager.py,sha256=1yEk1Zm2xS4G0VDFwKSZ4w0VUrcPWQ1nJnoBaz9xyKA,4251
+truefoundry/common/credential_provider.py,sha256=YQ6HKl8ZZFTg48vBZMauEAnM6IrEO3oOzM2DA47N-P0,4071
 truefoundry/common/entities.py,sha256=8O-EGPk4PKqnyoFMKUTxISCU19rz0KBnfRDJU695DhY,3797
 truefoundry/common/exceptions.py,sha256=ePpiQ_zmWe4e94gOgeMiyP_AZnKwjEBfyXsB5ScGYcI,329
 truefoundry/common/request_utils.py,sha256=-ss8033PClJOMJuS5Ue1zCFRVK7ZW9vjtva1b5G9uok,2849
-truefoundry/common/servicefoundry_client.py,sha256=ornmmhjSY2y-NJeBcoHPKxX3q9Z9XilcHERFSFk-dSU,2886
-truefoundry/common/utils.py,sha256=DbyLQM4WiKxzIStg2Ekv7fLs8amZR0HtsF75uqy8tMc,1741
+truefoundry/common/servicefoundry_client.py,sha256=qYSrsbFH8b7N_2oxlXfbJnWfdqHCBQ1dQ5Rr2tdp3zQ,2937
+truefoundry/common/utils.py,sha256=f5Av8X70vZf6eUK4Axm8q75It2-C-3o3uomA0GrFt_4,2506
 truefoundry/deploy/__init__.py,sha256=ugawKF2G02EmEXX35oZ2tec12d9oWN28Sf6mtGGIERY,2281
 truefoundry/deploy/auto_gen/models.py,sha256=I4-tV2V74mljN7lH606pCD5LArK6MvND_i7Vt_aRI5Q,79952
 truefoundry/deploy/builder/__init__.py,sha256=a1qR6nicHGcxRaeNTxWRsmDs8zsmXc7j13-I8q0qqVk,4938
@@ -88,9 +88,9 @@ 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/json_util.py,sha256=x_-7YYQ4_HUIJ8ofOcclAp9JWhgTWjR9Th6Q0FuRqGk,175
 truefoundry/deploy/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/deploy/lib/auth/servicefoundry_session.py,sha256=d4M56O_ZftFeDRiyRci11QABU9B1HW2eVOeVIrFvevE,1842
+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=fgou4wjuWJzqQgW7iatbSCvmstSjY0iUJldQK5vsruE,26018
+truefoundry/deploy/lib/clients/servicefoundry_client.py,sha256=yzbXnVGmILD5BUMLzCHloJiz7YBWU0OCtrE3V09R9vA,26069
 truefoundry/deploy/lib/clients/shell_client.py,sha256=tMrc0Ha1DmGtUCJrZD8eusOzfe8R_WIe6AAH7nxL0xA,461
 truefoundry/deploy/lib/const.py,sha256=FCQfnO7IecB1ikQHdLGNvvubq_iF900C9l5TJtDfvFc,314
 truefoundry/deploy/lib/dao/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -103,7 +103,7 @@ 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=8J8yd98iWtSy8giShdDRNyzbN1UgSXx4XtmZLljdWnE,8552
-truefoundry/deploy/lib/session.py,sha256=e0iA-fWPBzxm5JqT4hnfv4NOpYexbrtRLSWLqqVtFzs,4986
+truefoundry/deploy/lib/session.py,sha256=Vg6rCA315T0yS0xG4ayJ84Ia_9ZfibH8utOSwPBMAmw,4953
 truefoundry/deploy/lib/util.py,sha256=RlL3bjZu5Z0LU_OKYaMVfcMU8k7_rmkAp89_0CrZDLk,1520
 truefoundry/deploy/lib/win32.py,sha256=1RcvPTdlOAJ48rt8rCbE2Ufha2ztRqBAE9dueNXArrY,5009
 truefoundry/deploy/python_deploy_codegen.py,sha256=Ok7ufDY2x3aMJv9KpaRqxiS-ZI-kxBWauIUHst-ug7E,4020
@@ -126,14 +126,14 @@ truefoundry/ml/__init__.py,sha256=2A1l7pgqbVRt3cRW_0Lxg92hyJEkMxkCUh1EFprrmc0,94
 truefoundry/ml/artifact/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/ml/artifact/truefoundry_artifact_repo.py,sha256=dA_QYAyNCcCuOw5eVxPbU5OeMK8YBNh42vBnGf6I0Sw,44911
 truefoundry/ml/autogen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/ml/autogen/client/__init__.py,sha256=9tmMg8M1td9h0zbAAdl8Lla3A33iU1XNMKx6pZj1yzA,16524
+truefoundry/ml/autogen/client/__init__.py,sha256=M68r1z8Mi6dEFFKDsWPJ9NGdNOYdm6pcdH7_srCwkCI,16651
 truefoundry/ml/autogen/client/api/__init__.py,sha256=3sMSljMIS3UHYeF0BcNvrPPx6VbBSRt_1IfDn-13Kyc,752
 truefoundry/ml/autogen/client/api/auth_api.py,sha256=zpWzJhUmW6HHMY_atlUf0B25k77E1kue2hmix5I5Ih0,7017
 truefoundry/ml/autogen/client/api/deprecated_api.py,sha256=JCQ39Y3VHdgJ1zM4XVabdTl6QpOHtQFjh04XUb6mN9A,24218
 truefoundry/ml/autogen/client/api/experiments_api.py,sha256=vzlPMzclwk7oD2ekUBInIAHLNO3qCAPF38AAvxfDBSY,80682
 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=CzhvDh_ZrsKmVVeX2fqNbfbWkCvosUQhkTEHx2pLGCo,306479
+truefoundry/ml/autogen/client/api/mlfoundry_artifacts_api.py,sha256=Le1fSNI8NwsYKhs-LTpH1fVuTP5YUp99D3WbZkQeo_8,313447
 truefoundry/ml/autogen/client/api/python_deployment_config_api.py,sha256=8P53aegaaZp1LEV9_GDNSCofgxRXP3Atyprc2bUvvpI,8348
 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
@@ -141,7 +141,7 @@ truefoundry/ml/autogen/client/api_client.py,sha256=8qg-WpadDuKgbRt5yABJ4wVS4IRxd
 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=IS04fIw-KPC87XbVETy7dFS2Exrpm3IMupfBf8E5ym8,15140
+truefoundry/ml/autogen/client/models/__init__.py,sha256=9zOPn2LD-j3EbZmtt8ShBU8uGMRVpTW2-kvooBG9feE,15267
 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=rU0h96pEE8K1Ukw2pzDSjq0e6BgtDEuOctI-aZMrpUY,2653
 truefoundry/ml/autogen/client/models/agent.py,sha256=fnMWdEPe5Iw50WKydtu7QAxU419j3ju1IukpChUnGqY,3871
@@ -186,6 +186,7 @@ truefoundry/ml/autogen/client/models/dataset_dto.py,sha256=2GqBLyIFXjNJ8stQRQgB1
 truefoundry/ml/autogen/client/models/dataset_response_dto.py,sha256=5h3eHpiG4Zzqn5Hm0WpZgyXGjxujKem42_8YnGq5KLo,2205
 truefoundry/ml/autogen/client/models/delete_artifact_versions_request_dto.py,sha256=JF2dQuEe_W1c2ox9zZDER_pZSTbKwSnns8hIfG5o7oA,1905
 truefoundry/ml/autogen/client/models/delete_dataset_request_dto.py,sha256=bfzzgBoekOBcGg2Uzzel80rf2Z5fDgggShkYd0ZlbSM,2143
+truefoundry/ml/autogen/client/models/delete_files_for_dataset_request_dto.py,sha256=-J1HeTo1utlA9FtoUGXuLXxKZeHLRux1VnpiiwINPLE,2058
 truefoundry/ml/autogen/client/models/delete_model_version_request_dto.py,sha256=6abYuueylSgv7TLkIHq8X0NWeFr-J5k72caAyOnn6fw,1873
 truefoundry/ml/autogen/client/models/delete_run_request.py,sha256=jXQmROKfkPA8idoxkUXT-2O6QJ4OgxgFX-CEmjryIpY,1793
 truefoundry/ml/autogen/client/models/delete_tag_request_dto.py,sha256=eQSPMamCJbsehcwWoMxQxlzZKQfX5GEwm3KHGF2Ajts,1901
@@ -287,14 +288,15 @@ truefoundry/ml/autogen/client/models/user_message.py,sha256=6QOmKT_SDfAVZA8V3Bzo
 truefoundry/ml/autogen/client/models/validation_error.py,sha256=mFjwoc8g2-Usu1HXZhOQKQ4TGvLy4lwCzk8dHrJ69aA,2597
 truefoundry/ml/autogen/client/models/validation_error_loc_inner.py,sha256=nThJ5Gmy8W2Wok-ZOI4sK7uRe1BAkLS0qzq-XZbq8zs,4915
 truefoundry/ml/autogen/client/rest.py,sha256=9goba8qHjQuVx5O_yRaTKu7PvBnb7r7swfy3dwuTEgk,14281
-truefoundry/ml/autogen/client_README.md,sha256=lobq-OlEcetj90WhMC1EmFQ85Bivmg2OPY2_CT1zc94,32876
+truefoundry/ml/autogen/client_README.md,sha256=8Pd74fIjHyzidtS6HGpusUgHj__nIdd_W6mBsi7tgRQ,33233
+truefoundry/ml/autogen/entities/artifacts.py,sha256=iIzfMPKLXZpzxu55lG0k3edraaZqXS4DLc6-FMzxk6g,10159
 truefoundry/ml/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/ml/cli/cli.py,sha256=ckBcjUpqfhgrPE1okqT_G2iouOLt-0KjpLhHp2YdVFU,256
 truefoundry/ml/cli/commands/__init__.py,sha256=diDUiRUX4l6TtNLI4iF-ZblczkELM7FRViJ-8gGNJQY,82
 truefoundry/ml/cli/commands/download.py,sha256=cbz9KijiLKXj4-twlig3xZLTVRNm4fnjwpy0leZr31w,2342
 truefoundry/ml/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/ml/clients/entities.py,sha256=sNP4DnAVdQoMfc06s0r3VTzKHTo7jmxAOuTlQOVmsMs,151
-truefoundry/ml/clients/servicefoundry_client.py,sha256=lK7mepLNRHHS65Eebynk5HyCMSxqmduXRHO2k8pTIK0,1680
+truefoundry/ml/clients/servicefoundry_client.py,sha256=w8fgZWZF_JjRpL_ZRmLw1BiI7aqDHnJcAXluBgwft0s,1735
 truefoundry/ml/clients/utils.py,sha256=c0LdC8moejs-Zm30hu1sCqifLEmqhdq4SfZ_m0nUIDk,4402
 truefoundry/ml/constants.py,sha256=vDq72d4C9FSWqr9MMdjgTF4TuyNFApvo_6RVsSeAjB4,2837
 truefoundry/ml/entities.py,sha256=si5GAqZsWzKu5MPrU4Hk6se7bebHOYhTiNw69ai-Uk8,1485
@@ -304,7 +306,7 @@ 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=KrQhROx82rT9Trzoj_-Lq6Ya5vuIzoRVIbjBiPz_sNc,15511
+truefoundry/ml/log_types/artifacts/artifact.py,sha256=pNgLc6GaQ0UVJ6gaSRzpAW5CmGpJMC4iBgBY5FLlofc,17640
 truefoundry/ml/log_types/artifacts/constants.py,sha256=qKxQ5mMvJE4j83BvGW3qNTKunxCiBg_EEjTdgbgJtyE,1036
 truefoundry/ml/log_types/artifacts/dataset.py,sha256=oXi7itXOe3PrnN4_KY7s1GdhWV1lyvWXRWgY6_vPTAs,12981
 truefoundry/ml/log_types/artifacts/general_artifact.py,sha256=fY74EUeCGdHdlkrp_GMoaZPvl7EQctt8OlRM6yFgLF8,3772
@@ -320,10 +322,10 @@ truefoundry/ml/log_types/plot.py,sha256=oFnXNb2o5fVF0zsnRjvqjSjLaphQWUnQCdw72e2u
 truefoundry/ml/log_types/pydantic_base.py,sha256=eBlw_AEyAz4iJKDP4zgJOCFWcldwQqpf7FADW1jzIQY,272
 truefoundry/ml/log_types/utils.py,sha256=xjJ21jdPScvFmw3TbVh5NCzbzJwaqiXJyiiT4xxX1EI,335
 truefoundry/ml/logger.py,sha256=VT-BF3BnBYTWVq87O58F0c8uXMu94gYzsiFlGY3_7Ao,458
-truefoundry/ml/mlfoundry_api.py,sha256=7RZZaMd6Z1f-qekpEFhjgqgDAFUZ3_7iZbzrRAhVj58,59379
-truefoundry/ml/mlfoundry_run.py,sha256=CTUgf2-kPANie9Kog_rJRvpkDfotVzWTHvBg2ofv_FA,44341
+truefoundry/ml/mlfoundry_api.py,sha256=I2T8tXeAIWpD8EH05fm80mNyX6cs2S1ORI4qoo0HTpQ,60847
+truefoundry/ml/mlfoundry_run.py,sha256=gl1aA_j7KwtyOpZgTTi3boWxpdTyssGGFh_lP40mc1o,44387
 truefoundry/ml/run_utils.py,sha256=dzbQ_TGkSCYO-gg8tlr5tglR0p2SIQdy0wl4IEQE5JA,2899
-truefoundry/ml/session.py,sha256=gASORQ7TZ3kqAyqG64YPfn5Z8Uo2BjOvfhMmTX-X7uM,5263
+truefoundry/ml/session.py,sha256=_WmaGNiZCwQVmB-brbX-z38nFvuqP8f7jGxfqJoq3TM,5385
 truefoundry/ml/validation_utils.py,sha256=XBSUd9OoyriWJpT3M5LKz17iWY3yVMr3hM5vdaVjtf0,12082
 truefoundry/pydantic_v1.py,sha256=jSuhGtz0Mbk1qYu8jJ1AcnIDK4oxUsdhALc4spqstmM,345
 truefoundry/version.py,sha256=bqiT4Q-VWrTC6P4qfK43mez-Ppf-smWfrl6DcwV7mrw,137
@@ -338,7 +340,7 @@ truefoundry/workflow/map_task.py,sha256=2m3qGXQ90k9LdS45q8dqCCECc3qr8t2m_LMCVd1m
 truefoundry/workflow/python_task.py,sha256=SRXRLC4vdBqGjhkwuaY39LEWN6iPCpJAuW17URRdWTY,1128
 truefoundry/workflow/task.py,sha256=ToitYiKcNzFCtOVQwz1W8sRjbR97eVS7vQBdbgUQtKg,1779
 truefoundry/workflow/workflow.py,sha256=WaTqUjhwfAXDWu4E5ehuwAxrCbDJkoAf1oWmR2E9Qy0,4575
-truefoundry-0.4.1.dist-info/METADATA,sha256=rPzMg57cm984bDhYSIO7oqFqYCy-shU4fXOiVceU35Y,3137
-truefoundry-0.4.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-truefoundry-0.4.1.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
-truefoundry-0.4.1.dist-info/RECORD,,
+truefoundry-0.4.2rc1.dist-info/METADATA,sha256=WmKC8UIgG2oVxwmlurcKI9T3aL0TTxfB8oiKuywh-uU,3140
+truefoundry-0.4.2rc1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+truefoundry-0.4.2rc1.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
+truefoundry-0.4.2rc1.dist-info/RECORD,,