changed "data" of the http request

test_draftgpt.py

@@ -64,6 +64,18 @@ def draft_gpt(user_input, openai_api_key=os.environ["OPENAI_API_KEY"], gpt_model
         "Authorization": f"Bearer {openai_api_key}",
     }
 
+
+
+    data = {
+        "model": gpt_model,
+        "messages": [
+            {"role": "system", "content": "You are responsible for analyzing incident root causes at a software engineering company. Your tasks include extracting messages from users and systems and then determining the root cause of the incident in the incident report."},
+            {
+                "role": "user",
+                "content": user_input,
+            },
+        ],
+    }    
     if gpt_model[:2] == "gpt":
         url = "https://api.openai.com/v1/chat/completions"
 
@@ -74,17 +86,22 @@ def draft_gpt(user_input, openai_api_key=os.environ["OPENAI_API_KEY"], gpt_model
             "Authorization": f"Bearer {openai_api_key}",
             "OpenAI-Beta": "assistants=v2"
             }
+        data = {
 
-    data = {
         "model": gpt_model,
-        "messages": [
-            {"role": "system", "content": "You are responsible for analyzing incident root causes at a software engineering company. Your tasks include extracting messages from users and systems and then determining the root cause of the incident in the incident report."},
-            {
-                "role": "user",
-                "content": user_input,
-            },
-        ],
+        "messages":[
+    {
+      "role": "user",
+      "content": user_input,
+      "attachments": [
+        {
+          "tools": [{"type": "code_interpreter"}]
+        }
+      ]
     }
+  ]
+}
+
 
     response = requests.post(url, headers=headers, json=data)
     # Check if the request was successful