orq-ai-sdk 4.2.0rc28__py3-none-any.whl → 4.2.6__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,251 @@
+"""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 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,
+        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 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,
+            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,
+        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 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,
+            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)

orq_ai_sdk/transcriptions.py

@@ -0,0 +1,326 @@
+"""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 createtranscriptionop as models_createtranscriptionop
+from orq_ai_sdk.types import OptionalNullable, UNSET
+from orq_ai_sdk.utils import get_security_from_env
+from orq_ai_sdk.utils.unmarshal_json_response import unmarshal_json_response
+from typing import Any, List, Mapping, Optional, Union
+
+
+class Transcriptions(BaseSDK):
+    def create(
+        self,
+        *,
+        model: str,
+        prompt: Optional[str] = None,
+        enable_logging: Optional[bool] = True,
+        diarize: Optional[bool] = False,
+        response_format: Optional[
+            models_createtranscriptionop.CreateTranscriptionResponseFormat
+        ] = None,
+        tag_audio_events: Optional[bool] = True,
+        num_speakers: Optional[float] = None,
+        timestamps_granularity: Optional[
+            models_createtranscriptionop.TimestampsGranularity
+        ] = "word",
+        temperature: Optional[float] = None,
+        language: Optional[str] = None,
+        timestamp_granularities: Optional[
+            List[models_createtranscriptionop.TimestampGranularities]
+        ] = None,
+        orq: Optional[
+            Union[
+                models_createtranscriptionop.CreateTranscriptionOrq,
+                models_createtranscriptionop.CreateTranscriptionOrqTypedDict,
+            ]
+        ] = None,
+        file: Optional[
+            Union[
+                models_createtranscriptionop.CreateTranscriptionFile,
+                models_createtranscriptionop.CreateTranscriptionFileTypedDict,
+            ]
+        ] = None,
+        retries: OptionalNullable[utils.RetryConfig] = UNSET,
+        server_url: Optional[str] = None,
+        timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
+    ) -> models.CreateTranscriptionResponseBody:
+        r"""Create transcription
+
+        :param model: ID of the model to use
+        :param prompt: An optional text to guide the model's style or continue a previous audio segment. The prompt should match the audio language.
+        :param enable_logging: When enable_logging is set to false, zero retention mode is used. This disables history features like request stitching and is only available to enterprise customers.
+        :param diarize: Whether to annotate which speaker is currently talking in the uploaded file.
+        :param response_format: The format of the transcript output, in one of these options: json, text, srt, verbose_json, or vtt.
+        :param tag_audio_events: Whether to tag audio events like (laughter), (footsteps), etc. in the transcription.
+        :param num_speakers: The maximum amount of speakers talking in the uploaded file. Helps with predicting who speaks when, the maximum is 32.
+        :param timestamps_granularity: The granularity of the timestamps in the transcription. Word provides word-level timestamps and character provides character-level timestamps per word.
+        :param temperature: The sampling temperature, between 0 and 1. Higher values like 0.8 will make the output more random, while lower values like 0.2 will make it more focused and deterministic. If set to 0, the model will use log probability to automatically increase the temperature until certain thresholds are hit.
+        :param language: The language of the input audio. Supplying the input language in ISO-639-1 format will improve accuracy and latency.
+        :param timestamp_granularities: The timestamp granularities to populate for this transcription. response_format must be set to verbose_json to use timestamp granularities. Either or both of these options are supported: \"word\" or \"segment\". Note: There is no additional latency for segment timestamps, but generating word timestamps incurs additional latency.
+        :param orq:
+        :param file: The audio file object (not file name) to transcribe, in one of these formats: flac, mp3, mp4, mpeg, mpga, m4a, ogg, wav, or webm.
+        :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.CreateTranscriptionRequestBody(
+            model=model,
+            prompt=prompt,
+            enable_logging=enable_logging,
+            diarize=diarize,
+            response_format=response_format,
+            tag_audio_events=tag_audio_events,
+            num_speakers=num_speakers,
+            timestamps_granularity=timestamps_granularity,
+            temperature=temperature,
+            language=language,
+            timestamp_granularities=timestamp_granularities,
+            orq=utils.get_pydantic_model(orq, Optional[models.CreateTranscriptionOrq]),
+            file=utils.get_pydantic_model(
+                file, Optional[models.CreateTranscriptionFile]
+            ),
+        )
+
+        req = self._build_request(
+            method="POST",
+            path="/v2/router/audio/transcriptions",
+            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="application/json",
+            http_headers=http_headers,
+            security=self.sdk_configuration.security,
+            get_serialized_body=lambda: utils.serialize_request_body(
+                request,
+                False,
+                False,
+                "multipart",
+                models.CreateTranscriptionRequestBody,
+            ),
+            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="createTranscription",
+                oauth2_scopes=None,
+                security_source=get_security_from_env(
+                    self.sdk_configuration.security, models.Security
+                ),
+            ),
+            request=req,
+            error_status_codes=["422", "4XX", "5XX"],
+            retry_config=retry_config,
+        )
+
+        response_data: Any = None
+        if utils.match_response(http_res, "200", "application/json"):
+            return unmarshal_json_response(
+                models.CreateTranscriptionResponseBody, http_res
+            )
+        if utils.match_response(http_res, "422", "application/json"):
+            response_data = unmarshal_json_response(
+                models.CreateTranscriptionRouterAudioTranscriptionsResponseBodyData,
+                http_res,
+            )
+            raise models.CreateTranscriptionRouterAudioTranscriptionsResponseBody(
+                response_data, http_res
+            )
+        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,
+        *,
+        model: str,
+        prompt: Optional[str] = None,
+        enable_logging: Optional[bool] = True,
+        diarize: Optional[bool] = False,
+        response_format: Optional[
+            models_createtranscriptionop.CreateTranscriptionResponseFormat
+        ] = None,
+        tag_audio_events: Optional[bool] = True,
+        num_speakers: Optional[float] = None,
+        timestamps_granularity: Optional[
+            models_createtranscriptionop.TimestampsGranularity
+        ] = "word",
+        temperature: Optional[float] = None,
+        language: Optional[str] = None,
+        timestamp_granularities: Optional[
+            List[models_createtranscriptionop.TimestampGranularities]
+        ] = None,
+        orq: Optional[
+            Union[
+                models_createtranscriptionop.CreateTranscriptionOrq,
+                models_createtranscriptionop.CreateTranscriptionOrqTypedDict,
+            ]
+        ] = None,
+        file: Optional[
+            Union[
+                models_createtranscriptionop.CreateTranscriptionFile,
+                models_createtranscriptionop.CreateTranscriptionFileTypedDict,
+            ]
+        ] = None,
+        retries: OptionalNullable[utils.RetryConfig] = UNSET,
+        server_url: Optional[str] = None,
+        timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
+    ) -> models.CreateTranscriptionResponseBody:
+        r"""Create transcription
+
+        :param model: ID of the model to use
+        :param prompt: An optional text to guide the model's style or continue a previous audio segment. The prompt should match the audio language.
+        :param enable_logging: When enable_logging is set to false, zero retention mode is used. This disables history features like request stitching and is only available to enterprise customers.
+        :param diarize: Whether to annotate which speaker is currently talking in the uploaded file.
+        :param response_format: The format of the transcript output, in one of these options: json, text, srt, verbose_json, or vtt.
+        :param tag_audio_events: Whether to tag audio events like (laughter), (footsteps), etc. in the transcription.
+        :param num_speakers: The maximum amount of speakers talking in the uploaded file. Helps with predicting who speaks when, the maximum is 32.
+        :param timestamps_granularity: The granularity of the timestamps in the transcription. Word provides word-level timestamps and character provides character-level timestamps per word.
+        :param temperature: The sampling temperature, between 0 and 1. Higher values like 0.8 will make the output more random, while lower values like 0.2 will make it more focused and deterministic. If set to 0, the model will use log probability to automatically increase the temperature until certain thresholds are hit.
+        :param language: The language of the input audio. Supplying the input language in ISO-639-1 format will improve accuracy and latency.
+        :param timestamp_granularities: The timestamp granularities to populate for this transcription. response_format must be set to verbose_json to use timestamp granularities. Either or both of these options are supported: \"word\" or \"segment\". Note: There is no additional latency for segment timestamps, but generating word timestamps incurs additional latency.
+        :param orq:
+        :param file: The audio file object (not file name) to transcribe, in one of these formats: flac, mp3, mp4, mpeg, mpga, m4a, ogg, wav, or webm.
+        :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.CreateTranscriptionRequestBody(
+            model=model,
+            prompt=prompt,
+            enable_logging=enable_logging,
+            diarize=diarize,
+            response_format=response_format,
+            tag_audio_events=tag_audio_events,
+            num_speakers=num_speakers,
+            timestamps_granularity=timestamps_granularity,
+            temperature=temperature,
+            language=language,
+            timestamp_granularities=timestamp_granularities,
+            orq=utils.get_pydantic_model(orq, Optional[models.CreateTranscriptionOrq]),
+            file=utils.get_pydantic_model(
+                file, Optional[models.CreateTranscriptionFile]
+            ),
+        )
+
+        req = self._build_request_async(
+            method="POST",
+            path="/v2/router/audio/transcriptions",
+            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="application/json",
+            http_headers=http_headers,
+            security=self.sdk_configuration.security,
+            get_serialized_body=lambda: utils.serialize_request_body(
+                request,
+                False,
+                False,
+                "multipart",
+                models.CreateTranscriptionRequestBody,
+            ),
+            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="createTranscription",
+                oauth2_scopes=None,
+                security_source=get_security_from_env(
+                    self.sdk_configuration.security, models.Security
+                ),
+            ),
+            request=req,
+            error_status_codes=["422", "4XX", "5XX"],
+            retry_config=retry_config,
+        )
+
+        response_data: Any = None
+        if utils.match_response(http_res, "200", "application/json"):
+            return unmarshal_json_response(
+                models.CreateTranscriptionResponseBody, http_res
+            )
+        if utils.match_response(http_res, "422", "application/json"):
+            response_data = unmarshal_json_response(
+                models.CreateTranscriptionRouterAudioTranscriptionsResponseBodyData,
+                http_res,
+            )
+            raise models.CreateTranscriptionRouterAudioTranscriptionsResponseBody(
+                response_data, http_res
+            )
+        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)