Switched order of AzureAsyncOpenAI to AsyncAzureOpenAI

Switched order of AzureAsyncOpenAI to AsyncAzureOpenAI to match name in repo https://github.com/openai/openai-python/blob/main/src/openai/lib/azure.py
rlouf · Apr 11, 2024 · 366ea1b · 366ea1b
1 parent ae9ae50
commit 366ea1b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/outlines/models/openai.py b/outlines/models/openai.py
@@ -447,7 +447,7 @@ def azure_openai(
 ):
  try:
  import tiktoken
- from openai import AzureAsyncOpenAI
+ from openai import AsyncAzureOpenAI
  except ImportError:
  raise ImportError(
  "The `openai` and `tiktoken` libraries needs to be installed in order to use Outlines' Azure OpenAI integration."
@@ -458,7 +458,7 @@ def azure_openai(
  if config is None:
  config = OpenAIConfig(model=deployment_name)
 
- client = AzureAsyncOpenAI(
+ client = AsyncAzureOpenAI(
  azure_endpoint=azure_endpoint, api_version=api_version, api_key=api_key
  )
  tokenizer = tiktoken.encoding_for_model(deployment_name)