Add required env parameters for azure usage

This commit is contained in:
Kari Ahdan
2023-04-10 16:43:06 +03:00
parent 4d42e14d3d
commit a504f0bdd3
5 changed files with 12 additions and 4 deletions

View File

@@ -9,7 +9,7 @@ def create_chat_completion(messages, model=None, temperature=None, max_tokens=No
"""Create a chat completion using the OpenAI API"""
if cfg.use_azure:
response = openai.ChatCompletion.create(
deployment_id=cfg.openai_deployment_id,
deployment_id=cfg.azure_chat_deployment_id,
model=model,
messages=messages,
temperature=temperature,