@travisennis/acai 0.0.5 → 0.0.6

package/dist/repl-new.js

@@ -0,0 +1,374 @@
+import { PromptError, processPrompt } from "./mentions.js";
+import { getProjectStatusLine } from "./repl/project-status-line.js";
+import { AssistantMessageComponent } from "./tui/components/assistant-message.js";
+import { FooterComponent } from "./tui/components/footer.js";
+import { PromptStatusComponent } from "./tui/components/prompt-status.js";
+import { ThinkingBlockComponent } from "./tui/components/thinking-block.js";
+import { ToolExecutionComponent } from "./tui/components/tool-execution.js";
+import { Welcome } from "./tui/components/welcome.js";
+import { CombinedAutocompleteProvider, Container, Editor, Loader, ProcessTerminal, Spacer, TUI, UserMessageComponent, } from "./tui/index.js";
+export class NewRepl {
+    options;
+    tui;
+    welcome;
+    editor;
+    chatContainer;
+    statusContainer;
+    promptStatus;
+    footer;
+    editorContainer; // Container to swap between editor and selector
+    isInitialized;
+    onInputCallback;
+    loadingAnimation = null;
+    onInterruptCallback;
+    lastSigintTime = 0;
+    pendingTools;
+    // Streaming message tracking
+    streamingComponent = null;
+    // thinking block tracking
+    thinkingBlockComponent = null;
+    constructor(options) {
+        this.options = options;
+        this.tui = new TUI(new ProcessTerminal());
+        this.welcome = new Welcome();
+        this.editor = new Editor();
+        this.chatContainer = new Container();
+        this.statusContainer = new Container();
+        this.editorContainer = new Container(); // Container to hold editor or selector
+        this.footer = new FooterComponent(options.agent.state);
+        this.promptStatus = new PromptStatusComponent(options.modelManager, {
+            projectStatus: "",
+            currentContextWindow: 0,
+            contextWindow: options.modelManager.getModelMetadata("repl").contextWindow,
+        });
+        this.editorContainer.addChild(this.editor); // Start with editor
+        this.editor.onRenderRequested = () => this.tui.requestRender();
+        this.isInitialized = false;
+        this.pendingTools = new Map();
+    }
+    async init() {
+        if (this.isInitialized) {
+            return;
+        }
+        // Setup autocomplete for file paths and slash commands
+        const autocompleteProvider = new CombinedAutocompleteProvider([...(await this.options.commands.getCompletions())], this.options.workspace.allowedDirs);
+        this.editor.setAutocompleteProvider(autocompleteProvider);
+        const { promptManager, terminal, modelManager, messageHistory, commands, promptHistory, } = this.options;
+        const modelConfig = modelManager.getModelMetadata("repl");
+        this.promptStatus.setState({
+            projectStatus: await getProjectStatusLine(),
+            currentContextWindow: 0,
+            contextWindow: modelConfig.contextWindow,
+        });
+        this.tui.addChild(this.welcome);
+        this.tui.addChild(this.chatContainer);
+        this.tui.addChild(this.statusContainer);
+        this.tui.addChild(new Spacer(1));
+        this.tui.addChild(this.footer);
+        this.tui.addChild(this.editorContainer); // Use container that can hold editor or selector
+        this.tui.addChild(this.promptStatus);
+        this.tui.setFocus(this.editor);
+        // Set up custom key handlers on the editor
+        this.editor.onEscape = () => {
+            // Intercept Escape key when processing
+            if (this.loadingAnimation && this.onInterruptCallback) {
+                this.onInterruptCallback();
+            }
+        };
+        this.editor.onCtrlC = () => {
+            this.handleCtrlC();
+        };
+        // Create editor for input
+        this.editor.onSubmit = async (text) => {
+            if (text.trim()) {
+                // see if the text contains a command
+                const commandResult = await commands.handle2({ userInput: text }, {
+                    tui: this.tui,
+                    container: this.chatContainer,
+                    inputContainer: this.editorContainer,
+                    editor: this.editor,
+                });
+                if (commandResult.break) {
+                    this.stop();
+                    process.exit(0);
+                }
+                if (commandResult.continue) {
+                    this.editor.setText("");
+                    this.tui.requestRender();
+                    return;
+                }
+                if (!promptManager.isPending()) {
+                    try {
+                        const processedPrompt = await processPrompt(text, {
+                            baseDir: process.cwd(),
+                            model: modelConfig,
+                        });
+                        for (const context of processedPrompt.context) {
+                            promptManager.addContext(context);
+                        }
+                        promptManager.set(processedPrompt.message);
+                    }
+                    catch (error) {
+                        if (error instanceof PromptError) {
+                            terminal.error(`Prompt processing failed: ${error.message}`);
+                            if (error.cause &&
+                                typeof error.cause === "object" &&
+                                "command" in error.cause &&
+                                typeof error.cause.command === "string") {
+                                terminal.error(`Command: ${error.cause.command}`);
+                            }
+                        }
+                        throw error; // Re-throw other errors
+                    }
+                }
+                else {
+                    promptHistory.push(promptManager.get());
+                }
+                // flag to see if the user prompt has added context
+                const hasAddedContext = promptManager.hasContext();
+                if (hasAddedContext) {
+                    const contextTokenCount = promptManager.getContextTokenCount();
+                    terminal.info(`Context will be added to prompt. (${contextTokenCount} tokens)`);
+                    terminal.lineBreak();
+                }
+                const userPrompt = promptManager.get();
+                const userMsg = promptManager.getUserMessage();
+                messageHistory.appendUserMessage(userMsg);
+                if (this.onInputCallback) {
+                    this.onInputCallback(userPrompt);
+                }
+            }
+        };
+        // Start
+        this.tui.start();
+        this.isInitialized = true;
+    }
+    async handle(event, state) {
+        if (!this.isInitialized) {
+            await this.init();
+        }
+        // Update footer with current stats
+        this.footer.updateState(state);
+        this.promptStatus.setState({
+            projectStatus: await getProjectStatusLine(),
+            currentContextWindow: state.usage.totalTokens,
+            contextWindow: this.options.modelManager.getModelMetadata("repl").contextWindow,
+        });
+        const eventType = event.type;
+        switch (eventType) {
+            case "agent-start":
+                // Show loading animation
+                this.editor.disableSubmit = true;
+                // Stop old loader before clearing
+                if (this.loadingAnimation) {
+                    this.loadingAnimation.stop();
+                }
+                this.statusContainer.clear();
+                this.loadingAnimation = new Loader(this.tui, "Working... (esc to interrupt)");
+                this.statusContainer.addChild(this.loadingAnimation);
+                this.tui.requestRender();
+                break;
+            case "step-start":
+                this.tui.requestRender();
+                break;
+            case "step-stop":
+                this.pendingTools.clear();
+                this.tui.requestRender();
+                break;
+            case "message-start":
+                if (event.role === "assistant") {
+                    // Create assistant component for streaming
+                    const assistantMessageComponent = new AssistantMessageComponent();
+                    this.streamingComponent = assistantMessageComponent;
+                    this.chatContainer.addChild(assistantMessageComponent);
+                    this.streamingComponent.updateContent(event);
+                    this.tui.requestRender();
+                }
+                break;
+            case "message":
+                if (event.role === "user") {
+                    // Show user message immediately and clear editor
+                    this.addMessageToChat(event);
+                    this.editor.setText("");
+                    this.tui.requestRender();
+                }
+                else if (event.role === "assistant") {
+                    // Update streaming component
+                    if (this.streamingComponent && event.role === "assistant") {
+                        this.streamingComponent.updateContent(event);
+                        this.tui.requestRender();
+                    }
+                }
+                break;
+            case "message-end":
+                if (this.streamingComponent && event.role === "assistant") {
+                    this.streamingComponent.updateContent(event);
+                    this.streamingComponent = null;
+                }
+                this.tui.requestRender();
+                break;
+            case "tool-call-start": {
+                // Create tool component for new tool call
+                const component = this.pendingTools.get(event.toolCallId);
+                if (component) {
+                    component.update(event);
+                }
+                else {
+                    const newComponent = new ToolExecutionComponent(event);
+                    this.pendingTools.set(event.toolCallId, newComponent);
+                    this.chatContainer.addChild(newComponent);
+                }
+                this.tui.requestRender();
+                break;
+            }
+            case "tool-call-end": {
+                // Update the existing tool component with the result
+                const component = this.pendingTools.get(event.toolCallId);
+                if (component) {
+                    component.update(event);
+                    this.pendingTools.delete(event.toolCallId);
+                    this.tui.requestRender();
+                }
+                break;
+            }
+            case "tool-call-update": {
+                // Update the existing tool component with the result
+                const component = this.pendingTools.get(event.toolCallId);
+                if (component) {
+                    component.update(event);
+                }
+                else {
+                    // Defensive: create component if update arrives before start
+                    const newComponent = new ToolExecutionComponent(event);
+                    this.pendingTools.set(event.toolCallId, newComponent);
+                    this.chatContainer.addChild(newComponent);
+                }
+                this.tui.requestRender();
+                break;
+            }
+            case "tool-call-error": {
+                // Update the existing tool component with the result
+                const component = this.pendingTools.get(event.toolCallId);
+                if (component) {
+                    component.update(event);
+                    this.pendingTools.delete(event.toolCallId);
+                    this.tui.requestRender();
+                }
+                break;
+            }
+            case "agent-stop":
+                // Stop loading animation
+                if (this.loadingAnimation) {
+                    this.loadingAnimation.stop();
+                    this.loadingAnimation = null;
+                    this.statusContainer.clear();
+                }
+                // Clear streaming component reference
+                if (this.streamingComponent) {
+                    this.streamingComponent = null;
+                }
+                this.pendingTools.clear();
+                this.editor.disableSubmit = false;
+                this.tui.requestRender();
+                break;
+            case "agent-error":
+                // Stop loading animation
+                if (this.loadingAnimation) {
+                    this.loadingAnimation.stop();
+                    this.loadingAnimation = null;
+                    this.statusContainer.clear();
+                }
+                // Clear streaming component reference
+                if (this.streamingComponent) {
+                    this.streamingComponent = null;
+                }
+                this.pendingTools.clear();
+                this.editor.disableSubmit = false;
+                this.tui.requestRender();
+                break;
+            case "thinking-start": {
+                const component = new ThinkingBlockComponent();
+                this.thinkingBlockComponent = component;
+                this.chatContainer.addChild(component);
+                this.thinkingBlockComponent.updateContent(event);
+                this.tui.requestRender();
+                break;
+            }
+            case "thinking":
+                if (this.thinkingBlockComponent) {
+                    this.thinkingBlockComponent.updateContent(event);
+                }
+                this.tui.requestRender();
+                break;
+            case "thinking-end":
+                if (this.thinkingBlockComponent) {
+                    this.thinkingBlockComponent.updateContent(event);
+                    this.thinkingBlockComponent = null;
+                }
+                this.tui.requestRender();
+                break;
+            default:
+                eventType;
+        }
+    }
+    addMessageToChat(message) {
+        if (message.role === "user") {
+            // Extract text content from content blocks
+            const textContent = message.content;
+            if (textContent) {
+                const userComponent = new UserMessageComponent(textContent);
+                this.chatContainer.addChild(userComponent);
+            }
+        }
+    }
+    async getUserInput() {
+        return new Promise((resolve) => {
+            this.onInputCallback = (text) => {
+                this.onInputCallback = undefined;
+                resolve(text);
+            };
+        });
+    }
+    clearEditor() {
+        this.editor.setText("");
+        this.tui.requestRender();
+    }
+    setInterruptCallback(callback) {
+        this.onInterruptCallback = callback;
+    }
+    async rerender() {
+        this.footer.updateState(this.options.agent.state);
+        this.promptStatus.setState({
+            projectStatus: await getProjectStatusLine(),
+            currentContextWindow: this.options.agent.state.usage.totalTokens,
+            contextWindow: this.options.modelManager.getModelMetadata("repl").contextWindow,
+        });
+        this.tui.requestRender();
+    }
+    handleCtrlC() {
+        // Handle Ctrl+C double-press logic
+        const now = Date.now();
+        const timeSinceLastCtrlC = now - this.lastSigintTime;
+        if (timeSinceLastCtrlC < 500) {
+            // Second Ctrl+C within 500ms - exit
+            this.options.messageHistory.save();
+            this.stop();
+            process.exit(0);
+        }
+        else {
+            // First Ctrl+C - clear the editor
+            this.clearEditor();
+            this.lastSigintTime = now;
+        }
+    }
+    stop() {
+        if (this.loadingAnimation) {
+            this.loadingAnimation.stop();
+            this.loadingAnimation = null;
+        }
+        if (this.isInitialized) {
+            this.tui.stop();
+            this.isInitialized = false;
+        }
+    }
+}

package/dist/repl.d.ts

@@ -1,12 +1,11 @@
 import type { CommandManager } from "./commands/manager.ts";
+import type { WorkspaceContext } from "./index.ts";
 import type { MessageHistory } from "./messages.ts";
-import type { ModelManager } from "./models/manager.js";
+import type { ModelManager } from "./models/manager.ts";
 import type { PromptManager } from "./prompts/manager.ts";
 import type { Terminal } from "./terminal/index.ts";
 import type { TokenCounter } from "./tokens/counter.ts";
 import type { TokenTracker } from "./tokens/tracker.ts";
-import type { ToolExecutor } from "./tool-executor.ts";
-import type { Message } from "./tools/types.ts";
 interface ReplOptions {
     messageHistory: MessageHistory;
     promptManager: PromptManager;
@@ -16,10 +15,9 @@ interface ReplOptions {
     commands: CommandManager;
     config: Record<PropertyKey, unknown>;
     tokenCounter: TokenCounter;
-    toolEvents: Map<string, Message[]>;
     showLastMessage: boolean;
-    toolExecutor?: ToolExecutor;
     promptHistory: string[];
+    workspace: WorkspaceContext;
 }
 export declare class Repl {
     private options;

package/dist/repl.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"repl.d.ts","sourceRoot":"","sources":["../source/repl.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAI5D,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAEpD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAO1D,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,qBAAqB,CAAC;AAEpD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAEvD,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAEhD,UAAU,WAAW;IACnB,cAAc,EAAE,cAAc,CAAC;IAC/B,aAAa,EAAE,aAAa,CAAC;IAC7B,YAAY,EAAE,YAAY,CAAC;IAC3B,YAAY,EAAE,YAAY,CAAC;IAC3B,QAAQ,EAAE,QAAQ,CAAC;IACnB,QAAQ,EAAE,cAAc,CAAC;IACzB,MAAM,EAAE,MAAM,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;IACrC,YAAY,EAAE,YAAY,CAAC;IAC3B,UAAU,EAAE,GAAG,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;IACnC,eAAe,EAAE,OAAO,CAAC;IACzB,YAAY,CAAC,EAAE,YAAY,CAAC;IAC5B,aAAa,EAAE,MAAM,EAAE,CAAC;CACzB;AAED,qBAAa,IAAI;IACf,OAAO,CAAC,OAAO,CAAc;IAC7B,OAAO,CAAC,eAAe,CAAU;gBAErB,OAAO,EAAE,WAAW;IAK1B,GAAG;CAwVV"}
+{"version":3,"file":"repl.d.ts","sourceRoot":"","sources":["../source/repl.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAG5D,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAGnD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AACpD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAM1D,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,qBAAqB,CAAC;AAEpD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAGxD,UAAU,WAAW;IACnB,cAAc,EAAE,cAAc,CAAC;IAC/B,aAAa,EAAE,aAAa,CAAC;IAC7B,YAAY,EAAE,YAAY,CAAC;IAC3B,YAAY,EAAE,YAAY,CAAC;IAC3B,QAAQ,EAAE,QAAQ,CAAC;IACnB,QAAQ,EAAE,cAAc,CAAC;IACzB,MAAM,EAAE,MAAM,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;IACrC,YAAY,EAAE,YAAY,CAAC;IAC3B,eAAe,EAAE,OAAO,CAAC;IACzB,aAAa,EAAE,MAAM,EAAE,CAAC;IACxB,SAAS,EAAE,gBAAgB,CAAC;CAC7B;AAED,qBAAa,IAAI;IACf,OAAO,CAAC,OAAO,CAAc;IAC7B,OAAO,CAAC,eAAe,CAAU;gBAErB,OAAO,EAAE,WAAW;IAK1B,GAAG;CA0PV"}