client and api_key name

Author: manuelporrasojeda

toolium/utils/ai_utils/openai.py

@@ -45,22 +45,23 @@ def openai_request(system_message, user_message, model_name=None, azure=False, *
     config = DriverWrappersPool.get_default_wrapper().config
     model_name = model_name or config.get_optional('AI', 'openai_model', 'gpt-4o-mini')
     logger.info('Calling to OpenAI API with model %s', model_name)
-
     response_format = None
     if kwargs.get('response_format'):
         response_format = kwargs.pop('response_format')
         kwargs.pop('response_format', None)
-    client = AzureOpenAI(**kwargs) if azure else OpenAI(**kwargs)
     if azure:
         for key in ('azure_api_key', 'azure_endpoint', 'api_version', 'azure_deployment'):
             value = config.get_optional('AI', key)
             if value:
-                kwargs.setdefault(key, value)
+                kwargs_key = 'api_key' if key == 'azure_api_key' else key
+                kwargs.setdefault(kwargs_key, value)
     else:
         for key in ('openai_api_key', 'openai_temperature'):
             value = config.get_optional('AI', key)
             if value:
-                kwargs.setdefault(key, value)
+                kwargs_key = 'api_key' if key == 'openai_api_key' else key
+                kwargs.setdefault(kwargs_key, value)
+    client = AzureOpenAI(**kwargs) if azure else OpenAI(**kwargs)
     messages = []
     if isinstance(system_message, list):
         for prompt in system_message: