zames_pro 1.1.1 → 1.2.1

package/README.md

@@ -29,6 +29,11 @@ zames
 
 The agent works inside the directory it was started in and cannot leave it (sandbox).
 
+While the agent is working you can keep typing: press Enter to queue a message
+(it is sent right after the current task, in the same chat), or Esc / Ctrl+C to
+abort the current generation. This mirrors typing during generation on the
+DeepSeek website.
+
 ### Options
 
 ```

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "zames_pro",
-  "version": "1.1.1",
+  "version": "1.2.1",
   "description": "Terminal coding agent over chat.deepseek.com via Playwright",
   "type": "module",
   "bin": {

package/src/agent-loop.js

@@ -1,5 +1,6 @@
 import { buildSystemPrompt } from './system-prompt.js'
 import { getGitContext, formatGitContext } from './gitTools.js'
+import { parseXmlToolCalls } from './xml-toolcall.js'
 
 export async function runAgentLoop({
   browser,
@@ -47,6 +48,12 @@ export async function runAgentLoop({
   let message = task
   transcript?.log('task', { task })
 
+  // Счётчик «ответ похож на tool-call, но не распознан». Чтобы модель,
+  // написавшая битый JSON/XML, не останавливала агента молча, мы просим
+  // её переотправить вызов. Ограничиваем число таких попыток.
+  let malformedRetries = 0
+  const MAX_MALFORMED_RETRIES = 3
+
   for (let i = 0; i < maxIterations; i++) {
     onThinking()
     const rawResponse = await browser.ask(message)
@@ -69,6 +76,28 @@ export async function runAgentLoop({
     }
 
     if (!parsed) {
+      // Ответ не распознан как tool-call. Если он ПОХОЖ на попытку вызова
+      // (есть tool/invoke/parameter, но JSON/XML битый) — это почти всегда
+      // ошибка формата. Не считаем её финальным ответом (иначе агент молча
+      // остановится), а просим модель переотправить вызов корректно.
+      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 })
+        if (debugLog) {
+          console.error("внимание: ответ похож на tool-call, но не распознан (попытка " + malformedRetries + "/" + MAX_MALFORMED_RETRIES + ")")
+        }
+        message =
+          'Твой предыдущий ответ не распознан как вызов инструмента. ' +
+          'Ответь РОВНО одним JSON-объектом вызова инструмента, без текста до и после. ' +
+          'НЕ используй XML/DSML-теги — только JSON. ' +
+          'Например: {"tool": "Read", "args": {"path": "src/index.js"}}'
+        continue
+      }
+
       onAssistantMessage(rawResponse)
       transcript?.log('assistant_final', { message: rawResponse })
       return rawResponse
@@ -138,6 +167,27 @@ export async function runAgentLoop({
 
 // ============ JSON parsing ============
 
+function repairRawControlChars(str) {
+  let out = ''
+  let inString = false
+  let escape = false
+  for (let i = 0; i < str.length; i++) {
+    const c = str[i]
+    if (escape) { out += c; escape = false; continue }
+    if (c === '\\') { out += c; escape = true; continue }
+    if (c === '"') { inString = !inString; out += c; continue }
+    if (inString) {
+      if (c === '\n') { out += '\\n'; continue }
+      if (c === '\r') { out += '\\r'; continue }
+      if (c === '\t') { out += '\\t'; continue }
+      const code = c.charCodeAt(0)
+      if (code < 0x20) { out += '\\u' + code.toString(16).padStart(4, '0'); continue }
+    }
+    out += c
+  }
+  return out
+}
+
 function tryParse(str) {
   try {
     const obj = JSON.parse(str)
@@ -393,17 +443,34 @@ function parseToolCallPermissive(text) {
   // а не первый попавшийся '}'. Иначе вложенные объекты/массивы или
   // фигурные скобки внутри строк ломают разбор.
   const endIdx = findMatching(text, openIdx, '{', '}')
-  if (endIdx === -1) return null
 
-  const argsStr = text.slice(openIdx, endIdx + 1)
   if (tool === 'Edit') {
-    const e = parseEditArgs(argsStr)
-    if (e) return { tool, args: e }
+    // Edit-аргументы часто содержат сырые кавычки и переводы строк, из-за
+    // чего балансировка скобок сбоит. parseEditArgs рассчитан ровно на
+    // такой случай, поэтому пробуем его и на «хвосте» до конца текста,
+    // а не только на сбалансированном фрагменте.
+    const balanced = endIdx === -1 ? null : text.slice(openIdx, endIdx + 1)
+    const tailToEnd = text.slice(openIdx)
+    for (const frag of [balanced, tailToEnd]) {
+      if (!frag) continue
+      const e = parseEditArgs(frag)
+      if (e) return { tool, args: e }
+    }
+  }
+
+  // Балансировка скобок сбоит, если в строковых значениях есть сырые
+  // кавычки/скобки (частый случай для Bash/Write с кодом внутри). Тогда
+  // пробуем разобрать args из «хвоста» до конца текста и жадным парсером.
+  const frags = []
+  if (endIdx !== -1) frags.push(text.slice(openIdx, endIdx + 1))
+  frags.push(text.slice(openIdx))
+
+  for (const argsStr of frags) {
+    const args = parseArgsPermissive(argsStr)
+    if (args) return { tool, args }
+    const greedy = parseArgsGreedy(argsStr)
+    if (greedy) return { tool, args: greedy }
   }
-  const args = parseArgsPermissive(argsStr)
-  if (args) return { tool, args }
-  const greedy = parseArgsGreedy(argsStr)
-  if (greedy) return { tool, args: greedy }
   return null
 }
 
@@ -487,7 +554,7 @@ function findMatching(text, openIdx, openCh, closeCh) {
   return -1
 }
 
-function parseToolCall(text) {
+export function parseToolCall(text) {
   if (!text || typeof text !== 'string') return null
 
   let cleaned = text.trim()
@@ -515,9 +582,20 @@ function parseToolCall(text) {
     const repaired = raw.replace(/\\(?!["\\/bfnrtu])/g, '\\\\')
     const second = tryParse(repaired)
     if (second) return second
+
+    // Сырые переводы строк/табы внутри строковых значений (old_string,
+    // new_string, content) делают JSON невалидным. Экранируем их и
+    // пробуем снова, иначе многострочный вызов не распознаётся.
+    const ctrl = repairRawControlChars(raw)
+    const third = tryParse(ctrl)
+    if (third) return third
+    const ctrlArr = tryParseArray(ctrl)
+    if (ctrlArr) return ctrlArr
   }
 
-  const permissive = parseToolCallPermissive(cleaned)
+  const permissive =
+    parseToolCallPermissive(cleaned) ||
+    parseToolCallPermissive(repairRawControlChars(cleaned))
   if (permissive) return { ...permissive, _permissive: true }
 
   // Fallback: модель могла обернуть JSON в прозу и/или добавить мусорные
@@ -526,11 +604,20 @@ function parseToolCall(text) {
   const toolIdx = cleaned.search(/["']?tool["']?\s:/)
   if (toolIdx > 0) {
     let tail = cleaned.slice(toolIdx)
-    tail = tail.replace(/<[^>]>.*$/s, '').trim()
-    const tailPermissive = parseToolCallPermissive('{"' + tail)
+    // Регексп обязан матчить многосимвольные теги (<|DSML|invoke ...>),
+    // поэтому <[^>]*>, а не <[^>]> — иначе DSML-хвост не срезается,
+    // permissive-разбор падает и агент молча останавливается.
+    tail = tail.replace(/<[^>]*>.*$/s, '').trim()
+    const tailPermissive =
+      parseToolCallPermissive('{"' + tail) ||
+      parseToolCallPermissive('{"' + repairRawControlChars(tail))
     if (tailPermissive) return { ...tailPermissive, _permissive: true }
   }
 
+  // Последний fallback: модель ответила XML/DSML-блоком вместо JSON.
+  const xmlCalls = parseXmlToolCalls(cleaned)
+  if (xmlCalls) return Array.isArray(xmlCalls) ? xmlCalls : [xmlCalls]
+
   return null
 }
 

package/src/browser.js

@@ -3,6 +3,7 @@ import path from 'path'
 import os from 'os'
 import fs from 'fs/promises'
 import { execSync } from 'child_process'
+import { theme } from './theme.js'
 
 const USER_DATA_DIR = path.join(os.homedir(), '.zames', 'profile')
 const CHAT_URL = 'https://chat.deepseek.com/'
@@ -78,6 +79,7 @@ export class DeepSeekBrowser {
     askRetries = 3,
     stabilityChecks = 3,
     stabilityDelayMs = 1000,
+    minSendIntervalMs = 15000,
   } = {}) {
     this.headless = headless
     this.debug = debug
@@ -86,6 +88,10 @@ 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.context = null
     this.page = null
   }
@@ -418,6 +424,20 @@ export class DeepSeekBrowser {
     }
   }
 
+  // Ждём, пока с прошлой отправки пройдёт minSendIntervalMs. Защита от
+  // «Messages too frequent. Try again later.» на [chat.deepseek.com](https://chat.deepseek.com/).
+  async _waitForSendSlot() {
+    if (!this._lastSentAt) return // первая отправка — пауза не нужна
+    const gap = this.minSendIntervalMs - (Date.now() - this._lastSentAt)
+    if (gap <= 0) return
+    console.error(
+      theme.warn(
+        `⏳ пауза ${Math.ceil(gap / 1000)}с перед отправкой (лимит частоты DeepSeek)...`,
+      ),
+    )
+    await this.page.waitForTimeout(gap)
+  }
+
   async _askOnce(prompt, { timeout }) {
     const input = await this._findVisible(INPUT_SELECTORS, 10_000)
     if (!input) {
@@ -428,6 +448,7 @@ export class DeepSeekBrowser {
 
     const beforeText = await this._readLastAnswerTextClean().catch(() => '')
 
+    await this._waitForSendSlot()
     await this._setInputText(input, prompt)
     await this.page.waitForTimeout(200)
 
@@ -440,6 +461,7 @@ export class DeepSeekBrowser {
     } catch {
       await this.page.keyboard.press('Enter')
     }
+    this._lastSentAt = Date.now()
 
     const startDeadline = Date.now() + 15_000
     let started = false

package/src/config.js

@@ -43,6 +43,7 @@ const DEFAULTS = {
     askRetries: 3,
     stabilityChecks: 3,
     stabilityDelayMs: 1000,
+    minSendIntervalMs: 15000,
   },
 }