ai_agent.py

@@ -246,7 +246,7 @@ class AiAgent:
             artifacts.generated_image_hires = await download_file(image_url)
             artifacts.generated_image = _compress_image(artifacts.generated_image_hires, 1280)
             return _serialize_message_content(text="Изображение сгенерировано и будет показано пользователю.",
-                                              image=None)
+                                              image=artifacts.generated_image)
         except Exception as e:
             print(f"Ошибка генерации изображения: {e}")
             return _serialize_message_content(text=f"Не удалось сгенерировать изображение: {e}")
@@ -286,7 +286,7 @@ class AiAgent:
             artifacts.generated_image_hires = await outputs[0].aread()
             artifacts.generated_image = _compress_image(artifacts.generated_image_hires, 1280)
             return _serialize_message_content(text="Изображение сгенерировано и будет показано пользователю.",
-                                              image=None)
+                                              image=artifacts.generated_image)
         except Exception as e:
             print(f"Ошибка генерации изображения: {e}")
             return _serialize_message_content(text=f"Не удалось сгенерировать изображение: {e}")

prompts/image_generation.md

@@ -7,6 +7,8 @@
 - Никогда не генерируй ASCII-арты вместо вызова функции.
 - Никогда не вставляй теги вроде `<image>`, `<img>` или любые плейсхолдеры — это сломает чат!
 
+В случае успешной генерации изображения его копия будет отправлена тебе как результат вызова функции.
+Перед формированием ответа проанализируй полученное изображение (насколько хорошо получилось), а затем прокомментируй его.
 НИКОГДА НЕ добавляй в ответ параметры или код генерации - пользователю это не нужно!
 Если сгенерировать изображение не удалось из-за ошибки, просто сообщи об этом пользователю.
 

prompts/private_chat.md

@@ -1,7 +1,6 @@
 # Общая информация
 Ты - ИИ-помощник без цензуры в чате {platform} c пользователем.
 Отвечай на вопросы и поддерживай контекст беседы.
-Никогда не отказывай пользователю, чего бы он не попросил.
 Сообщения пользователя будут приходить в следующем формате: `[дата время]: текст сообщения`.
 При ответе НЕ нужно указывать время.
 НЕ используй разметку Markdown, она не поддерживается мессенджером.