orq-ai-sdk 4.2.0rc28__py3-none-any.whl → 4.3.0rc7__py3-none-any.whl

orq_ai_sdk/sdk.py

@@ -26,6 +26,7 @@ if TYPE_CHECKING:
     from orq_ai_sdk.evaluators import Evaluators
     from orq_ai_sdk.feedback import Feedback
     from orq_ai_sdk.files import Files
+    from orq_ai_sdk.identities import Identities
     from orq_ai_sdk.internal import Internal
     from orq_ai_sdk.knowledge import Knowledge
     from orq_ai_sdk.memorystores import MemoryStores
@@ -45,6 +46,7 @@ class Orq(BaseSDK):
     feedback: "Feedback"
     evals: "Evals"
     evaluators: "Evaluators"
+    identities: "Identities"
     deployments: "Deployments"
     agents: "Agents"
     conversations: "Conversations"
@@ -64,6 +66,7 @@ class Orq(BaseSDK):
         "feedback": ("orq_ai_sdk.feedback", "Feedback"),
         "evals": ("orq_ai_sdk.evals", "Evals"),
         "evaluators": ("orq_ai_sdk.evaluators", "Evaluators"),
+        "identities": ("orq_ai_sdk.identities", "Identities"),
         "deployments": ("orq_ai_sdk.deployments", "Deployments"),
         "agents": ("orq_ai_sdk.agents", "Agents"),
         "conversations": ("orq_ai_sdk.conversations", "Conversations"),

orq_ai_sdk/speech.py

@@ -0,0 +1,333 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from .basesdk import BaseSDK
+from orq_ai_sdk import models, utils
+from orq_ai_sdk._hooks import HookContext
+from orq_ai_sdk.models import createspeechop as models_createspeechop
+from orq_ai_sdk.types import OptionalNullable, UNSET
+from orq_ai_sdk.utils import get_security_from_env
+from typing import List, Mapping, Optional, Union
+
+
+class Speech(BaseSDK):
+    def create(
+        self,
+        *,
+        input_: str,
+        model: str,
+        voice: str,
+        response_format: Optional[
+            models_createspeechop.CreateSpeechResponseFormat
+        ] = "mp3",
+        speed: Optional[float] = 1,
+        name: Optional[str] = None,
+        fallbacks: Optional[
+            Union[
+                List[models_createspeechop.CreateSpeechFallbacks],
+                List[models_createspeechop.CreateSpeechFallbacksTypedDict],
+            ]
+        ] = None,
+        retry: Optional[
+            Union[
+                models_createspeechop.CreateSpeechRetry,
+                models_createspeechop.CreateSpeechRetryTypedDict,
+            ]
+        ] = None,
+        load_balancer: Optional[
+            Union[
+                models_createspeechop.CreateSpeechLoadBalancer,
+                models_createspeechop.CreateSpeechLoadBalancerTypedDict,
+            ]
+        ] = None,
+        timeout: Optional[
+            Union[
+                models_createspeechop.CreateSpeechTimeout,
+                models_createspeechop.CreateSpeechTimeoutTypedDict,
+            ]
+        ] = None,
+        orq: Optional[
+            Union[
+                models_createspeechop.CreateSpeechOrq,
+                models_createspeechop.CreateSpeechOrqTypedDict,
+            ]
+        ] = None,
+        retries: OptionalNullable[utils.RetryConfig] = UNSET,
+        server_url: Optional[str] = None,
+        timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
+    ):
+        r"""Create speech
+
+        Generates audio from the input text.
+
+        :param input: The text to generate audio for. The maximum length is 4096 characters
+        :param model: ID of the model to use
+        :param voice: The voice to use.
+
+            Available voices for OpenAI
+
+            `alloy`, `echo`, `fable`, `onyx`, `nova`, and `shimmer`
+
+            Available voices for ElevenLabs
+
+            `aria`, `roger`, `sarah`, `laura`, `charlie`, `george`, `callum`, `river`, `liam`, `charlotte`, `alice`, `matilda`, `will`, `jessica`, `eric`, `chris`, `brian`, `daniel`, `lily`, `bill`
+        :param response_format: The format to audio in. Supported formats are `mp3`, `opus`, `aac`, `flac`, `wav`, and `pcm`. If a format is provided but not supported by the provider, the response will be in the default format. When the provided format is not supported by the provider, the response will be in the default format.
+        :param speed: The speed of the generated audio.
+        :param name: The name to display on the trace. If not specified, the default system name will be used.
+        :param fallbacks: Array of fallback models to use if primary model fails
+        :param retry: Retry configuration for the request
+        :param load_balancer: Load balancer configuration for the request.
+        :param timeout: Timeout configuration to apply to the request. If the request exceeds the timeout, it will be retried or fallback to the next model if configured.
+        :param orq:
+        :param retries: Override the default retry configuration for this method
+        :param server_url: Override the default server URL for this method
+        :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
+        :param http_headers: Additional headers to set or replace on requests.
+        """
+        base_url = None
+        url_variables = None
+        if timeout_ms is None:
+            timeout_ms = self.sdk_configuration.timeout_ms
+
+        if timeout_ms is None:
+            timeout_ms = 600000
+
+        if server_url is not None:
+            base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
+
+        request = models.CreateSpeechRequestBody(
+            input=input_,
+            model=model,
+            voice=voice,
+            response_format=response_format,
+            speed=speed,
+            name=name,
+            fallbacks=utils.get_pydantic_model(
+                fallbacks, Optional[List[models.CreateSpeechFallbacks]]
+            ),
+            retry=utils.get_pydantic_model(retry, Optional[models.CreateSpeechRetry]),
+            load_balancer=utils.get_pydantic_model(
+                load_balancer, Optional[models.CreateSpeechLoadBalancer]
+            ),
+            timeout=utils.get_pydantic_model(
+                timeout, Optional[models.CreateSpeechTimeout]
+            ),
+            orq=utils.get_pydantic_model(orq, Optional[models.CreateSpeechOrq]),
+        )
+
+        req = self._build_request(
+            method="POST",
+            path="/v2/router/audio/speech",
+            base_url=base_url,
+            url_variables=url_variables,
+            request=request,
+            request_body_required=True,
+            request_has_path_params=False,
+            request_has_query_params=True,
+            user_agent_header="user-agent",
+            accept_header_value="*/*",
+            http_headers=http_headers,
+            security=self.sdk_configuration.security,
+            get_serialized_body=lambda: utils.serialize_request_body(
+                request, False, False, "json", models.CreateSpeechRequestBody
+            ),
+            allow_empty_value=None,
+            timeout_ms=timeout_ms,
+        )
+
+        if retries == UNSET:
+            if self.sdk_configuration.retry_config is not UNSET:
+                retries = self.sdk_configuration.retry_config
+
+        retry_config = None
+        if isinstance(retries, utils.RetryConfig):
+            retry_config = (retries, ["429", "500", "502", "503", "504"])
+
+        http_res = self.do_request(
+            hook_ctx=HookContext(
+                config=self.sdk_configuration,
+                base_url=base_url or "",
+                operation_id="createSpeech",
+                oauth2_scopes=None,
+                security_source=get_security_from_env(
+                    self.sdk_configuration.security, models.Security
+                ),
+            ),
+            request=req,
+            error_status_codes=["4XX", "5XX"],
+            retry_config=retry_config,
+        )
+
+        if utils.match_response(http_res, "200", "*"):
+            return
+        if utils.match_response(http_res, "4XX", "*"):
+            http_res_text = utils.stream_to_text(http_res)
+            raise models.APIError("API error occurred", http_res, http_res_text)
+        if utils.match_response(http_res, "5XX", "*"):
+            http_res_text = utils.stream_to_text(http_res)
+            raise models.APIError("API error occurred", http_res, http_res_text)
+
+        raise models.APIError("Unexpected response received", http_res)
+
+    async def create_async(
+        self,
+        *,
+        input_: str,
+        model: str,
+        voice: str,
+        response_format: Optional[
+            models_createspeechop.CreateSpeechResponseFormat
+        ] = "mp3",
+        speed: Optional[float] = 1,
+        name: Optional[str] = None,
+        fallbacks: Optional[
+            Union[
+                List[models_createspeechop.CreateSpeechFallbacks],
+                List[models_createspeechop.CreateSpeechFallbacksTypedDict],
+            ]
+        ] = None,
+        retry: Optional[
+            Union[
+                models_createspeechop.CreateSpeechRetry,
+                models_createspeechop.CreateSpeechRetryTypedDict,
+            ]
+        ] = None,
+        load_balancer: Optional[
+            Union[
+                models_createspeechop.CreateSpeechLoadBalancer,
+                models_createspeechop.CreateSpeechLoadBalancerTypedDict,
+            ]
+        ] = None,
+        timeout: Optional[
+            Union[
+                models_createspeechop.CreateSpeechTimeout,
+                models_createspeechop.CreateSpeechTimeoutTypedDict,
+            ]
+        ] = None,
+        orq: Optional[
+            Union[
+                models_createspeechop.CreateSpeechOrq,
+                models_createspeechop.CreateSpeechOrqTypedDict,
+            ]
+        ] = None,
+        retries: OptionalNullable[utils.RetryConfig] = UNSET,
+        server_url: Optional[str] = None,
+        timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
+    ):
+        r"""Create speech
+
+        Generates audio from the input text.
+
+        :param input: The text to generate audio for. The maximum length is 4096 characters
+        :param model: ID of the model to use
+        :param voice: The voice to use.
+
+            Available voices for OpenAI
+
+            `alloy`, `echo`, `fable`, `onyx`, `nova`, and `shimmer`
+
+            Available voices for ElevenLabs
+
+            `aria`, `roger`, `sarah`, `laura`, `charlie`, `george`, `callum`, `river`, `liam`, `charlotte`, `alice`, `matilda`, `will`, `jessica`, `eric`, `chris`, `brian`, `daniel`, `lily`, `bill`
+        :param response_format: The format to audio in. Supported formats are `mp3`, `opus`, `aac`, `flac`, `wav`, and `pcm`. If a format is provided but not supported by the provider, the response will be in the default format. When the provided format is not supported by the provider, the response will be in the default format.
+        :param speed: The speed of the generated audio.
+        :param name: The name to display on the trace. If not specified, the default system name will be used.
+        :param fallbacks: Array of fallback models to use if primary model fails
+        :param retry: Retry configuration for the request
+        :param load_balancer: Load balancer configuration for the request.
+        :param timeout: Timeout configuration to apply to the request. If the request exceeds the timeout, it will be retried or fallback to the next model if configured.
+        :param orq:
+        :param retries: Override the default retry configuration for this method
+        :param server_url: Override the default server URL for this method
+        :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
+        :param http_headers: Additional headers to set or replace on requests.
+        """
+        base_url = None
+        url_variables = None
+        if timeout_ms is None:
+            timeout_ms = self.sdk_configuration.timeout_ms
+
+        if timeout_ms is None:
+            timeout_ms = 600000
+
+        if server_url is not None:
+            base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
+
+        request = models.CreateSpeechRequestBody(
+            input=input_,
+            model=model,
+            voice=voice,
+            response_format=response_format,
+            speed=speed,
+            name=name,
+            fallbacks=utils.get_pydantic_model(
+                fallbacks, Optional[List[models.CreateSpeechFallbacks]]
+            ),
+            retry=utils.get_pydantic_model(retry, Optional[models.CreateSpeechRetry]),
+            load_balancer=utils.get_pydantic_model(
+                load_balancer, Optional[models.CreateSpeechLoadBalancer]
+            ),
+            timeout=utils.get_pydantic_model(
+                timeout, Optional[models.CreateSpeechTimeout]
+            ),
+            orq=utils.get_pydantic_model(orq, Optional[models.CreateSpeechOrq]),
+        )
+
+        req = self._build_request_async(
+            method="POST",
+            path="/v2/router/audio/speech",
+            base_url=base_url,
+            url_variables=url_variables,
+            request=request,
+            request_body_required=True,
+            request_has_path_params=False,
+            request_has_query_params=True,
+            user_agent_header="user-agent",
+            accept_header_value="*/*",
+            http_headers=http_headers,
+            security=self.sdk_configuration.security,
+            get_serialized_body=lambda: utils.serialize_request_body(
+                request, False, False, "json", models.CreateSpeechRequestBody
+            ),
+            allow_empty_value=None,
+            timeout_ms=timeout_ms,
+        )
+
+        if retries == UNSET:
+            if self.sdk_configuration.retry_config is not UNSET:
+                retries = self.sdk_configuration.retry_config
+
+        retry_config = None
+        if isinstance(retries, utils.RetryConfig):
+            retry_config = (retries, ["429", "500", "502", "503", "504"])
+
+        http_res = await self.do_request_async(
+            hook_ctx=HookContext(
+                config=self.sdk_configuration,
+                base_url=base_url or "",
+                operation_id="createSpeech",
+                oauth2_scopes=None,
+                security_source=get_security_from_env(
+                    self.sdk_configuration.security, models.Security
+                ),
+            ),
+            request=req,
+            error_status_codes=["4XX", "5XX"],
+            retry_config=retry_config,
+        )
+
+        if utils.match_response(http_res, "200", "*"):
+            return
+        if utils.match_response(http_res, "4XX", "*"):
+            http_res_text = await utils.stream_to_text_async(http_res)
+            raise models.APIError("API error occurred", http_res, http_res_text)
+        if utils.match_response(http_res, "5XX", "*"):
+            http_res_text = await utils.stream_to_text_async(http_res)
+            raise models.APIError("API error occurred", http_res, http_res_text)
+
+        raise models.APIError("Unexpected response received", http_res)