truefoundry 0.4.4rc9__py3-none-any.whl → 0.4.4rc11__py3-none-any.whl

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

@@ -0,0 +1,74 @@
+# 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, validator
+
+
+class UploadModelSource(BaseModel):
+    """
+    UploadModelSource
+    """
+
+    type: StrictStr = Field(
+        default=..., description="+label=Type +usage=Type of the source"
+    )
+    __properties = ["type"]
+
+    @validator("type")
+    def type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in ("upload"):
+            raise ValueError("must be one of enum values ('upload')")
+        return value
+
+    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) -> UploadModelSource:
+        """Create an instance of UploadModelSource 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) -> UploadModelSource:
+        """Create an instance of UploadModelSource from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return UploadModelSource.parse_obj(obj)
+
+        _obj = UploadModelSource.parse_obj({"type": obj.get("type")})
+        return _obj

truefoundry/ml/autogen/client_README.md

@@ -78,7 +78,6 @@ Class | Method | HTTP request | Description
 *ExperimentsApi* | [**list_experiments_get**](truefoundry/ml/autogen/client/docs/ExperimentsApi.md#list_experiments_get) | **GET** /api/2.0/mlflow/experiments/list | List Experiments
 *ExperimentsApi* | [**put_privacy_type_put**](truefoundry/ml/autogen/client/docs/ExperimentsApi.md#put_privacy_type_put) | **PUT** /api/2.0/mlflow/experiments/privacy-type | Put Privacy Type
 *ExperimentsApi* | [**restore_experiment_post**](truefoundry/ml/autogen/client/docs/ExperimentsApi.md#restore_experiment_post) | **POST** /api/2.0/mlflow/experiments/restore | Restore Experiment
-*ExperimentsApi* | [**seed_list_experiment_get**](truefoundry/ml/autogen/client/docs/ExperimentsApi.md#seed_list_experiment_get) | **GET** /api/2.0/mlflow/experiments/seed/list | Seed List Experiment
 *ExperimentsApi* | [**set_experiment_tag_post**](truefoundry/ml/autogen/client/docs/ExperimentsApi.md#set_experiment_tag_post) | **POST** /api/2.0/mlflow/experiments/set-experiment-tag | Set Experiment Tag
 *ExperimentsApi* | [**update_experiment_post**](truefoundry/ml/autogen/client/docs/ExperimentsApi.md#update_experiment_post) | **POST** /api/2.0/mlflow/experiments/update | Update Experiment
 *HealthApi* | [**health_get**](truefoundry/ml/autogen/client/docs/HealthApi.md#health_get) | **GET** /health | Health
@@ -166,6 +165,7 @@ Class | Method | HTTP request | Description
  - [ArtifactType](truefoundry/ml/autogen/client/docs/ArtifactType.md)
  - [ArtifactVersionDto](truefoundry/ml/autogen/client/docs/ArtifactVersionDto.md)
  - [ArtifactVersionResponseDto](truefoundry/ml/autogen/client/docs/ArtifactVersionResponseDto.md)
+ - [ArtifactVersionSerializationFormat](truefoundry/ml/autogen/client/docs/ArtifactVersionSerializationFormat.md)
  - [ArtifactVersionStatus](truefoundry/ml/autogen/client/docs/ArtifactVersionStatus.md)
  - [AssistantMessage](truefoundry/ml/autogen/client/docs/AssistantMessage.md)
  - [AuthorizeUserForModelRequestDto](truefoundry/ml/autogen/client/docs/AuthorizeUserForModelRequestDto.md)
@@ -206,10 +206,12 @@ Class | Method | HTTP request | Description
  - [ExperimentIdRequestDto](truefoundry/ml/autogen/client/docs/ExperimentIdRequestDto.md)
  - [ExperimentResponseDto](truefoundry/ml/autogen/client/docs/ExperimentResponseDto.md)
  - [ExperimentTagDto](truefoundry/ml/autogen/client/docs/ExperimentTagDto.md)
+ - [ExternalModelSource](truefoundry/ml/autogen/client/docs/ExternalModelSource.md)
  - [FeatureDto](truefoundry/ml/autogen/client/docs/FeatureDto.md)
  - [FeatureValueType](truefoundry/ml/autogen/client/docs/FeatureValueType.md)
  - [FileInfoDto](truefoundry/ml/autogen/client/docs/FileInfoDto.md)
  - [FinalizeArtifactVersionRequestDto](truefoundry/ml/autogen/client/docs/FinalizeArtifactVersionRequestDto.md)
+ - [Framework](truefoundry/ml/autogen/client/docs/Framework.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)
@@ -227,6 +229,7 @@ Class | Method | HTTP request | Description
  - [ImageUrl](truefoundry/ml/autogen/client/docs/ImageUrl.md)
  - [InternalMetadata](truefoundry/ml/autogen/client/docs/InternalMetadata.md)
  - [LatestRunLogDto](truefoundry/ml/autogen/client/docs/LatestRunLogDto.md)
+ - [LibraryName](truefoundry/ml/autogen/client/docs/LibraryName.md)
  - [ListArtifactVersionsRequestDto](truefoundry/ml/autogen/client/docs/ListArtifactVersionsRequestDto.md)
  - [ListArtifactVersionsResponseDto](truefoundry/ml/autogen/client/docs/ListArtifactVersionsResponseDto.md)
  - [ListArtifactsRequestDto](truefoundry/ml/autogen/client/docs/ListArtifactsRequestDto.md)
@@ -248,7 +251,6 @@ Class | Method | HTTP request | Description
  - [ListModelsResponseDto](truefoundry/ml/autogen/client/docs/ListModelsResponseDto.md)
  - [ListRunArtifactsResponseDto](truefoundry/ml/autogen/client/docs/ListRunArtifactsResponseDto.md)
  - [ListRunLogsResponseDto](truefoundry/ml/autogen/client/docs/ListRunLogsResponseDto.md)
- - [ListSeedExperimentsResponseDto](truefoundry/ml/autogen/client/docs/ListSeedExperimentsResponseDto.md)
  - [LogBatchRequestDto](truefoundry/ml/autogen/client/docs/LogBatchRequestDto.md)
  - [LogMetricRequestDto](truefoundry/ml/autogen/client/docs/LogMetricRequestDto.md)
  - [LogParamRequestDto](truefoundry/ml/autogen/client/docs/LogParamRequestDto.md)
@@ -261,6 +263,7 @@ Class | Method | HTTP request | Description
  - [ModelResponseDto](truefoundry/ml/autogen/client/docs/ModelResponseDto.md)
  - [ModelSchemaDto](truefoundry/ml/autogen/client/docs/ModelSchemaDto.md)
  - [ModelVersionDto](truefoundry/ml/autogen/client/docs/ModelVersionDto.md)
+ - [ModelVersionManifest](truefoundry/ml/autogen/client/docs/ModelVersionManifest.md)
  - [ModelVersionResponseDto](truefoundry/ml/autogen/client/docs/ModelVersionResponseDto.md)
  - [MultiPartUploadDto](truefoundry/ml/autogen/client/docs/MultiPartUploadDto.md)
  - [MultiPartUploadResponseDto](truefoundry/ml/autogen/client/docs/MultiPartUploadResponseDto.md)
@@ -284,17 +287,24 @@ Class | Method | HTTP request | Description
  - [SetExperimentTagRequestDto](truefoundry/ml/autogen/client/docs/SetExperimentTagRequestDto.md)
  - [SetTagRequestDto](truefoundry/ml/autogen/client/docs/SetTagRequestDto.md)
  - [SignedURLDto](truefoundry/ml/autogen/client/docs/SignedURLDto.md)
+ - [Source](truefoundry/ml/autogen/client/docs/Source.md)
  - [Stop](truefoundry/ml/autogen/client/docs/Stop.md)
  - [StoreRunLogsRequestDto](truefoundry/ml/autogen/client/docs/StoreRunLogsRequestDto.md)
+ - [Subject](truefoundry/ml/autogen/client/docs/Subject.md)
+ - [SubjectType](truefoundry/ml/autogen/client/docs/SubjectType.md)
  - [SystemMessage](truefoundry/ml/autogen/client/docs/SystemMessage.md)
+ - [TensorflowFramework](truefoundry/ml/autogen/client/docs/TensorflowFramework.md)
  - [Text](truefoundry/ml/autogen/client/docs/Text.md)
  - [TextContentPart](truefoundry/ml/autogen/client/docs/TextContentPart.md)
+ - [TransformersFramework](truefoundry/ml/autogen/client/docs/TransformersFramework.md)
+ - [TruefoundryModelSource](truefoundry/ml/autogen/client/docs/TruefoundryModelSource.md)
  - [UpdateArtifactVersionRequestDto](truefoundry/ml/autogen/client/docs/UpdateArtifactVersionRequestDto.md)
  - [UpdateDatasetRequestDto](truefoundry/ml/autogen/client/docs/UpdateDatasetRequestDto.md)
  - [UpdateExperimentRequestDto](truefoundry/ml/autogen/client/docs/UpdateExperimentRequestDto.md)
  - [UpdateModelVersionRequestDto](truefoundry/ml/autogen/client/docs/UpdateModelVersionRequestDto.md)
  - [UpdateRunRequestDto](truefoundry/ml/autogen/client/docs/UpdateRunRequestDto.md)
  - [UpdateRunResponseDto](truefoundry/ml/autogen/client/docs/UpdateRunResponseDto.md)
+ - [UploadModelSource](truefoundry/ml/autogen/client/docs/UploadModelSource.md)
  - [Url](truefoundry/ml/autogen/client/docs/Url.md)
  - [UserMessage](truefoundry/ml/autogen/client/docs/UserMessage.md)
  - [ValidationError](truefoundry/ml/autogen/client/docs/ValidationError.md)

truefoundry/ml/autogen/entities/artifacts.py

@@ -1,13 +1,13 @@
 # generated by datamodel-codegen:
 #   filename:  artifacts.json
-#   timestamp: 2024-09-12T13:40:36+00:00
+#   timestamp: 2024-10-25T06:52:16+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
+from truefoundry.pydantic_v1 import BaseModel, Field, conint, constr
 
 
 class Agent(BaseModel):
@@ -49,6 +49,17 @@ class AgentWithFQN(Agent):
     fqn: str
 
 
+class BaseArtifactVersion(BaseModel):
+    description: Optional[constr(max_length=512)] = Field(
+        None,
+        description="+label=Description\n+docs=Description of the artifact version",
+    )
+    metadata: Dict[str, Any] = Field(
+        ...,
+        description="+label=Metadata\n+docs=Metadata for the model version\n+usage=Metadata for the model version\n+uiType=JsonInput",
+    )
+
+
 class MimeType(str, Enum):
     """
     +label=MIME Type
@@ -78,6 +89,51 @@ class BlobStorageReference(BaseModel):
     )
 
 
+class ExternalModelSource(BaseModel):
+    """
+    +label=External
+    """
+
+    type: Literal["external"] = Field(
+        ...,
+        description="+label=Type\n+usage=Type of the source\n+value=external\n+type=External",
+    )
+    uri: str = Field(..., description="+label=URI\n+usage=URI of the model source")
+
+
+class FastAIFramework(BaseModel):
+    """
+    +docs=FastAI framework for the model version
+    +label=FastAI
+    """
+
+    type: Literal["fastai"] = Field(
+        ..., description="+label=Type\n+usage=Type of the framework\n+value=fastai"
+    )
+
+
+class GluonFramework(BaseModel):
+    """
+    +docs=Gluon framework for the model version
+    +label=Gluon
+    """
+
+    type: Literal["gluon"] = Field(
+        ..., description="+label=Type\n+usage=Type of the framework\n+value=gluon"
+    )
+
+
+class H2OFramework(BaseModel):
+    """
+    +docs=H2O framework for the model version
+    +label=H2O
+    """
+
+    type: Literal["h2o"] = Field(
+        ..., description="+label=Type\n+usage=Type of the framework\n+value=h2o"
+    )
+
+
 class ImageUrl(BaseModel):
     """
     +label=URL for the image
@@ -108,6 +164,28 @@ class ImageContentPart(BaseModel):
     )
 
 
+class KerasFramework(BaseModel):
+    """
+    +docs=Keras framework for the model version
+    +label=Keras
+    """
+
+    type: Literal["keras"] = Field(
+        ..., description="+label=Type\n+usage=Type of the framework\n+value=keras"
+    )
+
+
+class LightGBMFramework(BaseModel):
+    """
+    +docs=LightGBM framework for the model version
+    +label=LightGBM
+    """
+
+    type: Literal["lightgbm"] = Field(
+        ..., description="+label=Type\n+usage=Type of the framework\n+value=lightgbm"
+    )
+
+
 class Parameters(BaseModel):
     """
     +usage=Parameters for the provider
@@ -148,6 +226,61 @@ class ModelConfiguration(BaseModel):
     )
 
 
+class PaddleFramework(BaseModel):
+    """
+    +docs=PaddlePaddle framework for the model version
+    +label=Paddle
+    """
+
+    type: Literal["paddle"] = Field(
+        ..., description="+label=Type\n+usage=Type of the framework\n+value=paddle"
+    )
+
+
+class PyTorchFramework(BaseModel):
+    """
+    +docs=PyTorch framework for the model version
+    +label=PyTorch
+    """
+
+    type: Literal["pytorch"] = Field(
+        ..., description="+label=Type\n+usage=Type of the framework\n+value=pytorch"
+    )
+
+
+class SklearnFramework(BaseModel):
+    """
+    +docs=Scikit-learn framework for the model version
+    +label=Sklearn
+    """
+
+    type: Literal["sklearn"] = Field(
+        ..., description="+label=Type\n+usage=Type of the framework\n+value=sklearn"
+    )
+
+
+class SpaCyFramework(BaseModel):
+    """
+    +docs=spaCy framework for the model version
+    +label=SpaCy
+    """
+
+    type: Literal["spacy"] = Field(
+        ..., description="+label=Type\n+usage=Type of the framework\n+value=spacy"
+    )
+
+
+class StatsModelsFramework(BaseModel):
+    """
+    +docs=StatsModels framework for the model version
+    +label=StatsModels
+    """
+
+    type: Literal["statsmodels"] = Field(
+        ..., description="+label=Type\n+usage=Type of the framework\n+value=statsmodels"
+    )
+
+
 class SystemMessage(BaseModel):
     """
     +usage=System message
@@ -164,6 +297,17 @@ class SystemMessage(BaseModel):
     )
 
 
+class TensorFlowFramework(BaseModel):
+    """
+    +docs=TensorFlow framework for the model version
+    +label=TensorFlow
+    """
+
+    type: Literal["tensorflow"] = Field(
+        ..., description="+label=Type\n+usage=Type of the framework\n+value=tensorflow"
+    )
+
+
 class TextContentPart(BaseModel):
     """
     +usage=Text content
@@ -174,6 +318,53 @@ class TextContentPart(BaseModel):
     text: Union[constr(regex=r"^.[\s\S]*$"), BlobStorageReference]
 
 
+class LibraryName(str, Enum):
+    """
+    +label=Library Name
+    +usage=Name of the library for the framework
+    """
+
+    transformers = "transformers"
+    sentence_transformers = "sentence-transformers"
+    diffusers = "diffusers"
+
+
+class TransformersFramework(BaseModel):
+    """
+    +docs=Transformers framework for the model version
+    +label=Transformers
+    +value=transformers
+    """
+
+    type: Literal["transformers"] = Field(
+        ...,
+        description="+label=Type\n+usage=Type of the framework\n+value=transformers",
+    )
+    library_name: Optional[LibraryName] = Field(
+        "transformers",
+        description="+label=Library Name\n+usage=Name of the library for the framework",
+    )
+    pipeline_tag: Optional[str] = Field(
+        None,
+        description="+label=Pipeline Tag\n+usage=Pipeline tag\n+docs=Pipeline tag for the framework",
+    )
+
+
+class TruefoundryModelSource(BaseModel):
+    """
+    +usage=Source for the Model
+    +label=Upload
+    """
+
+    type: Literal["truefoundry"] = Field(
+        ...,
+        description="+usage=Type of the source\n+type=Upload\n+value=truefoundry\n+label=Upload",
+    )
+    uri: str = Field(
+        ..., description='+label=URI\n+uiType=UploadInput\n+uiProps={"hideClear":true}'
+    )
+
+
 class UserMessage(BaseModel):
     """
     +usage=User message
@@ -194,6 +385,17 @@ class UserMessage(BaseModel):
     )
 
 
+class XGBoostFramework(BaseModel):
+    """
+    +docs=XGBoost framework for the model version
+    +label=XGBoost
+    """
+
+    type: Literal["xgboost"] = Field(
+        ..., description="+label=Type\n+usage=Type of the framework\n+value=xgboost"
+    )
+
+
 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(
@@ -275,6 +477,36 @@ class ChatPrompt(BasePrompt):
     )
 
 
+class ModelVersion(BaseArtifactVersion):
+    type: Literal["model-version"] = Field(
+        ..., description='+label=Type\n+usage=Model Version\n+value="model-version"'
+    )
+    source: Union[TruefoundryModelSource, ExternalModelSource] = Field(
+        ...,
+        description="+label=Model Source\n+usage=Source for the model version\n+uiType=Group",
+    )
+    framework: Optional[
+        Union[
+            TransformersFramework,
+            TensorFlowFramework,
+            SklearnFramework,
+            PyTorchFramework,
+            KerasFramework,
+            XGBoostFramework,
+            LightGBMFramework,
+            FastAIFramework,
+            H2OFramework,
+            SpaCyFramework,
+            StatsModelsFramework,
+            GluonFramework,
+            PaddleFramework,
+        ]
+    ] = Field(
+        None, description="+label=Framework\n+usage=Framework for the model version"
+    )
+    step: conint(ge=0) = Field(0, description="+label=Step")
+
+
 class AgentApp(BaseModel):
     type: Literal["agent-app"] = Field(..., description="+value=agent-app")
     tools: List[AgentOpenAPIToolWithFQN]
@@ -282,5 +514,5 @@ class AgentApp(BaseModel):
     root_agent: constr(min_length=1)
 
 
-class Artifacts(BaseModel):
-    __root__: Union[ChatPrompt, AgentOpenAPITool, Agent, AgentApp]
+class ArtifactsVersion(BaseModel):
+    __root__: Union[ChatPrompt, AgentOpenAPITool, Agent, AgentApp, ModelVersion]

truefoundry/ml/mlfoundry_api.py

@@ -335,7 +335,6 @@ class MlFoundry:
         tags.update(_get_internal_env_vars_values())
         _run = self._runs_api.create_run_post(
             CreateRunRequestDto(
-                user_id="unknown",  # This does not matter, because on server we use the id from token
                 start_time=int(
                     time.time() * 1000
                 ),  # TODO (chiragjn): computing start time should be on server side!

{truefoundry-0.4.4rc9.dist-info → truefoundry-0.4.4rc11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: truefoundry
-Version: 0.4.4rc9
+Version: 0.4.4rc11
 Summary: Truefoundry CLI
 Author: Abhishek Choudhary
 Author-email: abhishek@truefoundry.com