Merge branch 'hotfixes' into integration

pm4py · Nov 20, 2023 · c38d105 · c38d105
2 parents afe8aff + e33cfe6
commit c38d105
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pm4py/algo/querying/llm/connectors/openai.py b/pm4py/algo/querying/llm/connectors/openai.py
@@ -24,4 +24,5 @@ def apply(query: str, parameters: Optional[Dict[Any, Any]] = None) -> str:
     message = {"role": "user", "content": query}
 
     response = client.chat.completions.create(model=model, messages=[message])
-    return response["choices"][0]["message"]["content"]
+
+    return response.choices[0].message.content