asteroid-odyssey 1.0.3__py3-none-any.whl → 1.2.0__py3-none-any.whl

asteroid_odyssey/{openapi_client → agents_v1_gen}/api/api_api.py

@@ -16,11 +16,11 @@ from pydantic import validate_call, Field, StrictFloat, StrictStr, StrictInt
 from typing import Any, Dict, List, Optional, Tuple, Union
 from typing_extensions import Annotated
 
-from asteroid_odyssey.openapi_client.models.health_check200_response import HealthCheck200Response
+from asteroid_odyssey.agents_v1_gen.models.health_check200_response import HealthCheck200Response
 
-from asteroid_odyssey.openapi_client.api_client import ApiClient, RequestSerialized
-from asteroid_odyssey.openapi_client.api_response import ApiResponse
-from asteroid_odyssey.openapi_client.rest import RESTResponseType
+from asteroid_odyssey.agents_v1_gen.api_client import ApiClient, RequestSerialized
+from asteroid_odyssey.agents_v1_gen.api_response import ApiResponse
+from asteroid_odyssey.agents_v1_gen.rest import RESTResponseType
 
 
 class APIApi:

asteroid_odyssey/{openapi_client/api/sdk_api.py → agents_v1_gen/api/execution_api.py}

@@ -16,21 +16,22 @@ from pydantic import validate_call, Field, StrictFloat, StrictStr, StrictInt
 from typing import Any, Dict, List, Optional, Tuple, Union
 from typing_extensions import Annotated
 
-from pydantic import Field, StrictStr
-from typing import Any, Dict
+from pydantic import Field, StrictBytes, StrictStr
+from typing import Any, Dict, List, Optional, Tuple, Union
 from typing_extensions import Annotated
-from asteroid_odyssey.openapi_client.models.browser_session_recording_response import BrowserSessionRecordingResponse
-from asteroid_odyssey.openapi_client.models.execution_response import ExecutionResponse
-from asteroid_odyssey.openapi_client.models.execution_result_response import ExecutionResultResponse
-from asteroid_odyssey.openapi_client.models.execution_status_response import ExecutionStatusResponse
-from asteroid_odyssey.openapi_client.models.structured_agent_execution_request import StructuredAgentExecutionRequest
+from asteroid_odyssey.agents_v1_gen.models.browser_session_recording_response import BrowserSessionRecordingResponse
+from asteroid_odyssey.agents_v1_gen.models.execution_response import ExecutionResponse
+from asteroid_odyssey.agents_v1_gen.models.execution_result_response import ExecutionResultResponse
+from asteroid_odyssey.agents_v1_gen.models.execution_status_response import ExecutionStatusResponse
+from asteroid_odyssey.agents_v1_gen.models.structured_agent_execution_request import StructuredAgentExecutionRequest
+from asteroid_odyssey.agents_v1_gen.models.upload_execution_files200_response import UploadExecutionFiles200Response
 
-from asteroid_odyssey.openapi_client.api_client import ApiClient, RequestSerialized
-from asteroid_odyssey.openapi_client.api_response import ApiResponse
-from asteroid_odyssey.openapi_client.rest import RESTResponseType
+from asteroid_odyssey.agents_v1_gen.api_client import ApiClient, RequestSerialized
+from asteroid_odyssey.agents_v1_gen.api_response import ApiResponse
+from asteroid_odyssey.agents_v1_gen.rest import RESTResponseType
 
 
-class SDKApi:
+class ExecutionApi:
     """NOTE: This class is auto generated by OpenAPI Generator
     Ref: https://openapi-generator.tech
 
@@ -61,7 +62,7 @@ class SDKApi:
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
     ) -> ExecutionResponse:
-        """Execute an agent
+        """(Deprecated) Execute an agent
 
         Executes an agent with the provided parameters
 
@@ -90,6 +91,7 @@ class SDKApi:
         :type _host_index: int, optional
         :return: Returns the result object.
         """ # noqa: E501
+        warnings.warn("POST /agent/{id} is deprecated.", DeprecationWarning)
 
         _param = self._execute_agent_serialize(
             id=id,
@@ -135,7 +137,7 @@ class SDKApi:
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
     ) -> ApiResponse[ExecutionResponse]:
-        """Execute an agent
+        """(Deprecated) Execute an agent
 
         Executes an agent with the provided parameters
 
@@ -164,6 +166,7 @@ class SDKApi:
         :type _host_index: int, optional
         :return: Returns the result object.
         """ # noqa: E501
+        warnings.warn("POST /agent/{id} is deprecated.", DeprecationWarning)
 
         _param = self._execute_agent_serialize(
             id=id,
@@ -209,7 +212,7 @@ class SDKApi:
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
     ) -> RESTResponseType:
-        """Execute an agent
+        """(Deprecated) Execute an agent
 
         Executes an agent with the provided parameters
 
@@ -238,6 +241,7 @@ class SDKApi:
         :type _host_index: int, optional
         :return: Returns the result object.
         """ # noqa: E501
+        warnings.warn("POST /agent/{id} is deprecated.", DeprecationWarning)
 
         _param = self._execute_agent_serialize(
             id=id,
@@ -1432,3 +1436,299 @@ class SDKApi:
         )
 
 
+
+
+    @validate_call
+    def upload_execution_files(
+        self,
+        id: Annotated[StrictStr, Field(description="The ID of the execution")],
+        files: Annotated[Optional[List[Union[StrictBytes, StrictStr, Tuple[StrictStr, StrictBytes]]]], Field(description="Files to upload to the execution")] = None,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> UploadExecutionFiles200Response:
+        """Upload files to an execution
+
+
+        :param id: The ID of the execution (required)
+        :type id: str
+        :param files: Files to upload to the execution
+        :type files: List[bytearray]
+        :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.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._upload_execution_files_serialize(
+            id=id,
+            files=files,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "UploadExecutionFiles200Response",
+            '400': "ErrorResponse",
+            '401': "ErrorResponse",
+            '404': "ErrorResponse",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        ).data
+
+
+    @validate_call
+    def upload_execution_files_with_http_info(
+        self,
+        id: Annotated[StrictStr, Field(description="The ID of the execution")],
+        files: Annotated[Optional[List[Union[StrictBytes, StrictStr, Tuple[StrictStr, StrictBytes]]]], Field(description="Files to upload to the execution")] = None,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> ApiResponse[UploadExecutionFiles200Response]:
+        """Upload files to an execution
+
+
+        :param id: The ID of the execution (required)
+        :type id: str
+        :param files: Files to upload to the execution
+        :type files: List[bytearray]
+        :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.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._upload_execution_files_serialize(
+            id=id,
+            files=files,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "UploadExecutionFiles200Response",
+            '400': "ErrorResponse",
+            '401': "ErrorResponse",
+            '404': "ErrorResponse",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        )
+
+
+    @validate_call
+    def upload_execution_files_without_preload_content(
+        self,
+        id: Annotated[StrictStr, Field(description="The ID of the execution")],
+        files: Annotated[Optional[List[Union[StrictBytes, StrictStr, Tuple[StrictStr, StrictBytes]]]], Field(description="Files to upload to the execution")] = None,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> RESTResponseType:
+        """Upload files to an execution
+
+
+        :param id: The ID of the execution (required)
+        :type id: str
+        :param files: Files to upload to the execution
+        :type files: List[bytearray]
+        :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.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._upload_execution_files_serialize(
+            id=id,
+            files=files,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "UploadExecutionFiles200Response",
+            '400': "ErrorResponse",
+            '401': "ErrorResponse",
+            '404': "ErrorResponse",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        return response_data.response
+
+
+    def _upload_execution_files_serialize(
+        self,
+        id,
+        files,
+        _request_auth,
+        _content_type,
+        _headers,
+        _host_index,
+    ) -> RequestSerialized:
+
+        _host = None
+
+        _collection_formats: Dict[str, str] = {
+            'files': 'csv',
+        }
+
+        _path_params: Dict[str, str] = {}
+        _query_params: List[Tuple[str, str]] = []
+        _header_params: Dict[str, Optional[str]] = _headers or {}
+        _form_params: List[Tuple[str, str]] = []
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
+        _body_params: Optional[bytes] = None
+
+        # process the path parameters
+        if id is not None:
+            _path_params['id'] = id
+        # process the query parameters
+        # process the header parameters
+        # process the form parameters
+        if files is not None:
+            _files['files'] = files
+        # process the body parameter
+
+
+        # set the HTTP header `Accept`
+        if 'Accept' not in _header_params:
+            _header_params['Accept'] = self.api_client.select_header_accept(
+                [
+                    'application/json'
+                ]
+            )
+
+        # set the HTTP header `Content-Type`
+        if _content_type:
+            _header_params['Content-Type'] = _content_type
+        else:
+            _default_content_type = (
+                self.api_client.select_header_content_type(
+                    [
+                        'multipart/form-data'
+                    ]
+                )
+            )
+            if _default_content_type is not None:
+                _header_params['Content-Type'] = _default_content_type
+
+        # authentication setting
+        _auth_settings: List[str] = [
+            'ApiKeyAuth'
+        ]
+
+        return self.api_client.param_serialize(
+            method='POST',
+            resource_path='/execution/{id}/files',
+            path_params=_path_params,
+            query_params=_query_params,
+            header_params=_header_params,
+            body=_body_params,
+            post_params=_form_params,
+            files=_files,
+            auth_settings=_auth_settings,
+            collection_formats=_collection_formats,
+            _host=_host,
+            _request_auth=_request_auth
+        )
+
+

asteroid_odyssey/{openapi_client → agents_v1_gen}/api_client.py

@@ -26,11 +26,11 @@ from urllib.parse import quote
 from typing import Tuple, Optional, List, Dict, Union
 from pydantic import SecretStr
 
-from asteroid_odyssey.openapi_client.configuration import Configuration
-from asteroid_odyssey.openapi_client.api_response import ApiResponse, T as ApiResponseT
-import asteroid_odyssey.openapi_client.models
-from asteroid_odyssey.openapi_client import rest
-from asteroid_odyssey.openapi_client.exceptions import (
+from asteroid_odyssey.agents_v1_gen.configuration import Configuration
+from asteroid_odyssey.agents_v1_gen.api_response import ApiResponse, T as ApiResponseT
+import asteroid_odyssey.agents_v1_gen.models
+from asteroid_odyssey.agents_v1_gen import rest
+from asteroid_odyssey.agents_v1_gen.exceptions import (
     ApiValueError,
     ApiException,
     BadRequestException,
@@ -453,7 +453,7 @@ class ApiClient:
             if klass in self.NATIVE_TYPES_MAPPING:
                 klass = self.NATIVE_TYPES_MAPPING[klass]
             else:
-                klass = getattr(asteroid_odyssey.openapi_client.models, klass)
+                klass = getattr(asteroid_odyssey.agents_v1_gen.models, klass)
 
         if klass in self.PRIMITIVE_TYPES:
             return self.__deserialize_primitive(data, klass)

asteroid_odyssey/{openapi_client → agents_v1_gen}/configuration.py

@@ -177,7 +177,7 @@ class Configuration:
 
     You can programmatically set the cookie:
 
-conf = asteroid_odyssey.openapi_client.Configuration(
+conf = asteroid_odyssey.agents_v1_gen.Configuration(
     api_key={'cookieAuth': 'abc123'}
     api_key_prefix={'cookieAuth': 'JSESSIONID'}
 )
@@ -252,7 +252,7 @@ conf = asteroid_odyssey.openapi_client.Configuration(
         self.logger = {}
         """Logging Settings
         """
-        self.logger["package_logger"] = logging.getLogger("asteroid_odyssey.openapi_client")
+        self.logger["package_logger"] = logging.getLogger("asteroid_odyssey.agents_v1_gen")
         self.logger["urllib3_logger"] = logging.getLogger("urllib3")
         self.logger_format = '%(asctime)s %(levelname)s %(message)s'
         """Log format

asteroid_odyssey/agents_v1_gen/models/__init__.py

@@ -0,0 +1,34 @@
+# coding: utf-8
+
+# flake8: noqa
+"""
+    Asteroid Agents API
+
+    Version 1 of the Asteroid Agents API
+
+    The version of the OpenAPI document: v1.0.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+# import models into model package
+from asteroid_odyssey.agents_v1_gen.models.agent_profile import AgentProfile
+from asteroid_odyssey.agents_v1_gen.models.browser_session_recording_response import BrowserSessionRecordingResponse
+from asteroid_odyssey.agents_v1_gen.models.country_code import CountryCode
+from asteroid_odyssey.agents_v1_gen.models.create_agent_profile_request import CreateAgentProfileRequest
+from asteroid_odyssey.agents_v1_gen.models.credential import Credential
+from asteroid_odyssey.agents_v1_gen.models.delete_agent_profile200_response import DeleteAgentProfile200Response
+from asteroid_odyssey.agents_v1_gen.models.error_response import ErrorResponse
+from asteroid_odyssey.agents_v1_gen.models.execution_response import ExecutionResponse
+from asteroid_odyssey.agents_v1_gen.models.execution_result import ExecutionResult
+from asteroid_odyssey.agents_v1_gen.models.execution_result_response import ExecutionResultResponse
+from asteroid_odyssey.agents_v1_gen.models.execution_status_response import ExecutionStatusResponse
+from asteroid_odyssey.agents_v1_gen.models.health_check200_response import HealthCheck200Response
+from asteroid_odyssey.agents_v1_gen.models.health_check500_response import HealthCheck500Response
+from asteroid_odyssey.agents_v1_gen.models.proxy_type import ProxyType
+from asteroid_odyssey.agents_v1_gen.models.status import Status
+from asteroid_odyssey.agents_v1_gen.models.structured_agent_execution_request import StructuredAgentExecutionRequest
+from asteroid_odyssey.agents_v1_gen.models.update_agent_profile_request import UpdateAgentProfileRequest
+from asteroid_odyssey.agents_v1_gen.models.upload_execution_files200_response import UploadExecutionFiles200Response

asteroid_odyssey/agents_v1_gen/models/agent_profile.py

@@ -0,0 +1,118 @@
+# coding: utf-8
+
+"""
+    Asteroid Agents API
+
+    Version 1 of the Asteroid Agents API
+
+    The version of the OpenAPI document: v1.0.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from datetime import datetime
+from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr
+from typing import Any, ClassVar, Dict, List
+from asteroid_odyssey.agents_v1_gen.models.country_code import CountryCode
+from asteroid_odyssey.agents_v1_gen.models.credential import Credential
+from asteroid_odyssey.agents_v1_gen.models.proxy_type import ProxyType
+from typing import Optional, Set
+from typing_extensions import Self
+
+class AgentProfile(BaseModel):
+    """
+    AgentProfile
+    """ # noqa: E501
+    id: StrictStr = Field(description="Unique identifier for the agent profile")
+    name: StrictStr = Field(description="Name of the agent profile (unique within organization)")
+    description: StrictStr = Field(description="Description of the agent profile")
+    organization_id: StrictStr = Field(description="The ID of the organization that the agent profile belongs to")
+    proxy_cc: CountryCode
+    proxy_type: ProxyType
+    captcha_solver_active: StrictBool = Field(description="Whether the captcha solver is active for this profile")
+    sticky_ip: StrictBool = Field(description="Whether the same IP address should be used for all executions of this profile")
+    credentials: List[Credential] = Field(description="List of credentials associated with this agent profile")
+    created_at: datetime = Field(description="The date and time the agent profile was created")
+    updated_at: datetime = Field(description="The last update time of the agent profile")
+    __properties: ClassVar[List[str]] = ["id", "name", "description", "organization_id", "proxy_cc", "proxy_type", "captcha_solver_active", "sticky_ip", "credentials", "created_at", "updated_at"]
+
+    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 AgentProfile 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 each item in credentials (list)
+        _items = []
+        if self.credentials:
+            for _item_credentials in self.credentials:
+                if _item_credentials:
+                    _items.append(_item_credentials.to_dict())
+            _dict['credentials'] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of AgentProfile from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "id": obj.get("id"),
+            "name": obj.get("name"),
+            "description": obj.get("description"),
+            "organization_id": obj.get("organization_id"),
+            "proxy_cc": obj.get("proxy_cc"),
+            "proxy_type": obj.get("proxy_type"),
+            "captcha_solver_active": obj.get("captcha_solver_active"),
+            "sticky_ip": obj.get("sticky_ip"),
+            "credentials": [Credential.from_dict(_item) for _item in obj["credentials"]] if obj.get("credentials") is not None else None,
+            "created_at": obj.get("created_at"),
+            "updated_at": obj.get("updated_at")
+        })
+        return _obj
+
+

asteroid_odyssey/agents_v1_gen/models/country_code.py

@@ -0,0 +1,44 @@
+# coding: utf-8
+
+"""
+    Asteroid Agents API
+
+    Version 1 of the Asteroid Agents API
+
+    The version of the OpenAPI document: v1.0.0
+    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 CountryCode(str, Enum):
+    """
+    Two-letter country code for proxy location
+    """
+
+    """
+    allowed enum values
+    """
+    US = 'us'
+    UK = 'uk'
+    FR = 'fr'
+    IT = 'it'
+    JP = 'jp'
+    AU = 'au'
+    DE = 'de'
+    FI = 'fi'
+    CA = 'ca'
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of CountryCode from a JSON string"""
+        return cls(json.loads(json_str))
+
+