scorable 1.6.4__py3-none-any.whl

root/generated/openapi_aclient/models/skill_test_input_request.py

@@ -0,0 +1,171 @@
+# coding: utf-8
+
+"""
+Root Signals API
+
+Root Signals JSON API provides a way to access Root Signals using provisioned API token
+
+The version of the OpenAPI document: 1.0.0 (latest)
+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 Any, ClassVar, Dict, List, Optional, Set
+
+from pydantic import BaseModel, ConfigDict, Field, StrictBool
+from typing_extensions import Annotated, Self
+
+from root.generated.openapi_aclient.models.input_variable_request import InputVariableRequest
+from root.generated.openapi_aclient.models.objective_request import ObjectiveRequest
+from root.generated.openapi_aclient.models.reference_variable_request import ReferenceVariableRequest
+from root.generated.openapi_aclient.models.skill_test_data_request_dataset_range import SkillTestDataRequestDatasetRange
+
+
+class SkillTestInputRequest(BaseModel):
+    """
+    SkillTestInputRequest
+    """  # noqa: E501
+
+    test_data: Optional[List[List[Annotated[str, Field(min_length=1, strict=True)]]]] = None
+    test_dataset_id: Optional[Annotated[str, Field(min_length=1, strict=True)]] = None
+    dataset_range: Optional[SkillTestDataRequestDatasetRange] = None
+    prompt: Annotated[str, Field(min_length=1, strict=True)]
+    reference_variables: Optional[List[ReferenceVariableRequest]] = None
+    input_variables: Optional[List[InputVariableRequest]] = None
+    models: Optional[List[Annotated[str, Field(min_length=1, strict=True)]]] = None
+    name: Optional[Annotated[str, Field(min_length=1, strict=True)]] = None
+    pii_filter: Optional[StrictBool] = False
+    objective: Optional[ObjectiveRequest] = None
+    is_evaluator: Optional[StrictBool] = False
+    __properties: ClassVar[List[str]] = [
+        "test_data",
+        "test_dataset_id",
+        "dataset_range",
+        "prompt",
+        "reference_variables",
+        "input_variables",
+        "models",
+        "name",
+        "pii_filter",
+        "objective",
+        "is_evaluator",
+    ]
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of SkillTestInputRequest from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of dataset_range
+        if self.dataset_range:
+            _dict["dataset_range"] = self.dataset_range.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of each item in reference_variables (list)
+        _items = []
+        if self.reference_variables:
+            for _item in self.reference_variables:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict["reference_variables"] = _items
+        # override the default output from pydantic by calling `to_dict()` of each item in input_variables (list)
+        _items = []
+        if self.input_variables:
+            for _item in self.input_variables:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict["input_variables"] = _items
+        # override the default output from pydantic by calling `to_dict()` of objective
+        if self.objective:
+            _dict["objective"] = self.objective.to_dict()
+        # set to None if test_data (nullable) is None
+        # and model_fields_set contains the field
+        if self.test_data is None and "test_data" in self.model_fields_set:
+            _dict["test_data"] = None
+
+        # set to None if dataset_range (nullable) is None
+        # and model_fields_set contains the field
+        if self.dataset_range is None and "dataset_range" in self.model_fields_set:
+            _dict["dataset_range"] = None
+
+        # set to None if name (nullable) is None
+        # and model_fields_set contains the field
+        if self.name is None and "name" in self.model_fields_set:
+            _dict["name"] = None
+
+        # set to None if objective (nullable) is None
+        # and model_fields_set contains the field
+        if self.objective is None and "objective" in self.model_fields_set:
+            _dict["objective"] = None
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of SkillTestInputRequest from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate(
+            {
+                "test_data": obj.get("test_data"),
+                "test_dataset_id": obj.get("test_dataset_id"),
+                "dataset_range": SkillTestDataRequestDatasetRange.from_dict(obj["dataset_range"])
+                if obj.get("dataset_range") is not None
+                else None,
+                "prompt": obj.get("prompt"),
+                "reference_variables": [
+                    ReferenceVariableRequest.from_dict(_item) for _item in obj["reference_variables"]
+                ]
+                if obj.get("reference_variables") is not None
+                else None,
+                "input_variables": [InputVariableRequest.from_dict(_item) for _item in obj["input_variables"]]
+                if obj.get("input_variables") is not None
+                else None,
+                "models": obj.get("models"),
+                "name": obj.get("name"),
+                "pii_filter": obj.get("pii_filter") if obj.get("pii_filter") is not None else False,
+                "objective": ObjectiveRequest.from_dict(obj["objective"]) if obj.get("objective") is not None else None,
+                "is_evaluator": obj.get("is_evaluator") if obj.get("is_evaluator") is not None else False,
+            }
+        )
+        return _obj

root/generated/openapi_aclient/models/skill_type_enum.py

@@ -0,0 +1,36 @@
+# coding: utf-8
+
+"""
+Root Signals API
+
+Root Signals JSON API provides a way to access Root Signals using provisioned API token
+
+The version of the OpenAPI document: 1.0.0 (latest)
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+from __future__ import annotations
+
+import json
+from enum import Enum
+
+from typing_extensions import Self
+
+
+class SkillTypeEnum(str, Enum):
+    """
+    * `managed` - managed * `native` - native
+    """
+
+    """
+    allowed enum values
+    """
+    MANAGED = "managed"
+    NATIVE = "native"
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of SkillTypeEnum from a JSON string"""
+        return cls(json.loads(json_str))

root/generated/openapi_aclient/models/status_change.py

@@ -0,0 +1,84 @@
+# coding: utf-8
+
+"""
+Root Signals API
+
+Root Signals JSON API provides a way to access Root Signals using provisioned API token
+
+The version of the OpenAPI document: 1.0.0 (latest)
+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 Any, ClassVar, Dict, List, Optional, Set
+
+from pydantic import BaseModel, ConfigDict
+from typing_extensions import Self
+
+from root.generated.openapi_aclient.models.status_change_status_enum import StatusChangeStatusEnum
+
+
+class StatusChange(BaseModel):
+    """
+    StatusChange
+    """  # noqa: E501
+
+    status: StatusChangeStatusEnum
+    __properties: ClassVar[List[str]] = ["status"]
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of StatusChange from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of StatusChange from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({"status": obj.get("status")})
+        return _obj

root/generated/openapi_aclient/models/status_change_request.py

@@ -0,0 +1,84 @@
+# coding: utf-8
+
+"""
+Root Signals API
+
+Root Signals JSON API provides a way to access Root Signals using provisioned API token
+
+The version of the OpenAPI document: 1.0.0 (latest)
+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 Any, ClassVar, Dict, List, Optional, Set
+
+from pydantic import BaseModel, ConfigDict
+from typing_extensions import Self
+
+from root.generated.openapi_aclient.models.status_change_status_enum import StatusChangeStatusEnum
+
+
+class StatusChangeRequest(BaseModel):
+    """
+    StatusChangeRequest
+    """  # noqa: E501
+
+    status: StatusChangeStatusEnum
+    __properties: ClassVar[List[str]] = ["status"]
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of StatusChangeRequest from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of StatusChangeRequest from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({"status": obj.get("status")})
+        return _obj

root/generated/openapi_aclient/models/status_change_status_enum.py

@@ -0,0 +1,36 @@
+# coding: utf-8
+
+"""
+Root Signals API
+
+Root Signals JSON API provides a way to access Root Signals using provisioned API token
+
+The version of the OpenAPI document: 1.0.0 (latest)
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+from __future__ import annotations
+
+import json
+from enum import Enum
+
+from typing_extensions import Self
+
+
+class StatusChangeStatusEnum(str, Enum):
+    """
+    * `unlisted` - unlisted * `listed` - listed
+    """
+
+    """
+    allowed enum values
+    """
+    UNLISTED = "unlisted"
+    LISTED = "listed"
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of StatusChangeStatusEnum from a JSON string"""
+        return cls(json.loads(json_str))

root/generated/openapi_aclient/models/status_enum.py

@@ -0,0 +1,38 @@
+# coding: utf-8
+
+"""
+Root Signals API
+
+Root Signals JSON API provides a way to access Root Signals using provisioned API token
+
+The version of the OpenAPI document: 1.0.0 (latest)
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+from __future__ import annotations
+
+import json
+from enum import Enum
+
+from typing_extensions import Self
+
+
+class StatusEnum(str, Enum):
+    """
+    * `unlisted` - unlisted * `listed` - listed * `public` - public * `public_unlisted` - public_unlisted
+    """
+
+    """
+    allowed enum values
+    """
+    UNLISTED = "unlisted"
+    LISTED = "listed"
+    PUBLIC = "public"
+    PUBLIC_UNLISTED = "public_unlisted"
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of StatusEnum from a JSON string"""
+        return cls(json.loads(json_str))

root/generated/openapi_aclient/models/validation_result_status.py

@@ -0,0 +1,36 @@
+# coding: utf-8
+
+"""
+Root Signals API
+
+Root Signals JSON API provides a way to access Root Signals using provisioned API token
+
+The version of the OpenAPI document: 1.0.0 (latest)
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+from __future__ import annotations
+
+import json
+from enum import Enum
+
+from typing_extensions import Self
+
+
+class ValidationResultStatus(str, Enum):
+    """
+    * `pending` - Pending * `finished` - Finished
+    """
+
+    """
+    allowed enum values
+    """
+    PENDING = "pending"
+    FINISHED = "finished"
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of ValidationResultStatus from a JSON string"""
+        return cls(json.loads(json_str))

root/generated/openapi_aclient/models/visibility_enum.py

@@ -0,0 +1,38 @@
+# coding: utf-8
+
+"""
+Root Signals API
+
+Root Signals JSON API provides a way to access Root Signals using provisioned API token
+
+The version of the OpenAPI document: 1.0.0 (latest)
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+from __future__ import annotations
+
+import json
+from enum import Enum
+
+from typing_extensions import Self
+
+
+class VisibilityEnum(str, Enum):
+    """
+    * `global` - global * `public` - public * `listed` - listed * `unlisted` - unlisted
+    """
+
+    """
+    allowed enum values
+    """
+    GLOBAL = "global"
+    PUBLIC = "public"
+    LISTED = "listed"
+    UNLISTED = "unlisted"
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of VisibilityEnum from a JSON string"""
+        return cls(json.loads(json_str))

root/generated/openapi_aclient/rest.py

@@ -0,0 +1,166 @@
+# coding: utf-8
+
+"""
+Root Signals API
+
+Root Signals JSON API provides a way to access Root Signals using provisioned API token
+
+The version of the OpenAPI document: 1.0.0 (latest)
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+import io
+import json
+import re
+import ssl
+from typing import Optional, Union
+
+import aiohttp
+import aiohttp_retry
+
+from root.generated.openapi_aclient.exceptions import ApiException, ApiValueError
+
+RESTResponseType = aiohttp.ClientResponse
+
+ALLOW_RETRY_METHODS = frozenset({"DELETE", "GET", "HEAD", "OPTIONS", "PUT", "TRACE"})
+
+
+class RESTResponse(io.IOBase):
+    def __init__(self, resp) -> None:
+        self.response = resp
+        self.status = resp.status
+        self.reason = resp.reason
+        self.data = None
+
+    async def read(self):
+        if self.data is None:
+            self.data = await self.response.read()
+        return self.data
+
+    def getheaders(self):
+        """Returns a CIMultiDictProxy of the response headers."""
+        return self.response.headers
+
+    def getheader(self, name, default=None):
+        """Returns a given response header."""
+        return self.response.headers.get(name, default)
+
+
+class RESTClientObject:
+    def __init__(self, configuration) -> None:
+        # maxsize is number of requests to host that are allowed in parallel
+        maxsize = configuration.connection_pool_maxsize
+
+        ssl_context = ssl.create_default_context(cafile=configuration.ssl_ca_cert)
+        if configuration.cert_file:
+            ssl_context.load_cert_chain(configuration.cert_file, keyfile=configuration.key_file)
+
+        if not configuration.verify_ssl:
+            ssl_context.check_hostname = False
+            ssl_context.verify_mode = ssl.CERT_NONE
+
+        connector = aiohttp.TCPConnector(limit=maxsize, ssl=ssl_context)
+
+        self.proxy = configuration.proxy
+        self.proxy_headers = configuration.proxy_headers
+
+        # https pool manager
+        self.pool_manager = aiohttp.ClientSession(connector=connector, trust_env=True)
+
+        retries = configuration.retries
+        self.retry_client: Optional[aiohttp_retry.RetryClient]
+        if retries is not None:
+            self.retry_client = aiohttp_retry.RetryClient(
+                client_session=self.pool_manager,
+                retry_options=aiohttp_retry.ExponentialRetry(
+                    attempts=retries, factor=0.0, start_timeout=0.0, max_timeout=120.0
+                ),
+            )
+        else:
+            self.retry_client = None
+
+    async def close(self):
+        await self.pool_manager.close()
+        if self.retry_client is not None:
+            await self.retry_client.close()
+
+    async def request(self, method, url, headers=None, body=None, post_params=None, _request_timeout=None):
+        """Execute request
+
+        :param method: http request method
+        :param url: http request url
+        :param headers: http request headers
+        :param body: request json body, for `application/json`
+        :param post_params: request post parameters,
+                            `application/x-www-form-urlencoded`
+                            and `multipart/form-data`
+        :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.
+        """
+        method = method.upper()
+        assert method in ["GET", "HEAD", "DELETE", "POST", "PUT", "PATCH", "OPTIONS"]
+
+        if post_params and body:
+            raise ApiValueError("body parameter cannot be used with post_params parameter.")
+
+        post_params = post_params or {}
+        headers = headers or {}
+        # url already contains the URL query string
+        timeout = _request_timeout or 5 * 60
+
+        if "Content-Type" not in headers:
+            headers["Content-Type"] = "application/json"
+
+        args = {"method": method, "url": url, "timeout": timeout, "headers": headers}
+
+        if self.proxy:
+            args["proxy"] = self.proxy
+        if self.proxy_headers:
+            args["proxy_headers"] = self.proxy_headers
+
+        # For `POST`, `PUT`, `PATCH`, `OPTIONS`, `DELETE`
+        if method in ["POST", "PUT", "PATCH", "OPTIONS", "DELETE"]:
+            if re.search("json", headers["Content-Type"], re.IGNORECASE):
+                if body is not None:
+                    body = json.dumps(body)
+                args["data"] = body
+            elif headers["Content-Type"] == "application/x-www-form-urlencoded":
+                args["data"] = aiohttp.FormData(post_params)
+            elif headers["Content-Type"] == "multipart/form-data":
+                # must del headers['Content-Type'], or the correct
+                # Content-Type which generated by aiohttp
+                del headers["Content-Type"]
+                data = aiohttp.FormData()
+                for param in post_params:
+                    k, v = param
+                    if isinstance(v, tuple) and len(v) == 3:
+                        data.add_field(k, value=v[1], filename=v[0], content_type=v[2])
+                    else:
+                        data.add_field(k, v)
+                args["data"] = data
+
+            # Pass a `bytes` parameter directly in the body to support
+            # other content types than Json when `body` argument is provided
+            # in serialized form
+            elif isinstance(body, bytes):
+                args["data"] = body
+            else:
+                # Cannot generate the request from given parameters
+                msg = """Cannot prepare a request message for provided
+                         arguments. Please check that your arguments match
+                         declared content type."""
+                raise ApiException(status=0, reason=msg)
+
+        pool_manager: Union[aiohttp.ClientSession, aiohttp_retry.RetryClient]
+        if self.retry_client is not None and method in ALLOW_RETRY_METHODS:
+            pool_manager = self.retry_client
+        else:
+            pool_manager = self.pool_manager
+
+        r = await pool_manager.request(**args)
+
+        return RESTResponse(r)