Function createJsonMathAgent

typescript/examples/multiSchema/src/agent.ts:62–111

(name: string, description: string,
     model: TypeChatLanguageModel,
     schema: string)

}

export function createJsonMathAgent<T extends object>
    (name: string, description: string,
     model: TypeChatLanguageModel,
     schema: string): MathAgent<T>
{
    async function _handleCall(func: string, args: any[]): Promise<unknown> {
        // implementation goes here
        console.log(`${func}(${args.map(arg => typeof arg === "number" ? arg : JSON.stringify(arg, undefined, 2)).join(", ")})`);
        switch (func) {
            case "add":
                return args[0] + args[1];
            case "sub":
                return args[0] - args[1];
            case "mul":
                return args[0] * args[1];
            case "div":
                return args[0] / args[1];
            case "neg":
                return -args[0];
            case "id":
                return args[0];
        }
        return NaN;
    }

    const _translator = createProgramTranslator(model, schema);
    const mathAgent : MathAgent<T> = {
        _translator,
        name: name,
        description: description,
        handleMessage: _handleMessage,
    };

    return mathAgent;

    async function _handleMessage(request: string): Promise<Result<T>> {
        const response = await _translator.translate(request);
        if (!response.success) {
            console.log(response.message);
            return response;
        }

        const program = response.data;
        console.log(getData(createModuleTextFromProgram(program)));
        console.log("Running program:");
        const result = await evaluateJsonProgram(program, _handleCall);
        console.log(`Result: ${typeof result === "number" ? result : "Error"}`);
        return success("Successful evaluation" as any);
    }
}