tuna-agent 0.1.35 → 0.1.37

package/dist/daemon/extension-handlers.d.ts

@@ -26,7 +26,7 @@ export interface VideoRecord {
     aspectRatio?: string;
 }
 export declare function handleGetHistory(ws: AgentWebSocketClient, code: string, taskId: string): void;
-export declare function handleGenerateIdeas(ws: AgentWebSocketClient, code: string, taskId: string, topic: string, styleName?: string, styleDesc?: string, language?: string): Promise<void>;
+export declare function handleGenerateIdeas(ws: AgentWebSocketClient, code: string, taskId: string, topic: string, styleName?: string, styleDesc?: string, language?: string, count?: number, ideasInstruction?: string): Promise<void>;
 export declare function handleGenerateScript(ws: AgentWebSocketClient, code: string, taskId: string, idea: string, topic: string, style?: string, duration?: number, language?: string, styleName?: string, styleGuidance?: string): Promise<void>;
 export declare function handleRetryVideo(ws: AgentWebSocketClient, code: string, taskId: string, videoId: string): void;
 export declare function handleGenerateScene(ws: AgentWebSocketClient, code: string, taskId: string, sceneIdx: number, prompt: string, aspectRatio: string): Promise<void>;

package/dist/daemon/extension-handlers.js

@@ -64,7 +64,8 @@ function hasContentCreator() {
     }
 }
 // ─── Handler: generate_ideas ──────────────────────────────────────────────────
-export async function handleGenerateIdeas(ws, code, taskId, topic, styleName, styleDesc, language) {
+export async function handleGenerateIdeas(ws, code, taskId, topic, styleName, styleDesc, language, count, ideasInstruction) {
+    console.log(`[generate_ideas] Received: topic="${topic}" count=${count} ideasInstruction=${ideasInstruction ? ideasInstruction.length + ' chars' : '(none)'} styleName=${styleName || '(none)'}`);
     if (!hasContentCreator()) {
         const error = 'content-creator agent not found on this machine';
         console.error(`[generate_ideas] ${error}`);
@@ -86,12 +87,16 @@ export async function handleGenerateIdeas(ws, code, taskId, topic, styleName, st
         systemParts.push(`The video style is "${styleName}"${styleDesc ? `: ${styleDesc}` : ''}.`);
         systemParts.push(`Generate ideas that are specifically suited for this style.`);
     }
+    if (ideasInstruction && ideasInstruction.trim()) {
+        systemParts.push(`\n\nAdditional instructions for idea generation:\n${ideasInstruction.trim()}`);
+    }
     const systemPrompt = systemParts.join(' ');
     const langReq = resolvedLang === 'vi'
         ? `- Titles must be in Vietnamese, natural and engaging`
         : `- Titles must be in English, natural and engaging`;
+    const n = count && count > 0 ? count : 10;
     const prompt = [
-        `Generate exactly 5 viral YouTube Shorts video ideas for the topic: "${topic}".`,
+        `Generate exactly ${n} viral YouTube Shorts video ideas for the topic: "${topic}".`,
         ``,
         `Requirements:`,
         `- Each idea is a catchy, scroll-stopping video title`,
@@ -100,7 +105,7 @@ export async function handleGenerateIdeas(ws, code, taskId, topic, styleName, st
         `- Mix different angles/formats for variety`,
         ...(styleName ? [`- Ideas must fit the "${styleName}" video style${styleDesc ? ` (${styleDesc})` : ''}`] : []),
         ``,
-        `Respond with ONLY a JSON array of 5 strings. No explanation, no markdown, no wrapping.`,
+        `Respond with ONLY a JSON array of ${n} strings. No explanation, no markdown, no wrapping.`,
         `Example format: ["title 1","title 2","title 3","title 4","title 5"]`,
     ].join('\n');
     try {
@@ -127,9 +132,9 @@ export async function handleGenerateIdeas(ws, code, taskId, topic, styleName, st
                 }
                 catch { /* skip non-JSON matches */ }
             }
-            // Prefer the array closest to 5 items
+            // Prefer the array closest to n items
             if (arrays.length > 0) {
-                ideas = arrays.reduce((best, cur) => Math.abs(cur.length - 5) < Math.abs(best.length - 5) ? cur : best);
+                ideas = arrays.reduce((best, cur) => Math.abs(cur.length - n) < Math.abs(best.length - n) ? cur : best);
             }
         }
         catch { /* fall through to fallback */ }

package/dist/daemon/index.js

@@ -367,13 +367,13 @@ ${skillContent.slice(0, 15000)}`;
                 }
                 if (extTask === 'generate_ideas') {
                     (async () => {
-                        await handleGenerateIdeas(ws, extCode, extTaskId, msg.topic, msg.styleName, msg.styleDesc, msg.language);
+                        await handleGenerateIdeas(ws, extCode, extTaskId, msg.topic, msg.styleName, msg.styleDesc, msg.language, msg.count, msg.ideasInstruction);
                     })();
                     break;
                 }
                 if (extTask === 'generate_script') {
                     (async () => {
-                        await handleGenerateScript(ws, extCode, extTaskId, msg.idea, msg.topic, msg.style, msg.duration, msg.language, msg.styleName, msg.styleGuidance);
+                        await handleGenerateScript(ws, extCode, extTaskId, msg.idea, msg.topic, msg.style, msg.duration, msg.language, msg.styleName, (msg.scriptInstruction || msg.styleGuidance));
                     })();
                     break;
                 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "tuna-agent",
-  "version": "0.1.35",
+  "version": "0.1.37",
   "description": "Tuna Agent - Run AI coding tasks on your machine",
   "bin": {
     "tuna-agent": "dist/cli/index.js"