truefoundry 0.5.9__py3-none-any.whl → 0.5.10rc2__py3-none-any.whl

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -0,0 +1,93 @@
+# coding: utf-8
+
+"""
+TrueFoundry ML API
+
+No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+The version of the OpenAPI document: 1.0.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.ml.autogen.client.models.artifact_path import ArtifactPath
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, conlist, validator
+
+
+class LocalArtifactSource(BaseModel):
+    """
+    LocalArtifactSource
+    """
+
+    type: StrictStr = Field(
+        default=..., description="+label=Type +usage=Type of the source +value=local"
+    )
+    paths: conlist(ArtifactPath) = Field(
+        default=...,
+        description="+label=Paths +usage=Array of ArtifactPath objects representing the source and destination paths",
+    )
+    __properties = ["type", "paths"]
+
+    @validator("type")
+    def type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in ("local",):
+            raise ValueError("must be one of enum values ('local')")
+        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) -> LocalArtifactSource:
+        """Create an instance of LocalArtifactSource 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)
+        # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of each item in paths (list)
+        _items = []
+        if self.paths:
+            for _item in self.paths:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict["paths"] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> LocalArtifactSource:
+        """Create an instance of LocalArtifactSource from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return LocalArtifactSource.parse_obj(obj)
+
+        _obj = LocalArtifactSource.parse_obj(
+            {
+                "type": obj.get("type"),
+                "paths": [ArtifactPath.from_dict(_item) for _item in obj.get("paths")]
+                if obj.get("paths") is not None
+                else None,
+            }
+        )
+        return _obj

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

@@ -0,0 +1,80 @@
+# coding: utf-8
+
+"""
+TrueFoundry ML API
+
+No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+The version of the OpenAPI document: 1.0.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 LocalModelSource(BaseModel):
+    """
+    LocalModelSource
+    """
+
+    type: StrictStr = Field(
+        default=..., description="+label=Type +usage=Type of the source +value=local"
+    )
+    file_or_folder: StrictStr = Field(
+        default=...,
+        description="+label=Path to the model file or folder +usage=Path to the model file or folder",
+    )
+    __properties = ["type", "file_or_folder"]
+
+    @validator("type")
+    def type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in ("local",):
+            raise ValueError("must be one of enum values ('local')")
+        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) -> LocalModelSource:
+        """Create an instance of LocalModelSource 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) -> LocalModelSource:
+        """Create an instance of LocalModelSource from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return LocalModelSource.parse_obj(obj)
+
+        _obj = LocalModelSource.parse_obj(
+            {"type": obj.get("type"), "file_or_folder": obj.get("file_or_folder")}
+        )
+        return _obj

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -18,9 +18,14 @@ import pprint
 import re  # noqa: F401
 from typing import TYPE_CHECKING, Any, List, Optional, Union
 
+from truefoundry.ml.autogen.client.models.agent_manifest import AgentManifest
+from truefoundry.ml.autogen.client.models.agent_open_api_tool_manifest import (
+    AgentOpenAPIToolManifest,
+)
 from truefoundry.ml.autogen.client.models.artifact_version_manifest import (
     ArtifactVersionManifest,
 )
+from truefoundry.ml.autogen.client.models.chat_prompt_manifest import ChatPromptManifest
 from truefoundry.ml.autogen.client.models.model_version_manifest import (
     ModelVersionManifest,
 )
@@ -31,7 +36,13 @@ from truefoundry.pydantic_v1 import (
     validator,
 )
 
-MANIFEST_ANY_OF_SCHEMAS = ["ArtifactVersionManifest", "ModelVersionManifest"]
+MANIFEST_ONE_OF_SCHEMAS = [
+    "AgentManifest",
+    "AgentOpenAPIToolManifest",
+    "ArtifactVersionManifest",
+    "ChatPromptManifest",
+    "ModelVersionManifest",
+]
 
 
 class Manifest(BaseModel):
@@ -40,18 +51,32 @@ class Manifest(BaseModel):
     """
 
     # data type: ModelVersionManifest
-    anyof_schema_1_validator: Optional[ModelVersionManifest] = None
+    oneof_schema_1_validator: Optional[ModelVersionManifest] = None
+    # data type: ChatPromptManifest
+    oneof_schema_2_validator: Optional[ChatPromptManifest] = None
+    # data type: AgentOpenAPIToolManifest
+    oneof_schema_3_validator: Optional[AgentOpenAPIToolManifest] = None
+    # data type: AgentManifest
+    oneof_schema_4_validator: Optional[AgentManifest] = None
     # data type: ArtifactVersionManifest
-    anyof_schema_2_validator: Optional[ArtifactVersionManifest] = None
+    oneof_schema_5_validator: Optional[ArtifactVersionManifest] = None
     if TYPE_CHECKING:
-        actual_instance: Union[ArtifactVersionManifest, ModelVersionManifest]
+        actual_instance: Union[
+            AgentManifest,
+            AgentOpenAPIToolManifest,
+            ArtifactVersionManifest,
+            ChatPromptManifest,
+            ModelVersionManifest,
+        ]
     else:
         actual_instance: Any
-    any_of_schemas: List[str] = Field(MANIFEST_ANY_OF_SCHEMAS, const=True)
+    one_of_schemas: List[str] = Field(MANIFEST_ONE_OF_SCHEMAS, const=True)
 
     class Config:
         validate_assignment = True
 
+    discriminator_value_class_map = {}
+
     def __init__(self, *args, **kwargs) -> None:
         if args:
             if len(args) > 1:
@@ -67,29 +92,55 @@ class Manifest(BaseModel):
             super().__init__(**kwargs)
 
     @validator("actual_instance")
-    def actual_instance_must_validate_anyof(cls, v):
+    def actual_instance_must_validate_oneof(cls, v):
         instance = Manifest.construct()
         error_messages = []
+        match = 0
         # validate data type: ModelVersionManifest
         if not isinstance(v, ModelVersionManifest):
             error_messages.append(
                 f"Error! Input type `{type(v)}` is not `ModelVersionManifest`"
             )
         else:
-            return v
-
+            match += 1
+        # validate data type: ChatPromptManifest
+        if not isinstance(v, ChatPromptManifest):
+            error_messages.append(
+                f"Error! Input type `{type(v)}` is not `ChatPromptManifest`"
+            )
+        else:
+            match += 1
+        # validate data type: AgentOpenAPIToolManifest
+        if not isinstance(v, AgentOpenAPIToolManifest):
+            error_messages.append(
+                f"Error! Input type `{type(v)}` is not `AgentOpenAPIToolManifest`"
+            )
+        else:
+            match += 1
+        # validate data type: AgentManifest
+        if not isinstance(v, AgentManifest):
+            error_messages.append(
+                f"Error! Input type `{type(v)}` is not `AgentManifest`"
+            )
+        else:
+            match += 1
         # validate data type: ArtifactVersionManifest
         if not isinstance(v, ArtifactVersionManifest):
             error_messages.append(
                 f"Error! Input type `{type(v)}` is not `ArtifactVersionManifest`"
             )
         else:
-            return v
-
-        if error_messages:
+            match += 1
+        if match > 1:
+            # more than 1 match
+            raise ValueError(
+                "Multiple matches found when setting `actual_instance` in Manifest with oneOf schemas: AgentManifest, AgentOpenAPIToolManifest, ArtifactVersionManifest, ChatPromptManifest, ModelVersionManifest. Details: "
+                + ", ".join(error_messages)
+            )
+        elif match == 0:
             # no match
             raise ValueError(
-                "No match found when setting the actual_instance in Manifest with anyOf schemas: ArtifactVersionManifest, ModelVersionManifest. Details: "
+                "No match found when setting `actual_instance` in Manifest with oneOf schemas: AgentManifest, AgentOpenAPIToolManifest, ArtifactVersionManifest, ChatPromptManifest, ModelVersionManifest. Details: "
                 + ", ".join(error_messages)
             )
         else:
@@ -104,23 +155,49 @@ class Manifest(BaseModel):
         """Returns the object represented by the json string"""
         instance = Manifest.construct()
         error_messages = []
-        # anyof_schema_1_validator: Optional[ModelVersionManifest] = None
+        match = 0
+
+        # deserialize data into ModelVersionManifest
         try:
             instance.actual_instance = ModelVersionManifest.from_json(json_str)
-            return instance
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into ChatPromptManifest
+        try:
+            instance.actual_instance = ChatPromptManifest.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into AgentOpenAPIToolManifest
+        try:
+            instance.actual_instance = AgentOpenAPIToolManifest.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into AgentManifest
+        try:
+            instance.actual_instance = AgentManifest.from_json(json_str)
+            match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
-        # anyof_schema_2_validator: Optional[ArtifactVersionManifest] = None
+        # deserialize data into ArtifactVersionManifest
         try:
             instance.actual_instance = ArtifactVersionManifest.from_json(json_str)
-            return instance
+            match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
 
-        if error_messages:
+        if match > 1:
+            # more than 1 match
+            raise ValueError(
+                "Multiple matches found when deserializing the JSON string into Manifest with oneOf schemas: AgentManifest, AgentOpenAPIToolManifest, ArtifactVersionManifest, ChatPromptManifest, ModelVersionManifest. Details: "
+                + ", ".join(error_messages)
+            )
+        elif match == 0:
             # no match
             raise ValueError(
-                "No match found when deserializing the JSON string into Manifest with anyOf schemas: ArtifactVersionManifest, ModelVersionManifest. Details: "
+                "No match found when deserializing the JSON string into Manifest with oneOf schemas: AgentManifest, AgentOpenAPIToolManifest, ArtifactVersionManifest, ChatPromptManifest, ModelVersionManifest. Details: "
                 + ", ".join(error_messages)
             )
         else:
@@ -140,10 +217,10 @@ class Manifest(BaseModel):
     def to_dict(self) -> dict:
         """Returns the dict representation of the actual instance"""
         if self.actual_instance is None:
-            return "null"
+            return None
 
-        to_json = getattr(self.actual_instance, "to_json", None)
-        if callable(to_json):
+        to_dict = getattr(self.actual_instance, "to_dict", None)
+        if callable(to_dict):
             return self.actual_instance.to_dict()
         else:
             # primitive type