增加提示词优化工具

src/agent.ts

@@ -49,6 +49,7 @@ GUIDELINES:
 2. ROBUSTNESS: Use standard Linux/Unix tools found in minimal images (Alpine/Debian).
 3. TOOLS: Use 'execute_shell_command' for actions, 'write_file' for code generation.
 4. CLARITY: Output concise logs. You are a worker unit, not a chat bot.
+5. OPTIMIZATION: When asked to generate creative content (images, stories, complex code), use 'optimize_prompt' first to ensure the best possible output quality.
 `
       }
     ];

src/tools/index.ts

@@ -6,6 +6,7 @@ import { NotifyTool } from './notify.js';
 import { BrowserTool } from './browser.js';
 import { ScreenshotTool } from './screenshot.js';
 import { ImageTool } from './image.js';
+import { PromptOptimizerTool } from './prompt-optimizer.js';
 
 // Central Registry of all available tools
 export const toolRegistry: ToolModule[] = [
@@ -13,6 +14,7 @@ export const toolRegistry: ToolModule[] = [
   ReadFileTool,
   WriteFileTool,
   DateTimeTool,
+  PromptOptimizerTool,
   EmailTool,
   SearchTool,
   NotifyTool,

src/tools/prompt-optimizer.ts

@@ -0,0 +1,69 @@
+import OpenAI from 'openai';
+import { ToolModule } from './interface.js';
+
+export const PromptOptimizerTool: ToolModule = {
+  name: "Prompt Optimizer",
+  definition: {
+    type: "function",
+    function: {
+      name: "optimize_prompt",
+      description: "Optimize a user's raw task description or prompt to be more professional, structured, and effective. STRONGLY RECOMMENDED for creative tasks (like image generation) or complex scripts to ensure high-quality results.",
+      parameters: {
+        type: "object",
+        properties: {
+          raw_prompt: { 
+            type: "string", 
+            description: "The original, raw prompt or task description provided by the user." 
+          },
+          context: { 
+            type: "string", 
+            description: "Optional context about the goal, audience, or specific requirements (e.g., 'for an image generator', 'for a code reviewer')." 
+          }
+        },
+        required: ["raw_prompt"]
+      }
+    }
+  },
+  handler: async (args: any, config: any) => {
+    if (!config?.apiKey) {
+      return "Error: OpenAI API Key is missing in the configuration. Please run 'autoclaw setup' or check your .env file.";
+    }
+
+    const client = new OpenAI({
+      apiKey: config.apiKey,
+      baseURL: config.baseUrl
+    });
+
+    const contextMsg = args.context ? `Context: ${args.context}` : "Context: General AI Assistant interaction.";
+
+    try {
+      const completion = await client.chat.completions.create({
+        model: config.model || 'gpt-4o',
+        messages: [
+          {
+            role: "system",
+            content: `You are an expert Prompt Engineer. Your goal is to rewrite the user's raw prompt to be clear, precise, and highly effective for LLMs or professional communication.
+            
+RULES:
+1. Preserve the original intent.
+2. Structure the prompt logically (e.g., Role, Context, Task, Constraints, Output Format).
+3. Use professional and concise language.
+4. Return ONLY the optimized prompt. Do not add conversational filler.`
+          },
+          {
+            role: "user",
+            content: `Raw Prompt: "${args.raw_prompt}"
+
+${contextMsg}
+
+Please optimize this prompt.`
+          }
+        ]
+      });
+
+      return completion.choices[0].message?.content || "Error: Failed to generate optimized prompt.";
+    } catch (error: any) {
+      return `Error optimizing prompt: ${error.message}`;
+    }
+  }
+};