truefoundry 0.3.2__py3-none-any.whl → 0.4.0.dev0__py3-none-any.whl

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

@@ -0,0 +1,75 @@
+# 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.ml.autogen.client.models.artifact_dto import ArtifactDto
+from truefoundry.pydantic_v1 import BaseModel, Field
+
+
+class ArtifactResponseDto(BaseModel):
+    """
+    ArtifactResponseDto
+    """
+
+    artifact: ArtifactDto = Field(...)
+    __properties = ["artifact"]
+
+    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) -> ArtifactResponseDto:
+        """Create an instance of ArtifactResponseDto 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 artifact
+        if self.artifact:
+            _dict["artifact"] = self.artifact.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> ArtifactResponseDto:
+        """Create an instance of ArtifactResponseDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return ArtifactResponseDto.parse_obj(obj)
+
+        _obj = ArtifactResponseDto.parse_obj(
+            {
+                "artifact": ArtifactDto.from_dict(obj.get("artifact"))
+                if obj.get("artifact") is not None
+                else None
+            }
+        )
+        return _obj

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

@@ -0,0 +1,39 @@
+# 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
+
+import json
+import re  # noqa: F401
+
+from aenum import Enum
+
+
+class ArtifactType(str, Enum):
+    """
+    An enumeration.
+    """
+
+    """
+    allowed enum values
+    """
+    ARTIFACT = "artifact"
+    MODEL = "model"
+    PLOT = "plot"
+    IMAGE = "image"
+    CHAT_PROMPT = "chat_prompt"
+    OPENAPI_TOOL = "openapi_tool"
+    AGENT = "agent"
+
+    @classmethod
+    def from_json(cls, json_str: str) -> ArtifactType:
+        """Create an instance of ArtifactType from a JSON string"""
+        return ArtifactType(json.loads(json_str))

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

@@ -0,0 +1,141 @@
+# 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 datetime import datetime
+from typing import Any, Dict, Optional
+
+from truefoundry.ml.autogen.client.models.artifact_version_status import (
+    ArtifactVersionStatus,
+)
+from truefoundry.ml.autogen.client.models.internal_metadata import InternalMetadata
+from truefoundry.pydantic_v1 import BaseModel, StrictInt, StrictStr
+
+
+class ArtifactVersionDto(BaseModel):
+    """
+    ArtifactVersionDto
+    """
+
+    id: Optional[StrictStr] = None
+    artifact_id: Optional[StrictStr] = None
+    version: Optional[StrictInt] = None
+    fqn: Optional[StrictStr] = None
+    artifact_storage_root: Optional[StrictStr] = None
+    artifact_metadata: Optional[Dict[str, Any]] = None
+    description: Optional[StrictStr] = None
+    status: Optional[ArtifactVersionStatus] = None
+    step: Optional[StrictInt] = 0
+    created_by: Optional[StrictStr] = None
+    created_at: Optional[datetime] = None
+    updated_at: Optional[datetime] = None
+    artifact_fqn: Optional[StrictStr] = None
+    data_path: Optional[StrictStr] = None
+    usage_code_snippet: Optional[StrictStr] = None
+    experiment_id: Optional[StrictStr] = None
+    run_id: Optional[StrictStr] = None
+    artifact_name: Optional[StrictStr] = None
+    internal_metadata: Optional[InternalMetadata] = None
+    artifact_size: Optional[StrictInt] = None
+    __properties = [
+        "id",
+        "artifact_id",
+        "version",
+        "fqn",
+        "artifact_storage_root",
+        "artifact_metadata",
+        "description",
+        "status",
+        "step",
+        "created_by",
+        "created_at",
+        "updated_at",
+        "artifact_fqn",
+        "data_path",
+        "usage_code_snippet",
+        "experiment_id",
+        "run_id",
+        "artifact_name",
+        "internal_metadata",
+        "artifact_size",
+    ]
+
+    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) -> ArtifactVersionDto:
+        """Create an instance of ArtifactVersionDto 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 internal_metadata
+        if self.internal_metadata:
+            _dict["internal_metadata"] = self.internal_metadata.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> ArtifactVersionDto:
+        """Create an instance of ArtifactVersionDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return ArtifactVersionDto.parse_obj(obj)
+
+        _obj = ArtifactVersionDto.parse_obj(
+            {
+                "id": obj.get("id"),
+                "artifact_id": obj.get("artifact_id"),
+                "version": obj.get("version"),
+                "fqn": obj.get("fqn"),
+                "artifact_storage_root": obj.get("artifact_storage_root"),
+                "artifact_metadata": obj.get("artifact_metadata"),
+                "description": obj.get("description"),
+                "status": obj.get("status"),
+                "step": obj.get("step") if obj.get("step") is not None else 0,
+                "created_by": obj.get("created_by"),
+                "created_at": obj.get("created_at"),
+                "updated_at": obj.get("updated_at"),
+                "artifact_fqn": obj.get("artifact_fqn"),
+                "data_path": obj.get("data_path"),
+                "usage_code_snippet": obj.get("usage_code_snippet"),
+                "experiment_id": obj.get("experiment_id"),
+                "run_id": obj.get("run_id"),
+                "artifact_name": obj.get("artifact_name"),
+                "internal_metadata": InternalMetadata.from_dict(
+                    obj.get("internal_metadata")
+                )
+                if obj.get("internal_metadata") is not None
+                else None,
+                "artifact_size": obj.get("artifact_size"),
+            }
+        )
+        return _obj

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

@@ -0,0 +1,77 @@
+# 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.ml.autogen.client.models.artifact_version_dto import ArtifactVersionDto
+from truefoundry.pydantic_v1 import BaseModel, Field
+
+
+class ArtifactVersionResponseDto(BaseModel):
+    """
+    ArtifactVersionResponseDto
+    """
+
+    artifact_version: ArtifactVersionDto = Field(...)
+    __properties = ["artifact_version"]
+
+    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) -> ArtifactVersionResponseDto:
+        """Create an instance of ArtifactVersionResponseDto 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 artifact_version
+        if self.artifact_version:
+            _dict["artifact_version"] = self.artifact_version.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> ArtifactVersionResponseDto:
+        """Create an instance of ArtifactVersionResponseDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return ArtifactVersionResponseDto.parse_obj(obj)
+
+        _obj = ArtifactVersionResponseDto.parse_obj(
+            {
+                "artifact_version": ArtifactVersionDto.from_dict(
+                    obj.get("artifact_version")
+                )
+                if obj.get("artifact_version") is not None
+                else None
+            }
+        )
+        return _obj

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

@@ -0,0 +1,35 @@
+# 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
+
+import json
+import re  # noqa: F401
+
+from aenum import Enum
+
+
+class ArtifactVersionStatus(str, Enum):
+    """
+    An enumeration.
+    """
+
+    """
+    allowed enum values
+    """
+    COMMITTED = "COMMITTED"
+    DELETED = "DELETED"
+    HARD_DELETED = "HARD_DELETED"
+
+    @classmethod
+    def from_json(cls, json_str: str) -> ArtifactVersionStatus:
+        """Create an instance of ArtifactVersionStatus from a JSON string"""
+        return ArtifactVersionStatus(json.loads(json_str))

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

@@ -0,0 +1,89 @@
+# 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 typing import Optional
+
+from truefoundry.ml.autogen.client.models.content import Content
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
+
+
+class AssistantMessage(BaseModel):
+    """
+    +usage=Assistant message +docs=Defines the structure of an assistant message, including role and content +label=Assistant Message  # noqa: E501
+    """
+
+    role: StrictStr = Field(...)
+    content: Content = Field(...)
+    name: Optional[StrictStr] = Field(
+        default=None, description="+label=Name +usage=Name of the assistant message"
+    )
+    __properties = ["role", "content", "name"]
+
+    @validator("role")
+    def role_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in ("assistant"):
+            raise ValueError("must be one of enum values ('assistant')")
+        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) -> AssistantMessage:
+        """Create an instance of AssistantMessage 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 content
+        if self.content:
+            _dict["content"] = self.content.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> AssistantMessage:
+        """Create an instance of AssistantMessage from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return AssistantMessage.parse_obj(obj)
+
+        _obj = AssistantMessage.parse_obj(
+            {
+                "role": obj.get("role"),
+                "content": Content.from_dict(obj.get("content"))
+                if obj.get("content") is not None
+                else None,
+                "name": obj.get("name"),
+            }
+        )
+        return _obj

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

@@ -0,0 +1,69 @@
+# 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 typing import Optional
+
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr
+
+
+class AuthorizeUserForModelRequestDto(BaseModel):
+    """
+    AuthorizeUserForModelRequestDto
+    """
+
+    id: StrictStr = Field(...)
+    role: Optional[StrictStr] = None
+    __properties = ["id", "role"]
+
+    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) -> AuthorizeUserForModelRequestDto:
+        """Create an instance of AuthorizeUserForModelRequestDto 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) -> AuthorizeUserForModelRequestDto:
+        """Create an instance of AuthorizeUserForModelRequestDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return AuthorizeUserForModelRequestDto.parse_obj(obj)
+
+        _obj = AuthorizeUserForModelRequestDto.parse_obj(
+            {"id": obj.get("id"), "role": obj.get("role")}
+        )
+        return _obj

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

@@ -0,0 +1,69 @@
+# 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 typing import Optional
+
+from truefoundry.pydantic_v1 import BaseModel, StrictStr
+
+
+class AuthorizeUserForModelVersionRequestDto(BaseModel):
+    """
+    AuthorizeUserForModelVersionRequestDto
+    """
+
+    id: Optional[StrictStr] = None
+    role: Optional[StrictStr] = None
+    __properties = ["id", "role"]
+
+    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) -> AuthorizeUserForModelVersionRequestDto:
+        """Create an instance of AuthorizeUserForModelVersionRequestDto 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) -> AuthorizeUserForModelVersionRequestDto:
+        """Create an instance of AuthorizeUserForModelVersionRequestDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return AuthorizeUserForModelVersionRequestDto.parse_obj(obj)
+
+        _obj = AuthorizeUserForModelVersionRequestDto.parse_obj(
+            {"id": obj.get("id"), "role": obj.get("role")}
+        )
+        return _obj

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

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