Update openai_client.py

openai_client.py

@@ -58,8 +58,19 @@ def refactor_code(code_snippet):
                 },
             ],
         )
-        refactored_code = response.choices[0].message.content
-        return refactored_code
+        refactor = response.text if hasattr(response, "text") else str(response)
+
+        # Check if feedback is a Message object and extract text if necessary
+        if hasattr(response, "content") and isinstance(response.content, list):
+            refactor = "\n\n".join(
+                [
+                    text_block.text
+                    for text_block in response.content
+                    if hasattr(text_block, "text")
+                ]
+            )
+
+        return refactor
     except Exception as e:
         return "Sorry, an error occurred while refactoring your code. Please try again later."
 
@@ -146,7 +157,18 @@ def remove_code_errors(code_snippet):
                 },
             ],
         )
-        error_removal_suggestions = response.choices[0].message.content
-        return error_removal_suggestions
+        code_errors = response.text if hasattr(response, "text") else str(response)
+
+        # Check if feedback is a Message object and extract text if necessary
+        if hasattr(response, "content") and isinstance(response.content, list):
+            code_errors = "\n\n".join(
+                [
+                    text_block.text
+                    for text_block in response.content
+                    if hasattr(text_block, "text")
+                ]
+            )
+
+        return code_errors
     except Exception as e:
         return "Sorry, an error occurred while removing code errors. Please try again later."