Commit a1366e0

benny-dou <60535774+benny-dou@users.noreply.github.com>
2026-01-26 12:02:38
fix(ai): handle media only message for text generation command
main
1 parent ddb188d
Changed files (1)
src
ai
main.py
src/ai/main.py
@@ -26,8 +26,11 @@ async def ai_text_generation(client: Client, message: Message, **kwargs) -> dict
     this_msg = message
     prompt = texts.removeprefix(PREFIX.AI_TEXT_GENERATION).strip()
     prompt = re.sub(r"^@([a-zA-Z0-9_\-\.]+)(\s+)?", "", prompt, flags=re.DOTALL).strip()
-    if not prompt and message.reply_to_message:
-        message = this_msg.reply_to_message
+    if not prompt:
+        if this_msg.media is not None:  # `/ai` + [meida]
+            pass
+        elif message.reply_to_message:  # `/ai` reply to [message]
+            message = this_msg.reply_to_message
     model_configs = await get_text_model_configs(this_msg)
     if not model_configs:
         return {}