zames_pro 2.0.0 → 2.1.1

package/dist/agent-loop.js

@@ -1,11 +1,29 @@
 import { buildSystemPrompt } from './system-prompt.js';
 import { getGitContext, formatGitContext } from './gitTools.js';
 import { parseXmlToolCalls } from './xml-toolcall.js';
-export async function runAgentLoop({ browser, tools, task, workdir, maxIterations = 40, freshChat = false, sendSystemPrompt = false, transcript = null, onThinking = () => { }, onAssistantThought = () => { }, onToolCall = () => { }, onToolResult = () => { }, onAssistantMessage = () => { }, debugLog = false, }) {
+export async function runAgentLoop({ browser, tools, task, workdir, maxIterations = 40, freshChat = false, sendSystemPrompt = false, transcript = null, onThinking = () => { }, onAssistantThought = () => { }, onToolCall = () => { }, onToolResult = () => { }, onAssistantMessage = () => { }, onChatReady = () => { }, debugLog = false, }) {
     if (freshChat) {
         await browser.newChat();
         transcript?.log('new_chat');
     }
+    // Сообщаем вызывающему актуальный chat id. После newChat() URL ещё без id
+    // (он появляется только после первой отправки), поэтому зовём колбэк и
+    // здесь, и после первой реальной отправки ниже.
+    let lastReportedChatId = null;
+    const reportChat = async () => {
+        let id = null;
+        try {
+            id = await browser.getCurrentChatId();
+        }
+        catch {
+            id = null;
+        }
+        if (id && id !== lastReportedChatId) {
+            lastReportedChatId = id;
+            onChatReady(id);
+        }
+    };
+    await reportChat();
     if (sendSystemPrompt) {
         let gitText = null;
         try {
@@ -26,6 +44,7 @@ export async function runAgentLoop({ browser, tools, task, workdir, maxIteration
         });
         onThinking();
         await browser.ask(systemPrompt, { timeout: 60_000 });
+        await reportChat();
     }
     let message = task;
     transcript?.log('task', { task });
@@ -37,6 +56,7 @@ export async function runAgentLoop({ browser, tools, task, workdir, maxIteration
     for (let i = 0; i < maxIterations; i++) {
         onThinking();
         const rawResponse = await browser.ask(message);
+        await reportChat();
         transcript?.log('assistant_raw', { response: rawResponse });
         const parsed = parseToolCall(rawResponse);
         if (parsed) {
@@ -60,9 +80,16 @@ export async function runAgentLoop({ browser, tools, task, workdir, maxIteration
             const looksLikeToolCall = /("tool"\s*:|\btool_calls?\b|\binvoke\b|\bparameter\b|DSML|function_call)/i.test(rawResponse) || /\{\s*"?(tool|name)"?\s*:/.test(rawResponse);
             if (looksLikeToolCall && malformedRetries < MAX_MALFORMED_RETRIES) {
                 malformedRetries++;
-                transcript?.log("malformed_toolcall", { attempt: malformedRetries, response: rawResponse });
+                transcript?.log('malformed_toolcall', {
+                    attempt: malformedRetries,
+                    response: rawResponse,
+                });
                 if (debugLog) {
-                    console.error("внимание: ответ похож на tool-call, но не распознан (попытка " + malformedRetries + "/" + MAX_MALFORMED_RETRIES + ")");
+                    console.error('внимание: ответ похож на tool-call, но не распознан (попытка ' +
+                        malformedRetries +
+                        '/' +
+                        MAX_MALFORMED_RETRIES +
+                        ')');
                 }
                 message =
                     'Твой предыдущий ответ не распознан как вызов инструмента. ' +
@@ -649,4 +676,3 @@ function extractPreToolText(text) {
         return '';
     return text.slice(0, earliest).trim();
 }
-//# sourceMappingURL=agent-loop.js.map

package/dist/browser.js

@@ -1,4 +1,5 @@
-import { chromium } from 'playwright';
+import { chromium, } from 'playwright';
+import { extractAnswer, dumpNetBody } from './net-capture.js';
 import path from 'path';
 import os from 'os';
 import fs from 'fs/promises';
@@ -12,10 +13,25 @@ const INPUT_SELECTORS = [
     '[role="textbox"]',
 ];
 const ANSWER_SELECTORS = [
+    'div.ds-assistant-message-main-content',
+    'div[class*="ds-assistant-message-main-content"]',
     'div[class*="ds-markdown"]',
     'div[class*="markdown"]',
 ];
+const SEND_SELECTORS = [
+    'div[role="button"].ds-button--primary.ds-button--circle',
+    'div[role="button"].ds-button--primary.ds-button--filled',
+    'button[type="submit"]',
+    'button[aria-label*="send" i]',
+    'button[aria-label*="отправ" i]',
+];
+// ВАЖНО: сюда НЕЛЬЗЯ добавлять общий 'div[role="button"][class*="ds-button--primary"]'
+// — под него попадает кнопка отправки, которая видна всегда, и тогда
+// _isGenerating() вечно возвращает true, из-за чего ответ никогда не
+// считается готовым.
 const STOP_SELECTORS = [
+    'div[role="button"][aria-label*="stop" i]',
+    'div[role="button"][aria-label*="останов" i]',
     'button:has-text("Stop")',
     'button:has-text("Остановить")',
     'button[aria-label*="Stop" i]',
@@ -71,6 +87,12 @@ export class DeepSeekBrowser {
     _abort;
     context;
     page;
+    _netCapture;
+    _netCaptureAt;
+    _netChatId;
+    _netSniff;
+    _netSniffLimit;
+    _netHookInstalled;
     constructor({ headless = false, debug = false, channel = 'chrome', answerTimeoutMs = 180000, askRetries = 3, stabilityChecks = 3, stabilityDelayMs = 1000, minSendIntervalMs = 15000, } = {}) {
         this.headless = headless;
         this.debug = debug;
@@ -79,11 +101,15 @@ export class DeepSeekBrowser {
         this.askRetries = askRetries;
         this.stabilityChecks = stabilityChecks;
         this.stabilityDelayMs = stabilityDelayMs;
-        // Минимальный интервал между отправками в чат: DeepSeek ограничивает
-        // частоту («Messages too frequent. Try again later.»).
         this.minSendIntervalMs = minSendIntervalMs;
         this._lastSentAt = 0;
         this._abort = false;
+        this._netCapture = '';
+        this._netCaptureAt = 0;
+        this._netChatId = null;
+        this._netSniff = [];
+        this._netSniffLimit = 5;
+        this._netHookInstalled = false;
     }
     async launch() {
         await killStaleChrome();
@@ -126,8 +152,51 @@ export class DeepSeekBrowser {
             }
         }
         this.page = this.context.pages()[0] || (await this.context.newPage());
+        this._installNetHook();
         await this.page.goto(CHAT_URL, { waitUntil: 'domcontentloaded' });
     }
+    _installNetHook() {
+        if (this._netHookInstalled)
+            return;
+        const pg = this['page'];
+        if (!pg)
+            return;
+        this._netHookInstalled = true;
+        pg.on('response', (resp) => {
+            void this._onResponse(resp).catch(() => { });
+        });
+    }
+    async _onResponse(resp) {
+        try {
+            const url = resp.url();
+            if (!/deepseek\.com/i.test(url))
+                return;
+            if (/\.(js|css|png|jpg|jpeg|svg|woff2?|ico|map)(\?|$)/i.test(url))
+                return;
+            const ct = (resp.headers()['content-type'] || '').toLowerCase();
+            const interesting = ct.includes('event-stream') ||
+                ct.includes('json') ||
+                ct.includes('text/plain');
+            if (!interesting)
+                return;
+            const body = await resp.text().catch(() => '');
+            if (!body)
+                return;
+            this._netSniff.push({ url, contentType: ct, body });
+            if (this._netSniff.length > this._netSniffLimit)
+                this._netSniff.shift();
+            void dumpNetBody(url, body);
+            const extracted = extractAnswer(body);
+            if (extracted) {
+                this._netCapture = extracted;
+                const cid = url.match(/[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}/);
+                if (cid)
+                    this._netChatId = cid[0];
+                this._netCaptureAt = Date.now();
+            }
+        }
+        catch { }
+    }
     async restart() {
         try {
             if (this.context)
@@ -193,6 +262,12 @@ export class DeepSeekBrowser {
         return null;
     }
     async _readLastAnswerText() {
+        // Если удалось перехватить сырой текст ответа по сети (без рендер-
+        // искажений DeepSeek) и он относится к текущему ответу — отдаём его.
+        // Это защищает $, экранированные переводы строк и т.п. в аргументах.
+        if (this._netCapture && this._netCaptureAt >= this._lastSentAt) {
+            return this._netCapture;
+        }
         return await this.page.evaluate((sels) => {
             let el = null;
             for (const s of sels) {
@@ -202,74 +277,8 @@ export class DeepSeekBrowser {
             }
             if (!el)
                 return '';
-            const clone = el.cloneNode(true);
-            Array.from(clone.querySelectorAll("pre")).forEach((pre) => {
-                const codeEl = pre.querySelector('code');
-                const source = codeEl || pre;
-                const text = (source.textContent || '').replace(/\n$/, '');
-                const langMatch = (source.className || '').match(/language-([\w-]+)/);
-                const lang = langMatch ? langMatch[1] : '';
-                const replacement = document.createTextNode('```' + lang + '\n' + text + '\n```');
-                if (pre.parentNode)
-                    pre.parentNode.replaceChild(replacement, pre);
-            });
-            Array.from(clone.querySelectorAll("code")).forEach((c) => {
-                const replacement = document.createTextNode('`' + (c.textContent || '') + '`');
-                if (c.parentNode)
-                    c.parentNode.replaceChild(replacement, c);
-            });
-            // textContent склеивает блоки без переводов строк, из-за чего
-            // Markdown-рендер получает одну длинную строку. Обходим DOM сами и
-            // расставляем переводы строк / маркеры Markdown по блочным элементам.
+            const out = el.innerText || el.textContent || '';
             const NL = String.fromCharCode(10);
-            const BULLET = String.fromCharCode(45) + ' '; // '- '
-            function domToMarkdown(node) {
-                if (node.nodeType === 3)
-                    return node.textContent || '';
-                if (node.nodeType !== 1)
-                    return '';
-                const tag = node.tagName.toLowerCase();
-                if (tag === 'br')
-                    return NL;
-                const inner = Array.from(node.childNodes)
-                    .map(domToMarkdown)
-                    .join('');
-                const STAR = String.fromCharCode(42); // '*'
-                if (tag === 'strong' || tag === 'b')
-                    return STAR + STAR + inner + STAR + STAR;
-                if (tag === 'em' || tag === 'i')
-                    return STAR + inner + STAR;
-                if (tag === 'del' || tag === 's')
-                    return '~~' + inner + '~~';
-                if (tag === 'a') {
-                    const href = node.getAttribute('href') || '';
-                    return href ? '[' + inner + '](' + href + ')' : inner;
-                }
-                if (/^h[1-6]$/.test(tag)) {
-                    const level = Number(tag[1]);
-                    const hashes = '#'.repeat(level);
-                    return NL + NL + hashes + ' ' + inner.trim() + NL + NL;
-                }
-                if (tag === 'li') {
-                    const text = inner.trim().replace(new RegExp(NL + '+', 'g'), ' ');
-                    return BULLET + text + NL;
-                }
-                if (tag === 'ul' || tag === 'ol' || tag === 'blockquote') {
-                    return NL + inner + NL;
-                }
-                if (tag === 'p' ||
-                    tag === 'div' ||
-                    tag === 'section' ||
-                    tag === 'article' ||
-                    tag === 'tr' ||
-                    tag === 'table') {
-                    const text = inner.trim();
-                    return text ? NL + NL + text : '';
-                }
-                return inner;
-            }
-            const out = domToMarkdown(clone);
-            // Схлопываем тройные+ переводы строк до двойных (разделитель блоков).
             return out.replace(new RegExp(NL + '{3,}', 'g'), NL + NL).trim();
         }, ANSWER_SELECTORS);
     }
@@ -278,7 +287,6 @@ export class DeepSeekBrowser {
         const t = (raw || '').trim();
         if (!t)
             return '';
-        // Отсекаем служебные статусы интерфейса (Reading..., Думаю...).
         if (STATUS_RE.test(t))
             return '';
         return raw;
@@ -287,8 +295,6 @@ export class DeepSeekBrowser {
         const stop = await this._findVisible(STOP_SELECTORS, 300);
         return !!stop;
     }
-    // Прервать текущую генерацию: нажать Stop в интерфейсе.
-    // Используется при нажатии Esc пользователем.
     async stopGeneration() {
         this._abort = true;
         const btn = await this._findVisible(STOP_SELECTORS, 500);
@@ -327,14 +333,6 @@ export class DeepSeekBrowser {
         }
         throw new Error(`ask() провалился после ${this.askRetries} попыток: ${lastErr?.message}`);
     }
-    // Вставка текста в поле ввода.
-    //
-    // fill()/insertText() ломаются на многострочном тексте в contenteditable-
-    // редакторах (ProseMirror/Lexical/...): символ перевода строки там
-    // трактуется как Enter, и в поле остаётся только первая строка. Поэтому
-    // для contenteditable и [role=textbox] эмулируем paste-событие с полным
-    // текстом — то же, что делает Shift+Insert. Для нативных textarea/input
-    // перевод строки работает и так.
     async _setInputText(input, text) {
         const tag = await input.evaluate((el) => el.tagName.toLowerCase());
         const isNative = tag === 'textarea' || tag === 'input';
@@ -346,7 +344,6 @@ export class DeepSeekBrowser {
             catch { }
         }
         await input.click();
-        // Выделяем всё содержимое, чтобы вставка заменила его целиком.
         await this.page.keyboard.press('Control+A');
         await this.page.keyboard.press('Delete');
         const ok = await input.evaluate((el, value) => {
@@ -359,32 +356,30 @@ export class DeepSeekBrowser {
                 cancelable: true,
             });
             el.dispatchEvent(ev);
-            // Если обработчик не отменил вставку и поле осталось пустым —
-            // пробуем через insertText вручную (fallback ниже).
             return true;
         }, text);
         if (!ok) {
             await this.page.keyboard.insertText(text);
         }
-        // Проверяем, что текст реально попал в поле. Если редактор проигнорировал
-        // paste — падаем на insertText.
         const got = await input.evaluate((el) => {
-            if (el.tagName.toLowerCase() === 'textarea' || el.tagName.toLowerCase() === 'input') {
+            if (el.tagName.toLowerCase() === 'textarea' ||
+                el.tagName.toLowerCase() === 'input') {
                 return el.value;
             }
             return el.innerText || el.textContent || '';
         });
-        const norm = (s) => (s || '').replace(/\r\n/g, '\n').replace(/\u00a0/g, ' ').trim();
+        const norm = (s) => (s || '')
+            .replace(/\r\n/g, '\n')
+            .replace(/\u00a0/g, ' ')
+            .trim();
         if (!norm(got)) {
             await input.click();
             await this.page.keyboard.insertText(text);
         }
     }
-    // Ждём, пока с прошлой отправки пройдёт minSendIntervalMs. Защита от
-    // «Messages too frequent. Try again later.» на [chat.deepseek.com](https://chat.deepseek.com/).
     async _waitForSendSlot() {
         if (!this._lastSentAt)
-            return; // первая отправка — пауза не нужна
+            return;
         const gap = this.minSendIntervalMs - (Date.now() - this._lastSentAt);
         if (gap <= 0)
             return;
@@ -398,25 +393,42 @@ export class DeepSeekBrowser {
         }
         const beforeText = await this._readLastAnswerTextClean().catch(() => '');
         await this._waitForSendSlot();
+        this._netCapture = '';
+        this._netCaptureAt = 0;
         await this._setInputText(input, prompt);
         await this.page.waitForTimeout(200);
-        const sendBtn = this.page
-            .locator('button')
-            .filter({ hasText: /send|отправить/i })
-            .first();
-        try {
-            await sendBtn.click({ timeout: 1500 });
+        let sent = false;
+        for (const sel of SEND_SELECTORS) {
+            const btn = this.page.locator(sel).last();
+            try {
+                if ((await btn.count()) === 0)
+                    continue;
+                if (!(await btn.isVisible().catch(() => false)))
+                    continue;
+                await btn.click({ timeout: 1500 });
+                sent = true;
+                break;
+            }
+            catch { }
         }
-        catch {
+        if (!sent) {
             await this.page.keyboard.press('Enter');
         }
         this._lastSentAt = Date.now();
+        // Ждём старта: либо появился Stop, либо изменился текст ответа,
+        // либо вырос общий объём текста на странице.
         const startDeadline = Date.now() + 15_000;
+        const startBodyLen = await this.page
+            .evaluate(() => document.body.innerText.length)
+            .catch(() => 0);
         let started = false;
         while (Date.now() < startDeadline) {
             const gen = await this._isGenerating();
             const cur = await this._readLastAnswerTextClean().catch(() => '');
-            if (gen || (cur && cur !== beforeText)) {
+            const bodyLen = await this.page
+                .evaluate(() => document.body.innerText.length)
+                .catch(() => 0);
+            if (gen || (cur && cur !== beforeText) || bodyLen > startBodyLen) {
                 started = true;
                 break;
             }
@@ -425,18 +437,20 @@ export class DeepSeekBrowser {
         if (!started) {
             throw new Error('Ответ не начал генерироваться за 15с. Возможно, сообщение не отправилось.');
         }
+        // Ждём, пока ответ перестанет меняться. Условие "не генерируется"
+        // проверяем через рост текста, а НЕ через _isGenerating(): кнопка Stop
+        // у DeepSeek ненадёжна, и если она не находится, ответ бы никогда не
+        // вернулся.
         const deadline = Date.now() + timeout;
         let last = '';
         let stable = 0;
         this._abort = false;
         while (Date.now() < deadline) {
             if (this._abort) {
-                // Пользователь нажал Esc — вернём то, что успело сгенерироваться.
                 return last || '(прервано пользователем)';
             }
-            const gen = await this._isGenerating();
             const cur = await this._readLastAnswerTextClean().catch(() => '');
-            if (cur && cur === last && !gen) {
+            if (cur && cur === last) {
                 stable++;
                 if (stable >= 2)
                     return cur;
@@ -475,7 +489,6 @@ export class DeepSeekBrowser {
         });
         return { file: filePath, selectors: report };
     }
-    // ---------- список чатов ----------
     async _ensureSidebarOpen() {
         const toggles = [
             'button[aria-label*="sidebar" i]',
@@ -555,10 +568,12 @@ export class DeepSeekBrowser {
         try {
             const url = this.page.url();
             const m = url.match(/\/chat\/s\/([a-zA-Z0-9_-]+)/);
-            return m ? m[1] : null;
+            if (m)
+                return m[1];
+            return this._netChatId;
         }
         catch {
-            return null;
+            return this._netChatId;
         }
     }
     async close() {
@@ -570,4 +585,3 @@ export class DeepSeekBrowser {
         await killStaleChrome();
     }
 }
-//# sourceMappingURL=browser.js.map

package/dist/config.js

@@ -71,4 +71,3 @@ function deepMerge(target, source) {
 }
 export const CONFIG_PATHS = { HOME_CONFIG, PROJECT_CONFIG };
 export { ZAMES_HOME };
-//# sourceMappingURL=config.js.map

package/dist/confirm.js

@@ -79,4 +79,3 @@ export function formatDiffPreview(oldText, newText, { label } = {}) {
     const diff = unifiedDiff(oldText, newText, { label });
     return colorDiff(diff);
 }
-//# sourceMappingURL=confirm.js.map

package/dist/diff.js

@@ -35,4 +35,3 @@ export function colorDiff(diffText) {
         .join(String.fromCharCode(10));
 }
 export default { unifiedDiff, colorDiff };
-//# sourceMappingURL=diff.js.map

package/dist/gitTools.js

@@ -216,4 +216,3 @@ export function createGitTools(workdir) {
         },
     ];
 }
-//# sourceMappingURL=gitTools.js.map