Skip to content
Merged
Changes from 1 commit
Commits
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
9 changes: 5 additions & 4 deletions toolium/utils/ai_utils/openai.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Comment thread
rgonalo marked this conversation as resolved.
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:
Expand Down
Loading