minovative-mind-cli 1.5.0 → 2.0.0

package/dist/services/ai.js

@@ -1,10 +1,46 @@
-import { DEFAULT_MODEL, GEMINI_MODELS, MAX_OUTPUT_TOKENS } from '../utils/config.js';
-import { toolDeclarations } from './agent-tools.js';
+import { GEMINI_MODELS, DEFAULT_MODEL, MAX_OUTPUT_TOKENS } from '../utils/config.js';
+import { getToolDeclarations } from './agent-tools.js';
 import { ProxyClient } from './proxyClient.js';
 import { getAuthorizedIdToken } from './auth.js';
 import { debugLog } from '../utils/logger.js';
-import { GENERAL_CHAT_INSTRUCTION, PLAN_EXECUTION_INSTRUCTION, CONTEXT_SYSTEM_INSTRUCTION, INTENT_ROUTER_SYSTEM_INSTRUCTION, WEB_SEARCH_SYSTEM_INSTRUCTION, } from '../utils/systemPrompts.js';
+import { GENERAL_CHAT_INSTRUCTION, PLAN_EXECUTION_INSTRUCTION, PLAN_MODE_INSTRUCTION, CONTEXT_SYSTEM_INSTRUCTION, INTENT_ROUTER_SYSTEM_INSTRUCTION, WEB_SEARCH_SYSTEM_INSTRUCTION, EXECUTION_COMPLEXITY_SYSTEM_INSTRUCTION, INVESTIGATION_COMPLEXITY_SYSTEM_INSTRUCTION, } from '../utils/systemPrompts.js';
+import { getMetricCollector } from './metrics.js';
+// ─── Model Overrides (for Regression Testing) ────────────────────────
+let contextModelOverride = null;
+let contextTempOverride = null;
+let executionModelOverride = null;
+let executionTempOverride = null;
+export function setModelOverride(agent, overrideStr) {
+    const [model, temp] = overrideStr.split(':');
+    if (agent === 'context') {
+        contextModelOverride = model || null;
+        if (temp)
+            contextTempOverride = parseFloat(temp);
+    }
+    else {
+        executionModelOverride = model || null;
+        if (temp)
+            executionTempOverride = parseFloat(temp);
+    }
+}
+export function clearModelOverrides() {
+    contextModelOverride = null;
+    contextTempOverride = null;
+    executionModelOverride = null;
+    executionTempOverride = null;
+}
 // ─── AI Service ──────────────────────────────────────────────────────
+let globalActiveModel = DEFAULT_MODEL;
+export function setGlobalActiveModel(model) {
+    globalActiveModel = model;
+}
+export function getGlobalActiveModel() {
+    return globalActiveModel;
+}
+let globalLatestUsageMetadata = undefined;
+export function getGlobalLatestUsageMetadata() {
+    return globalLatestUsageMetadata;
+}
 const proxyClient = new ProxyClient();
 // ─── History Limits ──────────────────────────────────────────────────
 /** Maximum number of Content entries to keep in the sliding history window. */
@@ -91,7 +127,7 @@ export class ProxyChatSession {
             this.history = this.history.slice(trimCount);
         }
     }
-    async sendMessage(message, additionalText, abortSignal) {
+    async sendMessage(message, additionalText, abortSignal, onChunk) {
         const idToken = await getAuthorizedIdToken();
         if (!idToken) {
             throw new Error('You are not signed in. Please run `minovative-mind-cli login` first.');
@@ -102,18 +138,29 @@ export class ProxyChatSession {
             newParts = [{ text: truncatePartText(message) }];
         }
         else {
-            // It's an array of function responses — truncate output strings
-            newParts = message.map((m) => ({
-                functionResponse: {
-                    name: m.functionResponse.name,
-                    response: {
-                        ...m.functionResponse.response,
-                        output: typeof m.functionResponse.response.output === 'string'
-                            ? truncatePartText(m.functionResponse.response.output)
-                            : m.functionResponse.response.output,
+            // It's an array of function responses
+            newParts = [];
+            for (const m of message) {
+                newParts.push({
+                    functionResponse: {
+                        name: m.functionResponse.name,
+                        response: {
+                            ...m.functionResponse.response,
+                            output: typeof m.functionResponse.response.output === 'string'
+                                ? truncatePartText(m.functionResponse.response.output)
+                                : m.functionResponse.response.output,
+                        },
                     },
-                },
-            }));
+                });
+                // If the tool execution returned inlineData (e.g. reading a PDF), append it as a sibling Part
+                if (m.functionResponse.response.inlineData) {
+                    newParts.push({
+                        inlineData: m.functionResponse.response.inlineData
+                    });
+                    // Remove it from the textual output payload to avoid schema violations in functionResponse
+                    delete newParts[newParts.length - 2].functionResponse.response.inlineData;
+                }
+            }
         }
         if (additionalText) {
             newParts.push({ text: truncatePartText(additionalText) });
@@ -125,8 +172,20 @@ export class ProxyChatSession {
         // Prune old history before sending to keep payload bounded
         this.pruneHistory();
         const effectiveGenerationConfig = { ...this.generationConfig };
-        const result = await proxyClient.generateFunctionCallViaProxy(idToken, this.modelName, this.history, this.tools, undefined, this.systemInstruction, effectiveGenerationConfig, undefined, abortSignal);
+        const result = await proxyClient.generateFunctionCallViaProxy(idToken, this.modelName, this.history, this.tools, undefined, // toolConfig
+        this.systemInstruction, effectiveGenerationConfig, onChunk ? { onChunk } : undefined, // streamCallbacks
+        abortSignal);
         this.latestUsageMetadata = result.usageMetadata;
+        if (result.usageMetadata) {
+            globalLatestUsageMetadata = result.usageMetadata;
+        }
+        // Track token usage metrics
+        if (result.usageMetadata) {
+            const collector = getMetricCollector();
+            if (collector) {
+                collector.recordTokenUsage(result.usageMetadata.promptTokens || 0, result.usageMetadata.candidatesTokens || 0, result.usageMetadata.cachedTokens || 0);
+            }
+        }
         // Append model response to history
         let modelParts = [];
         const allFunctionCalls = [...(result.functionCalls || [])];
@@ -184,9 +243,12 @@ export class ProxyChatSession {
  * Creates a new shared multi-turn chat session via the proxy.
  */
 export function createSharedChatSession() {
-    return new ProxyChatSession(DEFAULT_MODEL, GENERAL_CHAT_INSTRUCTION, [], {
+    let model = executionModelOverride || getGlobalActiveModel();
+    if (model === 'auto')
+        model = GEMINI_MODELS.FLASH_3_5; // Will be overridden per-turn in executeSingleTurn
+    return new ProxyChatSession(model, GENERAL_CHAT_INSTRUCTION, [], {
         maxOutputTokens: MAX_OUTPUT_TOKENS,
-        temperature: 1,
+        temperature: executionTempOverride !== null ? executionTempOverride : 1,
         topP: 0.95,
         topK: 40,
     });
@@ -200,22 +262,35 @@ export function getGeneralChatConfig() {
 export function getPlanExecutionConfig() {
     return {
         systemInstruction: PLAN_EXECUTION_INSTRUCTION,
-        tools: [{ functionDeclarations: toolDeclarations }],
+        tools: [{ functionDeclarations: getToolDeclarations() }],
+    };
+}
+export function getPlanModeConfig() {
+    return {
+        systemInstruction: PLAN_MODE_INSTRUCTION,
+        tools: [], // No tools allowed in plan mode
     };
 }
 /**
  * Compresses a large string of text using gemini-3.1-flash-lite.
  * Used for shrinking context payloads to prevent OOM/choking.
  */
-export async function compressTextUsingFlashLite(text, instruction = 'Summarize the following text concisely. Preserve the most critical technical details, function names, and architecture logic. Keep it under 1500 characters.') {
-    if (!text || text.length < 1000)
+export async function compressTextUsingFlashLite(text, instruction = 'Summarize the following text concisely. Preserve the most critical technical details, function names, and architecture logic. Keep it under 1500 characters.', inlineData) {
+    if (!text || (text.length < 1000 && !inlineData))
         return text; // Don't compress tiny texts
     try {
         const idToken = await getAuthorizedIdToken();
         if (!idToken)
             return text;
-        const contents = [{ role: 'user', parts: [{ text }] }];
-        const result = await proxyClient.generateFunctionCallViaProxy(idToken, 'gemini-3.1-flash-lite', contents, [], // no tools
+        let model = getGlobalActiveModel();
+        if (model === 'auto')
+            model = GEMINI_MODELS.FLASH_LITE_3_1;
+        const parts = [{ text }];
+        if (inlineData) {
+            parts.push({ inlineData });
+        }
+        const contents = [{ role: 'user', parts }];
+        const result = await proxyClient.generateFunctionCallViaProxy(idToken, model, contents, [], // no tools
         undefined, instruction, { temperature: 0.2 });
         let textPart = '';
         if (result.parts) {
@@ -236,9 +311,13 @@ export async function compressTextUsingFlashLite(text, instruction = 'Summarize
     }
 }
 // ─── Context Agent Service ───────────────────────────────────────────
-export const CONTEXT_AGENT_MODEL = DEFAULT_MODEL;
-export function createContextAgentSession() {
-    const contextTools = [
+/**
+ * Returns the tool declarations for the read-only Context Agent.
+ * Extracted as a reusable function so investigation sub-agents can import
+ * the same schema without duplicating the declarations.
+ */
+export function getContextToolDeclarations() {
+    return [
         {
             functionDeclarations: [
                 {
@@ -271,7 +350,7 @@ export function createContextAgentSession() {
                 },
                 {
                     name: 'read_file',
-                    description: 'Read a single file. Use startLine and endLine to read specific chunks of massive files.',
+                    description: 'Read a single file. Supports text files and native parsing of .pdf files (including math and diagrams). Use startLine and endLine to read specific chunks of massive text files.',
                     parameters: {
                         type: 'OBJECT',
                         properties: {
@@ -346,7 +425,8 @@ export function createContextAgentSession() {
                         "language's native AST parser (e.g., TypeScript compiler API, Python ast module, go/parser). " +
                         'The script is executed from a temporary directory and automatically cleaned up after execution. ' +
                         'Output should be structured JSON to stdout with name, type, startLine, endLine for each code element. ' +
-                        'Use the results to make precise read_file calls with exact startLine/endLine instead of guessing.',
+                        'Use the results to make precise read_file calls with exact startLine/endLine instead of guessing. ' +
+                        'CRITICAL: Do not use this tool on binary, document, or non-code files (e.g. PDF, image, audio, docx).',
                     parameters: {
                         type: 'OBJECT',
                         properties: {
@@ -393,26 +473,75 @@ export function createContextAgentSession() {
                         required: ['query'],
                     },
                 },
+                {
+                    name: 'semantic_search',
+                    description: 'Search the codebase by meaning and concept rather than exact text match. ' +
+                        'Use this when you need to find code related to a concept, pattern, or behavior ' +
+                        "but don't know the exact variable or function names to grep for. " +
+                        'Examples: "error handling for API requests", "user authentication flow", ' +
+                        '"database connection pooling logic". Returns ranked results with file paths, ' +
+                        'line ranges, and similarity scores.',
+                    parameters: {
+                        type: 'OBJECT',
+                        properties: {
+                            query: {
+                                type: 'STRING',
+                                description: "Natural language description of what you're looking for in the codebase.",
+                            },
+                            topK: {
+                                type: 'NUMBER',
+                                description: 'Number of results to return. Defaults to 5, maximum 15.',
+                            },
+                        },
+                        required: ['query'],
+                    },
+                },
             ],
         },
     ];
-    return new ProxyChatSession(CONTEXT_AGENT_MODEL, CONTEXT_SYSTEM_INSTRUCTION, contextTools, {
+}
+export function createContextAgentSession() {
+    const contextTools = getContextToolDeclarations();
+    let model = contextModelOverride || getGlobalActiveModel();
+    if (model === 'auto')
+        model = GEMINI_MODELS.FLASH_3_5;
+    return new ProxyChatSession(model, CONTEXT_SYSTEM_INSTRUCTION, contextTools, {
         maxOutputTokens: MAX_OUTPUT_TOKENS,
-        temperature: 1,
+        temperature: contextTempOverride !== null ? contextTempOverride : 1,
         topP: 0.95,
         topK: 40,
     });
 }
 // ─── Intent Router Service ───────────────────────────────────────────
-export const INTENT_ROUTER_MODEL = GEMINI_MODELS.FLASH_LITE_3_1;
 export function createIntentRouterSession() {
-    return new ProxyChatSession(INTENT_ROUTER_MODEL, INTENT_ROUTER_SYSTEM_INSTRUCTION, [], // no tools
+    let model = getGlobalActiveModel();
+    if (model === 'auto')
+        model = GEMINI_MODELS.FLASH_LITE_3_1;
+    return new ProxyChatSession(model, INTENT_ROUTER_SYSTEM_INSTRUCTION, [], // no tools
+    { temperature: 0, responseMimeType: 'application/json' });
+}
+// ─── Execution Complexity Router Service ──────────────────────────────
+export function createExecutionComplexitySession() {
+    let model = getGlobalActiveModel();
+    if (model === 'auto')
+        model = GEMINI_MODELS.FLASH_LITE_3_1;
+    return new ProxyChatSession(model, EXECUTION_COMPLEXITY_SYSTEM_INSTRUCTION, [], // no tools
+    { temperature: 0, responseMimeType: 'application/json' });
+}
+// ─── Investigation Complexity Router Service ─────────────────────────
+export function createInvestigationComplexitySession() {
+    let model = getGlobalActiveModel();
+    if (model === 'auto')
+        model = GEMINI_MODELS.FLASH_LITE_3_1;
+    return new ProxyChatSession(model, INVESTIGATION_COMPLEXITY_SYSTEM_INSTRUCTION, [], // no tools
     { temperature: 0, responseMimeType: 'application/json' });
 }
 // ─── Web Search Agent Service ───────────────────────────────────────────
-export const WEB_SEARCH_AGENT_MODEL = GEMINI_MODELS.FLASH_LITE_3_1;
 export function createWebSearchAgentSession() {
-    return new ProxyChatSession(WEB_SEARCH_AGENT_MODEL, WEB_SEARCH_SYSTEM_INSTRUCTION, [{ googleSearch: {} }], {
+    let model = getGlobalActiveModel();
+    if (model === 'auto')
+        model = GEMINI_MODELS.FLASH_3_5;
+    return new ProxyChatSession(model, WEB_SEARCH_SYSTEM_INSTRUCTION, [{ googleSearch: {} }], {
         maxOutputTokens: MAX_OUTPUT_TOKENS,
         temperature: 1,
         topP: 0.95,
@@ -432,7 +561,10 @@ export async function generateChatTitle(firstMessage) {
             return firstMessage.substring(0, maxLength);
         const instruction = "You are a helpful assistant that generates extremely concise chat titles (max 4-5 words) based on a user's first message. Output ONLY the title, no quotes, no markdown, no punctuation.";
         const contents = [{ role: 'user', parts: [{ text: firstMessage.substring(0, 500) }] }];
-        const result = await proxyClient.generateFunctionCallViaProxy(idToken, 'gemini-3.1-flash-lite', contents, [], // no tools
+        let model = getGlobalActiveModel();
+        if (model === 'auto')
+            model = GEMINI_MODELS.FLASH_LITE_3_1;
+        const result = await proxyClient.generateFunctionCallViaProxy(idToken, model, contents, [], // no tools
         undefined, instruction, { temperature: 0.2 });
         let title = '';
         if (result.parts) {

package/dist/services/changeLogger.d.ts

@@ -1,31 +1,173 @@
+/**
+ * Represents a single file modification tracked within a workspace changeset.
+ * Captures the state of a file before a change was applied, facilitating reversion.
+ */
 export interface FileChange {
+    /**
+     * Relative or absolute file path of the affected file.
+     */
     filePath: string;
+    /**
+     * The original content of the file before any modification in the changeset.
+     * A value of `null` indicates the file was newly created and did not exist previously.
+     */
     originalContent: string | null;
+    /**
+     * The nature of the file operation that occurred.
+     * - 'create': The file was newly written to disk.
+     * - 'modify': An existing file was updated.
+     * - 'delete': An existing file was removed.
+     */
     action: 'create' | 'modify' | 'delete';
 }
+/**
+ * Represents a transactional unit of workspace modifications.
+ * Groups related file changes under a shared timestamp and intent description.
+ */
 export interface ChangeSet {
+    /**
+     * Unix epoch timestamp (in milliseconds) when this changeset was initiated.
+     */
     timestamp: number;
+    /**
+     * Developer or system-provided description explaining the intent of the changes (e.g. "Edit index.ts").
+     */
     description: string;
+    /**
+     * Array of individual file modifications tracking the original states prior to this changeset.
+     */
     changes: FileChange[];
+    /**
+     * The execution outcome status of this changeset.
+     * - 'partial': The change operation is still in progress or was interrupted.
+     * - 'complete': All planned modifications in the transaction finished successfully.
+     */
     status?: 'complete' | 'partial';
 }
+/**
+ * Service class responsible for tracking, persisting, and reverting file modifications in the workspace.
+ * Uses a local project state file (`revert_state.json`) to persist state across CLI invocations.
+ */
 declare class ChangeLogger {
+    /**
+     * Ordered history stack of committed changesets, oldest first, newest last.
+     */
     private changeStack;
+    /**
+     * The active changeset currently receiving recorded modifications.
+     * Null if no change transaction is actively in progress.
+     */
     private currentChangeSet;
+    /**
+     * Absolute path to the active workspace root directory.
+     */
     private workspaceRoot;
+    /**
+     * Maximum number of historical changesets retained in memory and on disk.
+     * Older changesets are discarded on save once this limit is exceeded.
+     */
     private readonly MAX_HISTORY;
+    /**
+     * Whether the change logger is currently enabled.
+     */
+    private isEnabled;
+    getIsEnabled(): boolean;
+    setIsEnabled(enabled: boolean): void;
+    /**
+     * Initializes the ChangeLogger by setting the workspace root and loading existing
+     * history from the local `.minovativemind/revert_state.json` cache file.
+     *
+     * @param workspaceRoot - The absolute path to the project workspace root.
+     */
     init(workspaceRoot: string): void;
+    /**
+     * Persists the current committed change stack to the local cache directory.
+     * Enforces the `MAX_HISTORY` retention policy by slicing out older entries.
+     * Does nothing if the workspace root has not yet been initialized.
+     */
     private saveState;
+    /**
+     * Retrieves a copy of the active history stack.
+     *
+     * @returns A shallow copy of the array containing all tracked changesets.
+     */
     getHistory(): ChangeSet[];
+    /**
+     * Traverses backward through the history stack, popping and collecting changesets
+     * until a changeset matching the specified timestamp is encountered (inclusive).
+     * Persists the resulting truncated stack after execution.
+     *
+     * @param timestamp - The unique timestamp identifier of the target changeset to revert back to.
+     * @returns An array of popped changesets ordered from newest to oldest.
+     */
     popUntil(timestamp: number): ChangeSet[];
+    /**
+     * Initiates a new transactional changeset. If a previous changeset was in progress and
+     * uncommitted, it is automatically finalized and committed to the history stack first.
+     *
+     * @param description - Descriptive label summarizing the purpose of the new changeset.
+     */
     startChangeSet(description: string): void;
+    /**
+     * Records a file change within the active changeset.
+     * If no changeset is active, an anonymous default changeset is automatically initialized.
+     *
+     * @remarks
+     * Idempotency Guard: If the specified file already has an entry in the current changeset,
+     * the original state is preserved. This ensures that multiple successive modifications to
+     * the same file within a single transaction map back to the true original state prior to
+     * the onset of the transaction.
+     *
+     * @param filePath - The relative or absolute path of the file being altered.
+     * @param originalContent - The text content of the file prior to the change, or null if created.
+     * @param action - The classification of the operation ('create', 'modify', or 'delete').
+     */
     logChange(filePath: string, originalContent: string | null, action: 'create' | 'modify' | 'delete'): void;
+    /**
+     * Marks the status of the active changeset as 'complete', indicating that all scheduled
+     * operations within the scope of this transaction finished without interruption.
+     */
     markComplete(): void;
+    /**
+     * Retrieves a list of file paths that have been modified in the currently active changeset.
+     *
+     * @returns An array of string file paths, or an empty array if no changeset is active.
+     */
+    getChangedFiles(): string[];
+    /**
+     * Commits the active changeset to the history stack and persists the state to disk,
+     * provided that the changeset actually contains one or more recorded file modifications.
+     * Resets the active changeset state to null.
+     */
     commitChangeSet(): void;
+    /**
+     * Retrieves the most recently committed changeset from the history stack without removing it.
+     *
+     * @returns The latest committed ChangeSet, or null if the stack is currently empty.
+     */
     getLastChangeSet(): ChangeSet | null;
+    /**
+     * Returns the active, uncommitted changeset currently receiving modifications.
+     *
+     * @returns The in-progress ChangeSet, or null if no changeset is active.
+     */
     getCurrentChangeSet(): ChangeSet | null;
+    /**
+     * Removes and returns the most recently committed changeset from the history stack.
+     * Persists the newly truncated stack state back to the cache.
+     *
+     * @returns The removed ChangeSet, or null if no historical changes exist.
+     */
     popLastChangeSet(): ChangeSet | null;
+    /**
+     * Checks whether the history stack contains any tracked changesets.
+     *
+     * @returns True if at least one changeset is recorded; false otherwise.
+     */
     hasChanges(): boolean;
 }
+/**
+ * Singleton instance of the ChangeLogger service exported for application-wide tracking.
+ */
 export declare const changeLogger: ChangeLogger;
 export {};