Class MathAgent

python/examples/multiSchema/agents.py:46–89

class MathAgent:
    _validator: TypeChatProgramValidator
    _translator: TypeChatProgramTranslator

    def __init__(self, model: TypeChatLanguageModel):
        super().__init__()
        self._validator = TypeChatProgramValidator()
        self._translator = TypeChatProgramTranslator(model, self._validator, math_schema.MathAPI)

    async def _handle_json_program_call(self, func: str, args: Sequence[object]) -> int | float:
        print(f"{func}({json.dumps(args)}) ")
        
        for arg in args:
            if not isinstance(arg, (int, float)):
                raise ValueError("All arguments are expected to be numeric.")

        args = cast(Sequence[int | float], args)

        match 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 -1 * args[0]
            case "id":
                return args[0]
            case _:
                raise ValueError(f'Unexpected function name {func}')

    async def handle_request(self, line: str):
        result = await self._translator.translate(line)
        if isinstance(result, Failure):
            print(result.message)
        else:
            result = result.value
            print(json.dumps(result, indent=2))

            math_result = await evaluate_json_program(result, self._handle_json_program_call)
            print(f"Math Result: {math_result}")


class MusicAgent: