Простой «агент»: программа, выбирающая инструмент

Агент — это программа, которая разбивает задачу на шаги и при необходимости пользуется инструментами (калькулятор, поиск, файл). Типичный цикл: модель видит задачу и определяет, какой инструмент и с какими аргументами вызвать (часто в виде JSON); программа запускает инструмент и возвращает результат модели; модель строит ответ на основе этого результата.

В этом уроке самой модели нет: вместо неё даётся готовый словарь-«вызов», а мы пишем часть программы: по имени инструмента в вызове запускаем нужную функцию. Это упрощённый пример; в настоящих агентах есть и обработка ошибок, несколько шагов, разрешения и так далее.

def kvadrat(son):
    return son * son

VOSITALAR = {"kvadrat": kvadrat}

def agent(chaqiriq):
    nom = chaqiriq["vosita"]
    if nom in VOSITALAR:
        return VOSITALAR[nom](chaqiriq["son"])
    return "vosita topilmadi"

print(agent({"vosita": "kvadrat", "son": 7}))
print(agent({"vosita": "qidiruv", "son": 7}))

Результат49
vosita topilmadi

Правило безопасности: давайте агенту только необходимые инструменты и проверяйте результат. Неизвестный или неверный вызов не должен ронять программу: ситуация «vosita topilmadi» выше — пример такой обработки.