From 54376aa4275dc4d4e72142fd1aa843fe5619083f Mon Sep 17 00:00:00 2001
From: "Pavelka, Roman (ADV D EU CZ PDS1 CIO 1)" <roman.pavelka@siemens.com>
Date: Wed, 8 Nov 2023 14:34:27 +0100
Subject: [PATCH] Fix core

---
 core.py | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/core.py b/core.py
index 997bafe..2d85f55 100644
--- a/core.py
+++ b/core.py
@@ -6,9 +6,10 @@
 MODEL = "gpt-4"
 TEMPERATURE = 0.0
 
-api_key_path = os.path.join(os.path.dirname(__file__), '.api_key')
-with open(api_key_path, 'r') as f:
-    openai.api_key = f.read().strip()
+if 'OPENAI_API_KEY' not in os.environ:
+    api_key_path = os.path.join(os.path.dirname(__file__), '.api_key')
+    with open(api_key_path, 'r') as f:
+        os.environ['OPENAI_API_KEY'] = f.read().strip()
 
 
 class GptCore:
@@ -34,18 +35,20 @@ def __init__(self, input, output):
 
         self.messages = []
 
+        self.client = openai.OpenAI()
+
     def main(self):
         price = 0
         while prompt := self.input():
             self.messages.append({"role": "user", "content": prompt})
 
-            response = openai.ChatCompletion.create(
+            response = self.client.chat.completions.create(
                 model=MODEL, messages=self.messages, temperature=TEMPERATURE)
 
-            message = response.choices[0]["message"]
+            message = response.choices[0].message
             self.messages.append(message)
 
-            content = message["content"].strip()
+            content = message.content.strip()
 
             usage = response.usage
             prompt_tokens, completion_tokens = (