pokt-cli 1.0.3 → 1.0.4

package/dist/chat/loop.js

@@ -13,12 +13,17 @@ CORE CAPABILITIES:
 2.  **Autonomous Coding**: You can create new files, rewrite existing ones, and run terminal commands.
 3.  **Problem Solving**: You analyze errors and propose/apply fixes.
 
+CRITICAL - FILE CREATION/EDITS (this API does NOT support tool calls):
+- Do NOT reply with only "We will call read_file", "We will call write_file" or similar. Those tools will NOT run. The user will get no file.
+- You MUST output the complete file content in a markdown code block so the CLI can create/edit the file. Format: mention the filename (e.g. hello.py or **hello.py**) then a newline then \`\`\`python then newline then the full file content then \`\`\`.
+- For edits: first "read" the file by inferring its content from the user request and project context, then output the full updated file in a \`\`\`python (or correct language) block with the filename mentioned just above the block.
+- Never end your response with only an intention to call a tool. Always include the actual code in a block.
+
 GUIDELINES:
 - You will receive the user request first, then the current project structure. Use the project structure to understand the context before creating or editing anything.
 - When asked to fix something, first **read** the relevant files to understand the context.
-- When creating a project, start by planning the structure, then use \`write_file\` to create files.
-- **DO NOT repeat the generated code in your chat response** if you have already used the \`write_file\` tool. Simply state that the file has been updated or created.
-- You have full access to the current terminal. You can run \`npm install\`, \`tsc\`, or any other command.
+- When creating a project, start by planning the structure, then use \`write_file\` to create each file.
+- You have full access to the current terminal. You can run \`run_command\` for \`npm install\`, \`tsc\`, or any other command.
 - Be extremely concise in your explanations.
 - The current working directory is: ${process.cwd()}
 `;
@@ -58,6 +63,8 @@ export async function startChatLoop(modelConfig) {
     ];
     while (true) {
         console.log('');
+        const cwd = process.cwd();
+        console.log(ui.dim(`Diretório atual: ${cwd}`));
         console.log(ui.shortcutsLine('shift+tab to accept edits', '? for shortcuts'));
         const response = await prompts({
             type: 'text',
@@ -87,6 +94,83 @@ export async function startChatLoop(modelConfig) {
 }
 const MAX_429_RETRIES = 3;
 const BASE_429_DELAY_MS = 5000;
+/** Extensões que consideramos como arquivos de código para aplicar fallback */
+const CODE_EXT = /\.(py|js|ts|tsx|jsx|html|css|json|md|txt|java|go|rs|c|cpp|rb|php)$/i;
+/**
+ * Quando a API não retorna tool_calls, alguns backends só devolvem texto.
+ * Extrai blocos de código da resposta (```lang\n...\n```) e, se encontrar
+ * um nome de arquivo mencionado antes do bloco, aplica write_file.
+ */
+/** Blocos de comando "como rodar" (bash/sh de 1–2 linhas) não viram arquivo para não poluir. */
+function isRunCommandOnly(lang, code) {
+    const shellLike = /^(bash|sh|shell|zsh)$/i.test(lang);
+    const lines = code.split('\n').filter((l) => l.trim().length > 0);
+    return shellLike && lines.length <= 2;
+}
+/** Remove markdown/formatting do nome de arquivo (ex: **hello.py** → hello.py). */
+function cleanFilename(candidate) {
+    return candidate.replace(/^[\s*`'"]+/g, '').replace(/[\s*`'")\]\s]+$/g, '').trim();
+}
+/**
+ * Aplica blocos de código da resposta e retorna conteúdo para exibição (sem repetir o código).
+ * - Detecta nome de arquivo na mensagem (ex: "updated **hello.py**") para editar o existente.
+ * - Blocos já aplicados são substituídos por "[Código aplicado ao arquivo: path]" na mensagem exibida.
+ */
+/** Garante que o conteúdo da mensagem seja string (algumas APIs devolvem array). */
+function messageContentToString(content) {
+    if (typeof content === 'string')
+        return content;
+    if (Array.isArray(content)) {
+        return content
+            .map((part) => (typeof part === 'object' && part != null && 'text' in part ? part.text : String(part)))
+            .join('');
+    }
+    return content != null ? String(content) : '';
+}
+async function applyCodeBlocksFromContent(content) {
+    const codeBlockRe = /```(\w*)\n([\s\S]*?)```/g;
+    const appliedBlocks = [];
+    let applied = false;
+    let m;
+    const matches = [];
+    while ((m = codeBlockRe.exec(content)) !== null) {
+        matches.push({
+            fullMatch: m[0],
+            index: m.index,
+            lang: m[1] || '',
+            code: m[2].replace(/\r\n/g, '\n').trimEnd(),
+        });
+    }
+    for (const { fullMatch, index, lang, code } of matches) {
+        if (isRunCommandOnly(lang, code))
+            continue;
+        const beforeBlock = content.substring(0, index);
+        // Nome de arquivo: aceita "**hello.py**", "hello.py" antes de espaço/newline/backtick, etc.
+        const fileMatch = beforeBlock.match(/(\S+\.(?:py|js|ts|tsx|jsx|html|css|json|md|txt|java|go|rs|c|cpp|rb|php))(?=\s|$|[:.)\]*`"])/gi);
+        const rawCandidate = fileMatch ? fileMatch[fileMatch.length - 1].trim() : null;
+        const candidate = rawCandidate ? cleanFilename(rawCandidate) : null;
+        const path = candidate && CODE_EXT.test(candidate) ? candidate : (lang === 'python' ? 'generated.py' : lang ? `generated.${lang}` : null);
+        if (path && code) {
+            try {
+                console.log(ui.warn(`\n[Fallback] Aplicando código da resposta ao arquivo: ${path}`));
+                await executeTool('write_file', JSON.stringify({ path, content: code }));
+                applied = true;
+                appliedBlocks.push({ start: index, end: index + fullMatch.length, path });
+            }
+            catch {
+                // ignora falha em um bloco
+            }
+        }
+    }
+    // Monta mensagem para exibição: blocos aplicados viram uma linha curta (evita repetir o código)
+    let displayContent = content;
+    for (let i = appliedBlocks.length - 1; i >= 0; i--) {
+        const { start, end, path } = appliedBlocks[i];
+        const placeholder = `\n${ui.dim('[Código aplicado ao arquivo: ' + path + ']')}\n`;
+        displayContent = displayContent.substring(0, start) + placeholder + displayContent.substring(end);
+    }
+    return { applied, displayContent };
+}
 async function createCompletionWithRetry(client, modelId, messages, toolsList) {
     let lastError;
     for (let attempt = 0; attempt <= MAX_429_RETRIES; attempt++) {
@@ -117,10 +201,13 @@ async function processLLMResponse(client, modelId, messages, toolsList) {
         let completion = await createCompletionWithRetry(client, modelId, messages, toolsList);
         let message = completion.choices[0].message;
         spinner.stop();
+        let writeFileExecutedThisTurn = false;
         while (message.tool_calls && message.tool_calls.length > 0) {
             messages.push(message);
             for (const toolCall of message.tool_calls) {
                 const name = toolCall.function.name;
+                if (name === 'write_file')
+                    writeFileExecutedThisTurn = true;
                 const args = toolCall.function.arguments ?? '{}';
                 console.log(ui.warn(`\n[Executing Tool: ${name}]`));
                 console.log(ui.dim(`Arguments: ${args}`));
@@ -141,10 +228,59 @@ async function processLLMResponse(client, modelId, messages, toolsList) {
             message = completion.choices[0].message;
             spinner.stop();
         }
-        if (message.content) {
-            console.log('\n' + ui.labelPokt());
-            console.log(message.content);
-            messages.push({ role: 'assistant', content: message.content });
+        const rawContent = message.content;
+        let contentStr = messageContentToString(rawContent);
+        let finalContent = rawContent ?? contentStr;
+        // Quando a API não executa tools, tentar aplicar blocos de código da resposta
+        if (!writeFileExecutedThisTurn) {
+            let result = await applyCodeBlocksFromContent(contentStr);
+            // Se a IA só disse "We will call read_file/write_file" e não há código, pedir o código em um follow-up
+            const looksLikeToolIntentOnly = /(We will call|We need to call|Let's call|I will call)\s+(read_file|write_file|run_command)/i.test(contentStr)
+                || (/call\s+(read_file|write_file)/i.test(contentStr) && contentStr.length < 400);
+            if (!result.applied && looksLikeToolIntentOnly) {
+                messages.push({ role: 'assistant', content: rawContent ?? contentStr });
+                const followUpSystem = `This API does not support tool calls. You must NOT reply with "We will call X". Output the complete file content in a markdown code block so the user's CLI can create/edit the file. Format: mention the filename (e.g. hello.py) then newline then \`\`\`python then newline then the FULL file content then \`\`\`. Do that now for the user's last request.`;
+                messages.push({ role: 'system', content: followUpSystem });
+                spinner.start('Getting code...');
+                const followUp = await createCompletionWithRetry(client, modelId, messages, toolsList);
+                spinner.stop();
+                const followUpMsg = followUp.choices[0].message;
+                const followUpStr = messageContentToString(followUpMsg.content);
+                if (followUpStr.trim() !== '') {
+                    result = await applyCodeBlocksFromContent(followUpStr);
+                    contentStr = followUpStr;
+                    finalContent = followUpMsg.content ?? followUpStr;
+                    messages.push({ role: 'assistant', content: finalContent });
+                }
+                else {
+                    messages.push({ role: 'assistant', content: '' });
+                }
+            }
+            if (contentStr.trim() !== '') {
+                let contentToShow = result.applied ? result.displayContent : contentStr;
+                console.log('\n' + ui.labelPokt());
+                console.log(contentToShow);
+                if (!messages.some((m) => m.role === 'assistant' && m.content === finalContent)) {
+                    messages.push({ role: 'assistant', content: finalContent });
+                }
+            }
+            else {
+                console.log('\n' + ui.labelPokt());
+                console.log(ui.dim('(A IA não retornou código utilizável. Tente reformular o pedido.)'));
+                messages.push({ role: 'assistant', content: '' });
+            }
+        }
+        else {
+            if (contentStr.trim() !== '') {
+                console.log('\n' + ui.labelPokt());
+                console.log(contentStr);
+                messages.push({ role: 'assistant', content: finalContent });
+            }
+            else {
+                console.log('\n' + ui.labelPokt());
+                console.log(ui.dim('(Sem resposta de texto.)'));
+                messages.push({ role: 'assistant', content: '' });
+            }
         }
     }
     catch (error) {

package/dist/commands/chat.js

@@ -34,7 +34,7 @@ export const chatCommand = {
             console.log('');
         }
         console.log(ui.dim('Type "exit" or /quit to end the session.'));
-        console.log(ui.statusBar({ model: `/model ${activeModel.provider} (${activeModel.id})` }));
+        console.log(ui.statusBar({ cwd: process.cwd(), model: `/model ${activeModel.provider} (${activeModel.id})` }));
         console.log('');
         await startChatLoop(activeModel);
     }

package/dist/ui.js

@@ -2,7 +2,7 @@ import chalk from 'chalk';
 import fs from 'fs';
 import path from 'path';
 import { execSync } from 'child_process';
-const VERSION = '1.0.1';
+const VERSION = '1.0.4';
 /** Logo em estilo chevron com gradiente (azul → rosa → roxo) */
 function logo() {
     const c = (s, color) => color(s);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pokt-cli",
-  "version": "1.0.3",
+  "version": "1.0.4",
   "description": "Vibe Coding AI CLI for OpenRouter and Ollama",
   "main": "./dist/bin/pokt.js",
   "type": "module",