From b6b4070e38b695fb7b73c910023e8ede8dd765cd Mon Sep 17 00:00:00 2001 From: JB Griesner Date: Sun, 19 Nov 2023 23:54:45 +0100 Subject: [PATCH] Adapt openai calls to v1 migration --- pm4py/algo/querying/llm/connectors/openai.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pm4py/algo/querying/llm/connectors/openai.py b/pm4py/algo/querying/llm/connectors/openai.py index 887c6ff7c..b9b4af572 100644 --- 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"]