Class Codepilot

lib/Codepilot.js:44–178

* The main class for the Codepilot application.

 * The main class for the Codepilot application.
 */
class Codepilot {
    /**
     * Creates a new `Codepilot` instance.
     * @param index The code index to use.
     */
    constructor(index) {
        this._functions = new Map();
        this._index = index;
    }
    /**
     * Gets the code index.
     */
    get index() {
        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.
     */
    addFunction(schema, fn) {
        this._functions.set(schema.name, { schema, fn });
        return this;
    }
    /**
     * 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