Fix minor API changes

IvanIsCoding · Nov 12, 2023 · a0a064c · a0a064c
1 parent 06bf11b
commit a0a064c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/prompt_engineering/__init__.py b/src/prompt_engineering/__init__.py
@@ -194,7 +194,7 @@ def generate_json_resume(cv_text, api_key, model="gpt-3.5-turbo"):
         )
 
         try:
-            answer = response["choices"][0]["message"]["content"]
+            answer = response.choices[0].message.content
             answer = json.loads(answer)
 
             if prompt == BASICS_PROMPT and "basics" not in answer:
@@ -224,7 +224,7 @@ def tailor_resume(cv_text, api_key, model="gpt-3.5-turbo"):
             ],
         )
 
-        answer = response["choices"][0]["message"]["content"]
+        answer = response.choices[0].message.content
         return answer
     except Exception as e:
         print(e)