Merge pull request #456 from jbgriesner/fix/update-open-ai-v1-migration

Adapt openai calls to v1 migration
process-intelligence-solutions · Nov 20, 2023 · 698d53d · 698d53d
2 parents 1d058fc + b6b4070
commit 698d53d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pm4py/algo/querying/llm/connectors/openai.py b/pm4py/algo/querying/llm/connectors/openai.py
@@ -35,9 +35,9 @@ def apply(query: str, parameters: Optional[Dict[Any, Any]] = None) -> str:
 
     import openai
 
-    openai.api_key = api_key
+    client = openai.OpenAI(api_key=api_key)
 
     message = {"role": "user", "content": query}
 
-    response = openai.ChatCompletion.create(model=model, messages=[message])
+    response = client.chat.completions.create(model=model, messages=[message])
     return response["choices"][0]["message"]["content"]