From 6090b4b10e41462bc06551473aadb1a026c9d14a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Christian=20M=C3=BCrtz?= Date: Tue, 14 Mar 2023 15:07:29 +0100 Subject: [PATCH] Hardcode azure chat completions preview version --- openai/api_resources/chat_completion.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/openai/api_resources/chat_completion.py b/openai/api_resources/chat_completion.py index 39fb58b33a..e32f393249 100644 --- a/openai/api_resources/chat_completion.py +++ b/openai/api_resources/chat_completion.py @@ -3,12 +3,26 @@ from openai import util from openai.api_resources.abstract.engine_api_resource import EngineAPIResource from openai.error import TryAgain +from openai.util import ApiType class ChatCompletion(EngineAPIResource): engine_required = False OBJECT_NAME = "chat.completions" + _azure_preview_version = "2023-03-15-preview" + + @classmethod + def _get_api_type_and_version( + cls, api_type = None, api_version = None + ): + api_type, base_api_version = super()._get_api_type_and_version() + if api_type in (ApiType.AZURE, ApiType.AZURE_AD): + # This override is only temporary: DallE and GPT endpoint versioning is currently out of sync but will be aligned soon. + return (api_type, api_version or ChatCompletion._azure_preview_version) + else: + return (api_type, base_api_version) + @classmethod def create(cls, *args, **kwargs): """