Fixing errors due to using OpenAI's client object

openai_api.py

@@ -21,18 +21,18 @@ def get_embedding(text: str, model: str=EMBEDDING_MODEL) -> list[float]:
       model=model,
       input=text
     )
-    return result["data"][0]["embedding"]
+    return result.data[0].embedding
 
 
-def get_response(prompt: str, model: str=COMPLETIONS_MODEL) -> list[float]:
+def get_response(prompt: str, model: str=COMPLETIONS_MODEL) -> str:
     """Chat completion using OpenAI's GPT models."""
     response = client.chat.completions.create(
-        model = COMPLETIONS_MODEL, 
+        model=model,
         messages=[
-        {"role": "system", "content": META_PROMPT},
-        {"role": "user", "content": prompt}
+            {"role": "system", "content": META_PROMPT},
+            {"role": "user", "content": prompt}
         ],
         temperature=0,
-        max_completion_tokens=800
+        max_tokens=800
     )
-    return response["choices"][0]["message"]["content"]
+    return response.choices[0].message.content