Method chat

lib/Codepilot.js:74–156

* Starts the chat session and listens for user input.

()

     * Starts the chat session and listens for user input.
     */
    chat() {
        return __awaiter(this, void 0, void 0, function* () {
            // Create a readline interface object with the standard input and output streams
            const rl = readline.createInterface({
                input: process.stdin,
                output: process.stdout
            });
            // Create model and wave
            const model = this.createModel();
            const wave = new alphawave_1.AlphaWave({
                model,
                prompt: new promptrix_1.Prompt([
                    new promptrix_1.SystemMessage([
                        `You are an expert software developer.`,
                        `You are chatting with another developer who is asking for help with the project they're working on.`,
                    ].join('\n')),
                    new SourceCodeSection_1.SourceCodeSection(this._index, 0.6),
                    new promptrix_1.ConversationHistory('history', 0.4),
                    new promptrix_1.UserMessage('{{$input}}', 500)
                ])
            });
            // Define main chat loop
            const _that = this;
            function respond(botMessage) {
                return __awaiter(this, void 0, void 0, function* () {
                    function completePrompt(input) {
                        return __awaiter(this, void 0, void 0, function* () {
                            // Route users message to wave
                            const result = yield wave.completePrompt(input);
                            switch (result.status) {
                                case 'success':
                                    const message = result.message;
                                    if (message.function_call) {
                                        // Call function and add result to history
                                        const entry = _that._functions.get(message.function_call.name);
                                        if (entry) {
                                            const args = message.function_call.arguments ? JSON.parse(message.function_call.arguments) : {};
                                            const result = yield entry.fn(args);
                                            wave.addFunctionResultToHistory(message.function_call.name, result);
                                            // Call back in with the function result
                                            yield completePrompt('');
                                        }
                                        else {
                                            respond(internals_1.Colorize.error(`Function '${message.function_call.name}' was not found.`));
                                        }
                                    }
                                    else {
                                        // Call respond to display response and wait for user input
                                        yield respond(internals_1.Colorize.output(message.content));
                                    }
                                    break;
                                default:
                                    if (result.message) {
                                        yield respond(internals_1.Colorize.error(`${result.status}: ${result.message}`));
                                    }
                                    else {
                                        yield respond(internals_1.Colorize.error(`A result status of '${result.status}' was returned.`));
                                    }