needware-cli 1.7.3 → 1.7.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "needware-cli",
-  "version": "1.7.3",
+  "version": "1.7.5",
   "description": "一个功能强大的 Node.js 命令行工具",
   "type": "module",
   "main": "dist/index.js",

package/skills/ai-integration/SKILL.md

@@ -590,107 +590,60 @@ Requirements:
       }),
     });
 
-    if (!response.ok) {
-      const errorText = await response.text();
-      console.error("AI service error:", response.status, errorText);
-
-      if (response.status === 429) {
-        return new Response(
-          JSON.stringify({ error: "Request rate too high, please try again later" }),
-          { status: 429, headers: { "Content-Type": "application/json", ...corsHeaders } }
-        );
-      }
-
-      if (response.status === 402) {
-        return new Response(
-          JSON.stringify({ error: "AI service quota exhausted" }),
-          { status: 402, headers: { "Content-Type": "application/json", ...corsHeaders } }
-        );
-      }
+      // 非流式响应
+    const result = await response.json();
 
-      let errorMessage = "AI service temporarily unavailable";
-      try {
-        const errorJson = JSON.parse(errorText);
-        errorMessage = errorJson.error?.message || errorJson.message || errorMessage;
-      } catch {
-        if (errorText) errorMessage = errorText.slice(0, 200);
-      }
 
-      return new Response(
-        JSON.stringify({ error: errorMessage }),
-        { status: 500, headers: { "Content-Type": "application/json", ...corsHeaders } }
-      );
-    }
+    // 解析响应，提取生成的图片
+    let generatedImageUrl: string | null = null;
+    let textContent: string | null = null;
 
-    const data = await response.json();
-    console.log("AI response:", JSON.stringify(data, null, 2));
-
-    // ⚠️ IMPORTANT: Parse image from multiple possible response formats
-    let generatedImage: string | null = null;
-    const message = data.choices?.[0]?.message;
-
-    // Format 1: content_parts (Gemini native format)
-    if (message?.content_parts && Array.isArray(message.content_parts)) {
-      for (const part of message.content_parts) {
-        if (part.inline_data?.data) {
-          const mimeType = part.inline_data.mime_type || "image/png";
-          generatedImage = `data:${mimeType};base64,${part.inline_data.data}`;
-          break;
-        }
-      }
-    }
 
-    // Format 2: content array with various structures
-    if (!generatedImage && Array.isArray(message?.content)) {
-      for (const item of message.content) {
-        if (item.type === "image_url" && item.image_url?.url) {
-          generatedImage = item.image_url.url;
-          break;
-        }
-        if (item.type === "image" && item.data) {
-          generatedImage = `data:image/png;base64,${item.data}`;
-          break;
-        }
-        if (item.inline_data?.data) {
-          const mimeType = item.inline_data.mime_type || "image/png";
-          generatedImage = `data:${mimeType};base64,${item.inline_data.data}`;
-          break;
+    const messageContent = result.choices?.[0]?.message?.content;
+    
+    // 检查是否有图片内容
+    if (Array.isArray(messageContent)) {
+      for (const part of messageContent) {
+        if (part.type === 'image_url' && part.image_url?.url) {
+          generatedImageUrl = part.image_url.url;
+        } else if (part.type === 'text') {
+          textContent = part.text;
         }
       }
-    }
-
-    // Format 3: Direct base64 string in content
-    if (!generatedImage && typeof message?.content === "string" && message.content) {
-      if (message.content.startsWith("data:image")) {
-        generatedImage = message.content;
-      } else if (message.content.length > 1000 && /^[A-Za-z0-9+/=\s]+$/.test(message.content)) {
-        // Looks like raw base64 data
-        generatedImage = `data:image/png;base64,${message.content.replace(/\s/g, '')}`;
+    } else if (typeof messageContent === 'string') {
+      // 检查是否包含 base64 图片或图片 URL
+      const base64Match = messageContent.match(/data:image\/[^;]+;base64,[A-Za-z0-9+/=]+/);
+      const urlMatch = messageContent.match(/https?:\/\/[^\s"']+\.(jpg|jpeg|png|gif|webp)/i);
+      
+      if (base64Match) {
+        generatedImageUrl = base64Match[0];
+      } else if (urlMatch) {
+        generatedImageUrl = urlMatch[0];
       }
+      textContent = messageContent;
     }
 
-    if (!generatedImage) {
-      const reasoning = message?.reasoning;
-      console.error("No image in response. Full response:", JSON.stringify(data));
-
-      return new Response(
-        JSON.stringify({
-          error: "AI failed to generate try-on image, please retry or use a different photo",
-          debug: reasoning ? reasoning.slice(0, 100) : "Unable to extract image"
-        }),
-        { status: 500, headers: { "Content-Type": "application/json", ...corsHeaders } }
-      );
-    }
-
-    console.log("Virtual try-on completed successfully");
+    // 包装响应
+    const virtualTryOnResponse = {
+      success: response.ok,
+      data: {
+        generatedImageUrl,
+        textContent,
+        personImageUrl: body.personImageUrl,
+        garmentImageUrl: body.garmentImageUrl,
+        model: body.model || 'google/gemini-3-pro-image-preview',
+        usage: result.usage || null,
+      },
+      rawResponse: result,
+    };
 
-    return new Response(
-      JSON.stringify({
-        success: true,
-        image: generatedImage
-      }),
-      { status: 200, headers: { "Content-Type": "application/json", ...corsHeaders } }
-    );
+    return new Response(JSON.stringify(virtualTryOnResponse), {
+      status: response.status,
+      headers: {
+        "Content-Type": "application/json",
+        ...corsHeaders
+      },
+    });
 
   } catch (error: any) {
     console.error("Virtual try-on error:", error);