codeep 1.0.24 → 1.0.26

package/dist/utils/agent.js

@@ -205,7 +205,7 @@ async function agentChat(messages, systemPrompt, onChunk, abortSignal, dynamicTi
                 tools: getOpenAITools(),
                 tool_choice: 'auto',
                 temperature: config.get('temperature'),
-                max_tokens: config.get('maxTokens'),
+                max_tokens: Math.max(config.get('maxTokens'), 16384), // Ensure enough tokens for large file generation
             };
         }
         else {
@@ -216,7 +216,7 @@ async function agentChat(messages, systemPrompt, onChunk, abortSignal, dynamicTi
                 messages: messages,
                 tools: getAnthropicTools(),
                 temperature: config.get('temperature'),
-                max_tokens: config.get('maxTokens'),
+                max_tokens: Math.max(config.get('maxTokens'), 16384), // Ensure enough tokens for large file generation
             };
         }
         const response = await fetch(endpoint, {
@@ -626,13 +626,33 @@ export async function runAgent(prompt, projectContext, options = {}) {
                     toolCalls = textToolCalls;
                 }
             }
-            // If no tool calls, agent is done - simple and clean like Claude Code/Aider
+            // If no tool calls, check if model wants to continue or is really done
             if (toolCalls.length === 0) {
-                console.error(`[DEBUG] No tool calls at iteration ${iteration}, agent finished`);
+                console.error(`[DEBUG] No tool calls at iteration ${iteration}`);
                 // Remove <think>...</think> tags from response (some models include thinking)
                 finalResponse = content.replace(/<think>[\s\S]*?<\/think>/gi, '').trim();
-                // Trust the model - if it says it's done, it's done
-                // No forcing, no blocking, no artificial requirements
+                // Check if model indicates it wants to continue (incomplete response)
+                const continueIndicators = [
+                    'let me', 'i will', 'i\'ll', 'now i', 'next i',
+                    'creating', 'writing', 'generating',
+                    'let\'s', 'going to', 'need to create', 'need to write'
+                ];
+                const lowerResponse = finalResponse.toLowerCase();
+                const wantsToContinue = continueIndicators.some(indicator => lowerResponse.includes(indicator));
+                // Also check if there were tool call parsing failures in this iteration
+                // by looking for incomplete actions (e.g., write_file without content)
+                const hasIncompleteWork = iteration < 10 && wantsToContinue && finalResponse.length < 500;
+                if (hasIncompleteWork) {
+                    console.error(`[DEBUG] Model wants to continue, prompting for next action`);
+                    messages.push({ role: 'assistant', content });
+                    messages.push({
+                        role: 'user',
+                        content: 'Continue. Execute the tool calls now.'
+                    });
+                    continue;
+                }
+                // Model is done
+                console.error(`[DEBUG] Agent finished at iteration ${iteration}`);
                 break;
             }
             // Add assistant response to history

package/dist/utils/tools.js

@@ -204,20 +204,132 @@ function normalizeToolName(name) {
 export function parseOpenAIToolCalls(toolCalls) {
     if (!toolCalls || !Array.isArray(toolCalls))
         return [];
-    return toolCalls.map(tc => {
+    const parsed = [];
+    for (const tc of toolCalls) {
+        const toolName = normalizeToolName(tc.function?.name || '');
+        if (!toolName)
+            continue;
         let parameters = {};
+        const rawArgs = tc.function?.arguments || '{}';
         try {
-            parameters = JSON.parse(tc.function?.arguments || '{}');
+            parameters = JSON.parse(rawArgs);
         }
-        catch {
-            parameters = {};
+        catch (e) {
+            // JSON parsing failed - likely truncated response
+            // Try to extract what we can from partial JSON
+            console.error(`[DEBUG] Failed to parse tool arguments for ${toolName}, attempting partial extraction...`);
+            const partialParams = extractPartialToolParams(toolName, rawArgs);
+            if (partialParams) {
+                parameters = partialParams;
+                console.error(`[DEBUG] Successfully extracted partial params for ${toolName}`);
+            }
+            else {
+                console.error(`[DEBUG] Could not extract params, skipping ${toolName}`);
+                continue;
+            }
         }
-        return {
-            tool: normalizeToolName(tc.function?.name || ''),
+        // Validate required parameters for specific tools
+        if (toolName === 'write_file' && (!parameters.path || parameters.content === undefined)) {
+            console.error(`[DEBUG] Skipping write_file with missing path or content`);
+            continue;
+        }
+        if (toolName === 'read_file' && !parameters.path) {
+            console.error(`[DEBUG] Skipping read_file with missing path`);
+            continue;
+        }
+        if (toolName === 'edit_file' && (!parameters.path || parameters.old_text === undefined || parameters.new_text === undefined)) {
+            console.error(`[DEBUG] Skipping edit_file with missing parameters`);
+            continue;
+        }
+        parsed.push({
+            tool: toolName,
             parameters,
             id: tc.id,
-        };
-    }).filter(tc => tc.tool);
+        });
+    }
+    return parsed;
+}
+/**
+ * Extract parameters from truncated/partial JSON for tool calls
+ * This is a fallback when JSON.parse fails due to API truncation
+ */
+function extractPartialToolParams(toolName, rawArgs) {
+    try {
+        // For write_file, try to extract path and content
+        if (toolName === 'write_file') {
+            const pathMatch = rawArgs.match(/"path"\s*:\s*"([^"]+)"/);
+            const contentMatch = rawArgs.match(/"content"\s*:\s*"([\s\S]*?)(?:"|$)/);
+            if (pathMatch && contentMatch) {
+                // Unescape the content
+                let content = contentMatch[1];
+                content = content
+                    .replace(/\\n/g, '\n')
+                    .replace(/\\t/g, '\t')
+                    .replace(/\\r/g, '\r')
+                    .replace(/\\"/g, '"')
+                    .replace(/\\\\/g, '\\');
+                // If content appears truncated (doesn't end properly), add a comment
+                if (!content.endsWith('\n') && !content.endsWith('}') && !content.endsWith(';') && !content.endsWith('>')) {
+                    content += '\n<!-- Content may be truncated -->\n';
+                }
+                return { path: pathMatch[1], content };
+            }
+        }
+        // For read_file, just need path
+        if (toolName === 'read_file') {
+            const pathMatch = rawArgs.match(/"path"\s*:\s*"([^"]+)"/);
+            if (pathMatch) {
+                return { path: pathMatch[1] };
+            }
+        }
+        // For list_files, just need path
+        if (toolName === 'list_files') {
+            const pathMatch = rawArgs.match(/"path"\s*:\s*"([^"]+)"/);
+            if (pathMatch) {
+                return { path: pathMatch[1] };
+            }
+        }
+        // For create_directory, just need path
+        if (toolName === 'create_directory') {
+            const pathMatch = rawArgs.match(/"path"\s*:\s*"([^"]+)"/);
+            if (pathMatch) {
+                return { path: pathMatch[1] };
+            }
+        }
+        // For edit_file, need path, old_text, new_text
+        if (toolName === 'edit_file') {
+            const pathMatch = rawArgs.match(/"path"\s*:\s*"([^"]+)"/);
+            const oldTextMatch = rawArgs.match(/"old_text"\s*:\s*"([\s\S]*?)(?:"|$)/);
+            const newTextMatch = rawArgs.match(/"new_text"\s*:\s*"([\s\S]*?)(?:"|$)/);
+            if (pathMatch && oldTextMatch && newTextMatch) {
+                return {
+                    path: pathMatch[1],
+                    old_text: oldTextMatch[1].replace(/\\n/g, '\n').replace(/\\"/g, '"'),
+                    new_text: newTextMatch[1].replace(/\\n/g, '\n').replace(/\\"/g, '"'),
+                };
+            }
+        }
+        // For execute_command
+        if (toolName === 'execute_command') {
+            const commandMatch = rawArgs.match(/"command"\s*:\s*"([^"]+)"/);
+            if (commandMatch) {
+                const argsMatch = rawArgs.match(/"args"\s*:\s*\[([\s\S]*?)\]/);
+                let args = [];
+                if (argsMatch) {
+                    const argStrings = argsMatch[1].match(/"([^"]+)"/g);
+                    if (argStrings) {
+                        args = argStrings.map(s => s.replace(/"/g, ''));
+                    }
+                }
+                return { command: commandMatch[1], args };
+            }
+        }
+        return null;
+    }
+    catch (e) {
+        console.error(`[DEBUG] Error in extractPartialToolParams:`, e);
+        return null;
+    }
 }
 /**
  * Parse tool calls from Anthropic response

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "codeep",
-  "version": "1.0.24",
+  "version": "1.0.26",
   "description": "AI-powered coding assistant built for the terminal. Multiple LLM providers, project-aware context, and a seamless development workflow.",
   "type": "module",
   "main": "dist/index.js",