Update requirements

Author: BenConstable9

text_2_sql/autogen/src/autogen_text_2_sql/creators/llm_model_creator.py

@@ -1,6 +1,6 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
-from autogen_ext.models import AzureOpenAIChatCompletionClient
+from autogen_ext.models.openai import AzureOpenAIChatCompletionClient
 from text_2_sql_core.utils.environment import IdentityType, get_identity_type
 
 from azure.identity import DefaultAzureCredential, get_bearer_token_provider
@@ -57,7 +57,7 @@ def gpt_4o_mini_model(cls) -> AzureOpenAIChatCompletionClient:
         return AzureOpenAIChatCompletionClient(
             azure_deployment=os.environ["OpenAI__MiniCompletionDeployment"],
             model=os.environ["OpenAI__MiniCompletionDeployment"],
-            api_version="2024-08-01-preview",
+            api_version=os.environ["OpenAI__ApiVersion"],
             azure_endpoint=os.environ["OpenAI__Endpoint"],
             azure_ad_token_provider=token_provider,
             api_key=api_key,
@@ -75,7 +75,7 @@ def gpt_4o_model(cls) -> AzureOpenAIChatCompletionClient:
         return AzureOpenAIChatCompletionClient(
             azure_deployment=os.environ["OpenAI__CompletionDeployment"],
             model=os.environ["OpenAI__CompletionDeployment"],
-            api_version="2024-08-01-preview",
+            api_version=os.environ["OpenAI__ApiVersion"],
             azure_endpoint=os.environ["OpenAI__Endpoint"],
             azure_ad_token_provider=token_provider,
             api_key=api_key,

text_2_sql/text_2_sql_core/src/text_2_sql_core/connectors/open_ai.py

@@ -34,10 +34,13 @@ def get_authentication_properties(cls) -> dict:
         return token_provider, api_key
 
     async def run_completion_request(self, messages: list[dict], temperature=0):
+        token_provider, api_key = self.get_authentication_properties()
         async with AsyncAzureOpenAI(
-            api_key=os.environ["OpenAI__ApiKey"],
-            azure_endpoint=os.environ["OpenAI__Endpoint"],
+            azure_deployment=os.environ["OpenAI__MiniCompletionDeployment"],
             api_version=os.environ["OpenAI__ApiVersion"],
+            azure_endpoint=os.environ["OpenAI__Endpoint"],
+            azure_ad_token_provider=token_provider,
+            api_key=api_key,
         ) as open_ai_client:
             response = await open_ai_client.chat.completions.create(
                 model=os.environ["OpenAI__MiniCompletionDeployment"],