polyaxon-sdk 1.18.0rc7__py2.py3-none-any.whl → 2.1.6__py2.py3-none-any.whl

polyaxon_sdk/models/v1_event_kind.py

@@ -1,158 +1,65 @@
-#!/usr/bin/python
-#
-# Copyright 2018-2022 Polyaxon, Inc.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#      http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
 # coding: utf-8
 
 """
     Polyaxon SDKs and REST API specification.
 
-    Polyaxon SDKs and REST API specification.  # noqa: E501
+         # noqa: E501
 
-    The version of the OpenAPI document: 1.18.0
+    The version of the OpenAPI document: 2.1.6
     Contact: contact@polyaxon.com
-    Generated by: https://openapi-generator.tech
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
 """
 
 
-try:
-    from inspect import getfullargspec
-except ImportError:
-    from inspect import getargspec as getfullargspec
+from inspect import getfullargspec
 import pprint
 import re  # noqa: F401
-import six
+from aenum import Enum, no_arg
 
-from polyaxon_sdk.configuration import Configuration
 
 
-class V1EventKind(object):
-    """NOTE: This class is auto generated by OpenAPI Generator.
-    Ref: https://openapi-generator.tech
 
-    Do not edit the class manually.
-    """
 
+class V1EventKind(str, Enum):
     """
-    allowed enum values
+    V1EventKind
     """
-    RUN_STATUS_CREATED = "run_status_created"
-    RUN_STATUS_RESUMING = "run_status_resuming"
-    RUN_STATUS_COMPILED = "run_status_compiled"
-    RUN_STATUS_ON_SCHEDULE = "run_status_on_schedule"
-    RUN_STATUS_QUEUED = "run_status_queued"
-    RUN_STATUS_SCHEDULED = "run_status_scheduled"
-    RUN_STATUS_STARTING = "run_status_starting"
-    RUN_STATUS_RUNNING = "run_status_running"
-    RUN_STATUS_PROCESSING = "run_status_processing"
-    RUN_STATUS_STOPPING = "run_status_stopping"
-    RUN_STATUS_FAILED = "run_status_failed"
-    RUN_STATUS_STOPPED = "run_status_stopped"
-    RUN_STATUS_SUCCEEDED = "run_status_succeeded"
-    RUN_STATUS_SKIPPED = "run_status_skipped"
-    RUN_STATUS_WARNING = "run_status_warning"
-    RUN_STATUS_UNSCHEDULABLE = "run_status_unschedulable"
-    RUN_STATUS_UPSTREAM_FAILED = "run_status_upstream_failed"
-    RUN_STATUS_RETRYING = "run_status_retrying"
-    RUN_STATUS_UNKNOWN = "run_status_unknown"
-    RUN_STATUS_DONE = "run_status_done"
-    RUN_APPROVED_ACTOR = "run_approved_actor"
-    RUN_INVALIDATED_ACTOR = "run_invalidated_actor"
-    RUN_NEW_ARTIFACTS = "run_new_artifacts"
-    CONNECTION_GIT_COMMIT = "connection_git_commit"
-    CONNECTION_DATASET_VERSION = "connection_dataset_version"
-    CONNECTION_REGISTRY_IMAGE = "connection_registry_image"
-    ALERT_INFO = "alert_info"
-    ALERT_WARNING = "alert_warning"
-    ALERT_CRITICAL = "alert_critical"
-    MODEL_VERSION_NEW_METRIC = "model_version_new_metric"
-    PROJECT_CUSTOM_EVENT = "project_custom_event"
-    ORG_CUSTOM_EVENT = "org_custom_event"
-
-    allowable_values = [RUN_STATUS_CREATED, RUN_STATUS_RESUMING, RUN_STATUS_COMPILED, RUN_STATUS_ON_SCHEDULE, RUN_STATUS_QUEUED, RUN_STATUS_SCHEDULED, RUN_STATUS_STARTING, RUN_STATUS_RUNNING, RUN_STATUS_PROCESSING, RUN_STATUS_STOPPING, RUN_STATUS_FAILED, RUN_STATUS_STOPPED, RUN_STATUS_SUCCEEDED, RUN_STATUS_SKIPPED, RUN_STATUS_WARNING, RUN_STATUS_UNSCHEDULABLE, RUN_STATUS_UPSTREAM_FAILED, RUN_STATUS_RETRYING, RUN_STATUS_UNKNOWN, RUN_STATUS_DONE, RUN_APPROVED_ACTOR, RUN_INVALIDATED_ACTOR, RUN_NEW_ARTIFACTS, CONNECTION_GIT_COMMIT, CONNECTION_DATASET_VERSION, CONNECTION_REGISTRY_IMAGE, ALERT_INFO, ALERT_WARNING, ALERT_CRITICAL, MODEL_VERSION_NEW_METRIC, PROJECT_CUSTOM_EVENT, ORG_CUSTOM_EVENT]  # noqa: E501
 
     """
-    Attributes:
-      openapi_types (dict): The key is attribute name
-                            and the value is attribute type.
-      attribute_map (dict): The key is attribute name
-                            and the value is json key in definition.
+    allowed enum values
     """
-    openapi_types = {
-    }
-
-    attribute_map = {
-    }
-
-    def __init__(self, local_vars_configuration=None):  # noqa: E501
-        """V1EventKind - a model defined in OpenAPI"""  # noqa: E501
-        if local_vars_configuration is None:
-            local_vars_configuration = Configuration.get_default_copy()
-        self.local_vars_configuration = local_vars_configuration
-        self.discriminator = None
-
-    def to_dict(self, serialize=False):
-        """Returns the model properties as a dict"""
-        result = {}
-
-        def convert(x):
-            if hasattr(x, "to_dict"):
-                args = getfullargspec(x.to_dict).args
-                if len(args) == 1:
-                    return x.to_dict()
-                else:
-                    return x.to_dict(serialize)
-            else:
-                return x
-
-        for attr, _ in six.iteritems(self.openapi_types):
-            value = getattr(self, attr)
-            attr = self.attribute_map.get(attr, attr) if serialize else attr
-            if isinstance(value, list):
-                result[attr] = list(map(
-                    lambda x: convert(x),
-                    value
-                ))
-            elif isinstance(value, dict):
-                result[attr] = dict(map(
-                    lambda item: (item[0], convert(item[1])),
-                    value.items()
-                ))
-            else:
-                result[attr] = convert(value)
-
-        return result
-
-    def to_str(self):
-        """Returns the string representation of the model"""
-        return pprint.pformat(self.to_dict())
-
-    def __repr__(self):
-        """For `print` and `pprint`"""
-        return self.to_str()
-
-    def __eq__(self, other):
-        """Returns true if both objects are equal"""
-        if not isinstance(other, V1EventKind):
-            return False
-
-        return self.to_dict() == other.to_dict()
-
-    def __ne__(self, other):
-        """Returns true if both objects are not equal"""
-        if not isinstance(other, V1EventKind):
-            return True
+    RUN_STATUS_CREATED = 'run_status_created'
+    RUN_STATUS_RESUMING = 'run_status_resuming'
+    RUN_STATUS_COMPILED = 'run_status_compiled'
+    RUN_STATUS_ON_SCHEDULE = 'run_status_on_schedule'
+    RUN_STATUS_QUEUED = 'run_status_queued'
+    RUN_STATUS_SCHEDULED = 'run_status_scheduled'
+    RUN_STATUS_STARTING = 'run_status_starting'
+    RUN_STATUS_RUNNING = 'run_status_running'
+    RUN_STATUS_PROCESSING = 'run_status_processing'
+    RUN_STATUS_STOPPING = 'run_status_stopping'
+    RUN_STATUS_FAILED = 'run_status_failed'
+    RUN_STATUS_STOPPED = 'run_status_stopped'
+    RUN_STATUS_SUCCEEDED = 'run_status_succeeded'
+    RUN_STATUS_SKIPPED = 'run_status_skipped'
+    RUN_STATUS_WARNING = 'run_status_warning'
+    RUN_STATUS_UNSCHEDULABLE = 'run_status_unschedulable'
+    RUN_STATUS_UPSTREAM_FAILED = 'run_status_upstream_failed'
+    RUN_STATUS_RETRYING = 'run_status_retrying'
+    RUN_STATUS_UNKNOWN = 'run_status_unknown'
+    RUN_STATUS_DONE = 'run_status_done'
+    RUN_APPROVED_ACTOR = 'run_approved_actor'
+    RUN_INVALIDATED_ACTOR = 'run_invalidated_actor'
+    RUN_NEW_ARTIFACTS = 'run_new_artifacts'
+    CONNECTION_GIT_COMMIT = 'connection_git_commit'
+    CONNECTION_DATASET_VERSION = 'connection_dataset_version'
+    CONNECTION_REGISTRY_IMAGE = 'connection_registry_image'
+    ALERT_INFO = 'alert_info'
+    ALERT_WARNING = 'alert_warning'
+    ALERT_CRITICAL = 'alert_critical'
+    MODEL_VERSION_NEW_METRIC = 'model_version_new_metric'
+    PROJECT_CUSTOM_EVENT = 'project_custom_event'
+    ORG_CUSTOM_EVENT = 'org_custom_event'
 
-        return self.to_dict() != other.to_dict()

polyaxon_sdk/models/v1_event_model.py

@@ -1,200 +1,75 @@
-#!/usr/bin/python
-#
-# Copyright 2018-2022 Polyaxon, Inc.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#      http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
 # coding: utf-8
 
 """
     Polyaxon SDKs and REST API specification.
 
-    Polyaxon SDKs and REST API specification.  # noqa: E501
+         # noqa: E501
 
-    The version of the OpenAPI document: 1.18.0
+    The version of the OpenAPI document: 2.1.6
     Contact: contact@polyaxon.com
-    Generated by: https://openapi-generator.tech
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
 """
 
 
-try:
-    from inspect import getfullargspec
-except ImportError:
-    from inspect import getargspec as getfullargspec
+from __future__ import annotations
+from inspect import getfullargspec
 import pprint
 import re  # noqa: F401
-import six
-
-from polyaxon_sdk.configuration import Configuration
+import json
 
 
-class V1EventModel(object):
-    """NOTE: This class is auto generated by OpenAPI Generator.
-    Ref: https://openapi-generator.tech
-
-    Do not edit the class manually.
-    """
+from typing import Any, Dict, Optional
+from pydantic import BaseModel, StrictStr
 
+class V1EventModel(BaseModel):
     """
-    Attributes:
-      openapi_types (dict): The key is attribute name
-                            and the value is attribute type.
-      attribute_map (dict): The key is attribute name
-                            and the value is json key in definition.
+    V1EventModel
     """
-    openapi_types = {
-        'framework': 'str',
-        'path': 'str',
-        'spec': 'object'
-    }
-
-    attribute_map = {
-        'framework': 'framework',
-        'path': 'path',
-        'spec': 'spec'
-    }
-
-    def __init__(self, framework=None, path=None, spec=None, local_vars_configuration=None):  # noqa: E501
-        """V1EventModel - a model defined in OpenAPI"""  # noqa: E501
-        if local_vars_configuration is None:
-            local_vars_configuration = Configuration.get_default_copy()
-        self.local_vars_configuration = local_vars_configuration
-
-        self._framework = None
-        self._path = None
-        self._spec = None
-        self.discriminator = None
-
-        if framework is not None:
-            self.framework = framework
-        if path is not None:
-            self.path = path
-        if spec is not None:
-            self.spec = spec
-
-    @property
-    def framework(self):
-        """Gets the framework of this V1EventModel.  # noqa: E501
-
-
-        :return: The framework of this V1EventModel.  # noqa: E501
-        :rtype: str
-        """
-        return self._framework
-
-    @framework.setter
-    def framework(self, framework):
-        """Sets the framework of this V1EventModel.
-
-
-        :param framework: The framework of this V1EventModel.  # noqa: E501
-        :type framework: str
-        """
-
-        self._framework = framework
-
-    @property
-    def path(self):
-        """Gets the path of this V1EventModel.  # noqa: E501
-
-
-        :return: The path of this V1EventModel.  # noqa: E501
-        :rtype: str
-        """
-        return self._path
-
-    @path.setter
-    def path(self, path):
-        """Sets the path of this V1EventModel.
-
-
-        :param path: The path of this V1EventModel.  # noqa: E501
-        :type path: str
-        """
-
-        self._path = path
-
-    @property
-    def spec(self):
-        """Gets the spec of this V1EventModel.  # noqa: E501
-
-
-        :return: The spec of this V1EventModel.  # noqa: E501
-        :rtype: object
-        """
-        return self._spec
-
-    @spec.setter
-    def spec(self, spec):
-        """Sets the spec of this V1EventModel.
-
-
-        :param spec: The spec of this V1EventModel.  # noqa: E501
-        :type spec: object
-        """
-
-        self._spec = spec
-
-    def to_dict(self, serialize=False):
-        """Returns the model properties as a dict"""
-        result = {}
-
-        def convert(x):
-            if hasattr(x, "to_dict"):
-                args = getfullargspec(x.to_dict).args
-                if len(args) == 1:
-                    return x.to_dict()
-                else:
-                    return x.to_dict(serialize)
-            else:
-                return x
-
-        for attr, _ in six.iteritems(self.openapi_types):
-            value = getattr(self, attr)
-            attr = self.attribute_map.get(attr, attr) if serialize else attr
-            if isinstance(value, list):
-                result[attr] = list(map(
-                    lambda x: convert(x),
-                    value
-                ))
-            elif isinstance(value, dict):
-                result[attr] = dict(map(
-                    lambda item: (item[0], convert(item[1])),
-                    value.items()
-                ))
-            else:
-                result[attr] = convert(value)
-
-        return result
-
-    def to_str(self):
-        """Returns the string representation of the model"""
-        return pprint.pformat(self.to_dict())
-
-    def __repr__(self):
-        """For `print` and `pprint`"""
-        return self.to_str()
-
-    def __eq__(self, other):
-        """Returns true if both objects are equal"""
-        if not isinstance(other, V1EventModel):
-            return False
-
-        return self.to_dict() == other.to_dict()
-
-    def __ne__(self, other):
-        """Returns true if both objects are not equal"""
-        if not isinstance(other, V1EventModel):
-            return True
+    framework: Optional[StrictStr] = None
+    path: Optional[StrictStr] = None
+    spec: Optional[Dict[str, Any]] = None
+    __properties = ["framework", "path", "spec"]
+
+    class Config:
+        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) -> V1EventModel:
+        """Create an instance of V1EventModel 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) -> V1EventModel:
+        """Create an instance of V1EventModel from a dict"""
+        if obj is None:
+            return None
+
+        if type(obj) is not dict:
+            return V1EventModel.parse_obj(obj)
+
+        _obj = V1EventModel.parse_obj({
+            "framework": obj.get("framework"),
+            "path": obj.get("path"),
+            "spec": obj.get("spec")
+        })
+        return _obj
 
-        return self.to_dict() != other.to_dict()

polyaxon_sdk/models/v1_event_span.py

@@ -0,0 +1,110 @@
+# coding: utf-8
+
+"""
+    Polyaxon SDKs and REST API specification.
+
+         # noqa: E501
+
+    The version of the OpenAPI document: 2.1.6
+    Contact: contact@polyaxon.com
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""
+
+
+from __future__ import annotations
+from inspect import getfullargspec
+import pprint
+import re  # noqa: F401
+import json
+
+from datetime import datetime
+from typing import Any, Dict, List, Optional
+from pydantic import BaseModel, StrictStr, conlist
+from polyaxon_sdk.models.v1_event_span_kind import V1EventSpanKind
+from polyaxon_sdk.models.v1_status_condition import V1StatusCondition
+from polyaxon_sdk.models.v1_statuses import V1Statuses
+
+class V1EventSpan(BaseModel):
+    """
+    V1EventSpan
+    """
+    uuid: Optional[StrictStr] = None
+    name: Optional[StrictStr] = None
+    tags: Optional[conlist(StrictStr)] = None
+    started_at: Optional[datetime] = None
+    finished_at: Optional[datetime] = None
+    status: Optional[V1Statuses] = None
+    status_conditions: Optional[conlist(V1StatusCondition)] = None
+    metadata: Optional[Dict[str, Any]] = None
+    inputs: Optional[Dict[str, Any]] = None
+    outputs: Optional[Dict[str, Any]] = None
+    children: Optional[conlist(V1EventSpan)] = None
+    kind: Optional[V1EventSpanKind] = None
+    __properties = ["uuid", "name", "tags", "started_at", "finished_at", "status", "status_conditions", "metadata", "inputs", "outputs", "children", "kind"]
+
+    class Config:
+        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) -> V1EventSpan:
+        """Create an instance of V1EventSpan 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 pydantic by calling `to_dict()` of each item in status_conditions (list)
+        _items = []
+        if self.status_conditions:
+            for _item in self.status_conditions:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict['status_conditions'] = _items
+        # override the default output from pydantic by calling `to_dict()` of each item in children (list)
+        _items = []
+        if self.children:
+            for _item in self.children:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict['children'] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> V1EventSpan:
+        """Create an instance of V1EventSpan from a dict"""
+        if obj is None:
+            return None
+
+        if type(obj) is not dict:
+            return V1EventSpan.parse_obj(obj)
+
+        _obj = V1EventSpan.parse_obj({
+            "uuid": obj.get("uuid"),
+            "name": obj.get("name"),
+            "tags": obj.get("tags"),
+            "started_at": obj.get("started_at"),
+            "finished_at": obj.get("finished_at"),
+            "status": obj.get("status"),
+            "status_conditions": [V1StatusCondition.from_dict(_item) for _item in obj.get("status_conditions")] if obj.get("status_conditions") is not None else None,
+            "metadata": obj.get("metadata"),
+            "inputs": obj.get("inputs"),
+            "outputs": obj.get("outputs"),
+            "children": [V1EventSpan.from_dict(_item) for _item in obj.get("children")] if obj.get("children") is not None else None,
+            "kind": obj.get("kind")
+        })
+        return _obj
+

polyaxon_sdk/models/v1_event_span_kind.py

@@ -0,0 +1,39 @@
+# coding: utf-8
+
+"""
+    Polyaxon SDKs and REST API specification.
+
+         # noqa: E501
+
+    The version of the OpenAPI document: 2.1.6
+    Contact: contact@polyaxon.com
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""
+
+
+from inspect import getfullargspec
+import pprint
+import re  # noqa: F401
+from aenum import Enum, no_arg
+
+
+
+
+
+class V1EventSpanKind(str, Enum):
+    """
+    V1EventSpanKind
+    """
+
+    """
+    allowed enum values
+    """
+    LLM = 'llm'
+    CHAIN = 'chain'
+    AGENT = 'agent'
+    TOOL = 'tool'
+    EMBEDDING = 'embedding'
+    RETRIEVER = 'retriever'
+