Class Codepilot

src/Codepilot.ts:11–150

 * The main class for the Codepilot application.
 */
export class Codepilot {
    private readonly _index: CodeIndex;
    private readonly _functions: Map<string, FunctionEntry> = new Map();

    /**
     * Creates a new `Codepilot` instance.
     * @param index The code index to use.
     */
    public constructor(index: CodeIndex) {
        this._index = index;
    }

    /**
     * Gets the code index.
     */
    public get index(): CodeIndex {
        return this._index;
    }

    /**
     * Registers a new function to be used in the chat completion.
     * @remarks
     * This is used to add new capabilities to Codepilot's chat feature
     * @param name The name of the function.
     * @param schema The schema of the function.
     * @param fn The function to be executed.
     */
    public addFunction<TArgs = Record<string, any>>(schema: ChatCompletionFunction, fn: (args: TArgs) => Promise<any>): this {
        this._functions.set(schema.name, { schema, fn });
        return this;
    }

    /**
     * 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