@@ -766,18 +766,12 @@ class Chatgpt(SimpleService):
                 and (data := resp.json())
                 and data.get('output')
                 and (
-                    output := data['output']
+                    content := data['output'][-1]['content'][0]['text']
                 )
             ):
                 input_tokens = resp.json()['usage']['input_tokens']
                 output_tokens = resp.json()['usage']['output_tokens']
-                response = AIMessage(content=output[-1]['content'][0]['text'])
-                # for o in output:
-                #     if o['type'] == 'code_interpreter_call':
-                #         content += f'Код: {o['code']}\n\n'
-                #     if o['type'] == 'message':
-                #         content += o['content'][0]['text']
-                #     response = AIMessage(content=content)
+                response = AIMessage(content=content)
                 return input_tokens, output_tokens, response
             else:
                 raise Exception('GPT not answer correctly, please retry later')