@probelabs/probe-chat 0.6.0-rc57 → 0.6.0-rc59

package/index.js

@@ -61,7 +61,7 @@ export function main() {
     .option('--images <urls>', 'Comma-separated list of image URLs to include in the message')
     .option('--json', 'Output the response as JSON in non-interactive mode')
     .option('--max-iterations <number>', 'Maximum number of tool iterations allowed (default: 30)')
-    .option('--prompt <value>', 'Use a custom prompt (values: architect, code-review, support, path to a file, or arbitrary string)')
+    .option('--prompt <value>', 'Use a custom prompt (values: architect, code-review, code-review-template, support, path to a file, or arbitrary string)')
     .option('--allow-edit', 'Enable the implement tool for editing files')
     .option('--implement-tool-backend <backend>', 'Choose implementation tool backend (aider, claude-code)')
     .option('--implement-tool-timeout <ms>', 'Implementation tool timeout in milliseconds')
@@ -250,7 +250,7 @@ export function main() {
   let customPrompt = null;
   if (options.prompt) {
     // Check if it's one of the predefined prompts
-    const predefinedPrompts = ['architect', 'code-review', 'support', 'engineer'];
+    const predefinedPrompts = ['architect', 'code-review', 'code-review-template', 'support', 'engineer'];
     if (predefinedPrompts.includes(options.prompt)) {
       process.env.PROMPT_TYPE = options.prompt;
       logInfo(chalk.blue(`Using predefined prompt: ${options.prompt}`));
@@ -358,7 +358,7 @@ export function main() {
         sessionId: options.sessionId,
         isNonInteractive: true,
         customPrompt: customPrompt,
-        promptType: options.prompt && ['architect', 'code-review', 'support', 'engineer'].includes(options.prompt) ? options.prompt : null,
+        promptType: options.prompt && ['architect', 'code-review', 'code-review-template', 'support', 'engineer'].includes(options.prompt) ? options.prompt : null,
         allowEdit: options.allowEdit
       });
       // Model/Provider info is logged via logInfo above if debug enabled
@@ -467,7 +467,7 @@ export function main() {
       sessionId: options.sessionId,
       isNonInteractive: false,
       customPrompt: customPrompt,
-      promptType: options.prompt && ['architect', 'code-review', 'support', 'engineer'].includes(options.prompt) ? options.prompt : null,
+      promptType: options.prompt && ['architect', 'code-review', 'code-review-template', 'support', 'engineer'].includes(options.prompt) ? options.prompt : null,
       allowEdit: options.allowEdit
     });
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@probelabs/probe-chat",
-  "version": "0.6.0-rc57",
+  "version": "0.6.0-rc59",
   "description": "CLI and web interface for Probe code search (formerly @probelabs/probe-web and @probelabs/probe-chat)",
   "main": "index.js",
   "type": "module",
@@ -32,7 +32,7 @@
     "@ai-sdk/anthropic": "^1.2.0",
     "@ai-sdk/google": "^1.2.1",
     "@ai-sdk/openai": "^1.3.0",
-    "@buger/probe": "^0.6.0-rc57",
+    "@buger/probe": "^0.6.0-rc59",
     "@opentelemetry/api": "^1.9.0",
     "@opentelemetry/exporter-trace-otlp-http": "^0.203.0",
     "@opentelemetry/resources": "^2.0.1",

package/probeChat.js

@@ -625,6 +625,8 @@ If you detect critical issues that should prevent the code from being merged, in
 
 The <fail> tag will cause the GitHub check to fail, drawing immediate attention to these critical issues.`,
 
+      'code-review-template': `You are going to perform code review according to provided user rules. Ensure to review only code provided in diff and latest commit, if provided. However you still need to fully understand how modified code works, and read dependencies if something is not clear.`,
+
       'engineer': `You are senior engineer focused on software architecture and design.
 Before jumping on the task you first, in details analyse user request, and try to provide elegant and concise solution.
 If solution is clear, you can jump to implementation right away, if not, you can ask user a clarification question, by calling attempt_completion tool, with required details.
@@ -1107,6 +1109,20 @@ When troubleshooting:
           });
         }
 
+        // Add warning message when reaching the last iteration
+        if (currentIteration === MAX_TOOL_ITERATIONS) {
+          const warningMessage = `⚠️ WARNING: You have reached the maximum tool iterations limit (${MAX_TOOL_ITERATIONS}). This is your final message. Please respond with the data you have so far. If something was not completed, honestly state what was not done and provide any partial results or recommendations you can offer.`;
+          
+          currentMessages.push({
+            role: 'user',
+            content: warningMessage
+          });
+          
+          if (this.debug) {
+            console.log(`[DEBUG] Added max iterations warning message at iteration ${currentIteration}`);
+          }
+        }
+
         this.tokenCounter.calculateContextSize(currentMessages);
         if (this.debug) console.log(`[DEBUG] Estimated context tokens BEFORE LLM call (Iter ${currentIteration}): ${this.tokenCounter.contextSize}`);