Skip to content

enable azure for audio/whisper #3

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
wants to merge 6 commits into
base: main
Choose a base branch
from
Draft

enable azure for audio/whisper #3

Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
9adb1b2
enable azure for audio
kristapratico Aug 30, 2023
310e415
reorder overloads
kristapratico Aug 30, 2023
d66a949
add additional tests
kristapratico Sep 11, 2023
021c95b
add helper function to utils
kristapratico Sep 12, 2023
6be320b
Merge branch 'main' into whisper-1
kristapratico Sep 12, 2023
8ae05cf
simplify - azure users will just need to pass model and deployment_id
kristapratico Sep 21, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
44 changes: 35 additions & 9 deletions openai/api_resources/audio.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,9 @@ class Audio(APIResource):
OBJECT_NAME = "audio"

@classmethod
def _get_url(cls, action):
def _get_url(cls, action, deployment_id=None, api_type=None, api_version=None):
if api_type in (util.ApiType.AZURE, util.ApiType.AZURE_AD):
return f"/{cls.azure_api_prefix}/deployments/{deployment_id}/audio/{action}?api-version={api_version}"
return cls.class_url() + f"/{action}"

@classmethod
Expand Down Expand Up @@ -50,6 +52,8 @@ def transcribe(
api_type=None,
api_version=None,
organization=None,
*,
deployment_id=None,
**params,
):
requestor, files, data = cls._prepare_request(
Expand All @@ -63,7 +67,8 @@ def transcribe(
organization=organization,
**params,
)
url = cls._get_url("transcriptions")
api_type, api_version = cls._get_api_type_and_version(api_type, api_version)
url = cls._get_url("transcriptions", deployment_id=deployment_id, api_type=api_type, api_version=api_version)
response, _, api_key = requestor.request("post", url, files=files, params=data)
return util.convert_to_openai_object(
response, api_key, api_version, organization
Expand All @@ -79,6 +84,8 @@ def translate(
api_type=None,
api_version=None,
organization=None,
*,
deployment_id=None,
**params,
):
requestor, files, data = cls._prepare_request(
Expand All @@ -92,7 +99,8 @@ def translate(
organization=organization,
**params,
)
url = cls._get_url("translations")
api_type, api_version = cls._get_api_type_and_version(api_type, api_version)
url = cls._get_url("translations", deployment_id=deployment_id, api_type=api_type, api_version=api_version)
response, _, api_key = requestor.request("post", url, files=files, params=data)
return util.convert_to_openai_object(
response, api_key, api_version, organization
Expand All @@ -109,6 +117,8 @@ def transcribe_raw(
api_type=None,
api_version=None,
organization=None,
*,
deployment_id=None,
**params,
):
requestor, files, data = cls._prepare_request(
Expand All @@ -122,7 +132,8 @@ def transcribe_raw(
organization=organization,
**params,
)
url = cls._get_url("transcriptions")
api_type, api_version = cls._get_api_type_and_version(api_type, api_version)
url = cls._get_url("transcriptions", deployment_id=deployment_id, api_type=api_type, api_version=api_version)
response, _, api_key = requestor.request("post", url, files=files, params=data)
return util.convert_to_openai_object(
response, api_key, api_version, organization
Expand All @@ -139,6 +150,8 @@ def translate_raw(
api_type=None,
api_version=None,
organization=None,
*,
deployment_id=None,
**params,
):
requestor, files, data = cls._prepare_request(
Expand All @@ -152,7 +165,8 @@ def translate_raw(
organization=organization,
**params,
)
url = cls._get_url("translations")
api_type, api_version = cls._get_api_type_and_version(api_type, api_version)
url = cls._get_url("translations", deployment_id=deployment_id, api_type=api_type, api_version=api_version)
response, _, api_key = requestor.request("post", url, files=files, params=data)
return util.convert_to_openai_object(
response, api_key, api_version, organization
Expand All @@ -168,6 +182,8 @@ async def atranscribe(
api_type=None,
api_version=None,
organization=None,
*,
deployment_id=None,
**params,
):
requestor, files, data = cls._prepare_request(
Expand All @@ -181,7 +197,8 @@ async def atranscribe(
organization=organization,
**params,
)
url = cls._get_url("transcriptions")
api_type, api_version = cls._get_api_type_and_version(api_type, api_version)
url = cls._get_url("transcriptions", deployment_id=deployment_id, api_type=api_type, api_version=api_version)
response, _, api_key = await requestor.arequest(
"post", url, files=files, params=data
)
Expand All @@ -199,6 +216,8 @@ async def atranslate(
api_type=None,
api_version=None,
organization=None,
*,
deployment_id=None,
**params,
):
requestor, files, data = cls._prepare_request(
Expand All @@ -212,7 +231,8 @@ async def atranslate(
organization=organization,
**params,
)
url = cls._get_url("translations")
api_type, api_version = cls._get_api_type_and_version(api_type, api_version)
url = cls._get_url("translations", deployment_id=deployment_id, api_type=api_type, api_version=api_version)
response, _, api_key = await requestor.arequest(
"post", url, files=files, params=data
)
Expand All @@ -231,6 +251,8 @@ async def atranscribe_raw(
api_type=None,
api_version=None,
organization=None,
*,
deployment_id=None,
**params,
):
requestor, files, data = cls._prepare_request(
Expand All @@ -244,7 +266,8 @@ async def atranscribe_raw(
organization=organization,
**params,
)
url = cls._get_url("transcriptions")
api_type, api_version = cls._get_api_type_and_version(api_type, api_version)
url = cls._get_url("transcriptions", deployment_id=deployment_id, api_type=api_type, api_version=api_version)
response, _, api_key = await requestor.arequest(
"post", url, files=files, params=data
)
Expand All @@ -263,6 +286,8 @@ async def atranslate_raw(
api_type=None,
api_version=None,
organization=None,
*,
deployment_id=None,
**params,
):
requestor, files, data = cls._prepare_request(
Expand All @@ -276,7 +301,8 @@ async def atranslate_raw(
organization=organization,
**params,
)
url = cls._get_url("translations")
api_type, api_version = cls._get_api_type_and_version(api_type, api_version)
url = cls._get_url("translations", deployment_id=deployment_id, api_type=api_type, api_version=api_version)
response, _, api_key = await requestor.arequest(
"post", url, files=files, params=data
)
Expand Down