Update main.py

main.py

@@ -407,12 +407,26 @@ async def process_image_generation(msg_data: str, session: requests.Session, hea
         job_id, prompt, negative = match.groups()
         print(f"Starting image generation process for job_id: {job_id}")
         
-        # 构建思考消息的开始部分
-        think_msg = "<think>\n"
-        think_msg += "🎨 Generating image...\n\n"
-        think_msg += f"Prompt: {prompt}\n"
+        # 返回多个消息块
+        messages = []
+        
+        # 第一个消息块：开始思考
+        start_msg = "<think>\n🎨 Generating image...\n\n"
+        start_msg += f"Prompt: {prompt}\n"
+        
+        messages.append({
+            "id": f"chatcmpl-{chat_id}-start",
+            "object": "chat.completion.chunk",
+            "created": int(time.time()),
+            "model": "AkashGen",
+            "choices": [{
+                "delta": {"content": start_msg},
+                "index": 0,
+                "finish_reason": None
+            }]
+        })
         
-        # 记录开始时间（在检查图片状态前）
+        # 记录开始时间
         start_time = time.time()
         
         # 检查图片状态和上传
@@ -421,27 +435,22 @@ async def process_image_generation(msg_data: str, session: requests.Session, hea
         # 计算实际经过的时间
         elapsed_time = time.time() - start_time
         
-        # 完成思考部分
-        think_msg += f"\n🤔 Thinking for {elapsed_time:.1f}s...\n"
-        think_msg += "</think>"
-        
-        # 返回两个独立的消息块
-        messages = []
+        # 第二个消息块：结束思考
+        end_think_msg = f"\n🤔 Thinking for {elapsed_time:.1f}s...\n</think>"
         
-        # 第一个消息块：思考过程
         messages.append({
-            "id": f"chatcmpl-{chat_id}-think",
+            "id": f"chatcmpl-{chat_id}-end-think",
             "object": "chat.completion.chunk",
             "created": int(time.time()),
             "model": "AkashGen",
             "choices": [{
-                "delta": {"content": think_msg},
+                "delta": {"content": end_think_msg},
                 "index": 0,
                 "finish_reason": None
             }]
         })
         
-        # 第二个消息块：图片结果
+        # 第三个消息块：图片结果
         if result:
             image_msg = f"\n\n![Generated Image]({result})"
             messages.append({