Method chat

src/Codepilot.ts:46–124

* Starts the chat session and listens for user input.

()

     * Starts the chat session and listens for user input.
     */
    public async chat(): Promise<void> {
        // 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({
            model,
            prompt: new Prompt([
                new 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(this._index, 0.6),
                new ConversationHistory('history', 0.4),
                new UserMessage('{{$input}}', 500)
            ])
        });

        // Define main chat loop
        const _that = this;
        async function respond(botMessage: string) {
            async function completePrompt(input: string) {
                // Route users message to wave
                const result = await wave.completePrompt<string>(input);
                switch (result.status) {
                    case 'success':
                        const message = result.message as Message<string>;
                        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 = await entry.fn(args);
                                wave.addFunctionResultToHistory(message.function_call.name!, result);

                                // Call back in with the function result
                                await completePrompt('');
                            } else {
                                respond(Colorize.error(`Function '${message.function_call.name}' was not found.`));
                            }
                        } else {
                            // Call respond to display response and wait for user input
                            await respond(Colorize.output(message.content!));
                        }
                        break;
                    default:
                        if (result.message) {
                            await respond(Colorize.error(`${result.status}: ${result.message}`));
                        } else {
                            await respond(Colorize.error(`A result status of '${result.status}' was returned.`));
                        }
                        break;
                }
            }