Агентный RAG, часть 6: запросы к SQL-базе

О чём урок

В предыдущем руководстве мы рассмотрели, как агентский RAG может обрабатывать сложные запросы к структурированным данным в виде таблиц с использованием pandas. Теперь мы увидим, как мы можем сделать то же самое для баз данных SQL. Рассмотрим сценарий, аналогичный предыдущему руководству, где у нас есть результаты оценки для приложения LLM. Однако вместо файла CSV эти данные теперь хранятся в базе данных SQLite. Пользователи по-прежнему могут задавать вопросы типа "Каков средний балл для конкретного варианта использования?" или "Какая конфигурация имеет наименьшую задержку?", но теперь мы будем отвечать на них с помощью SQL-запросов вместо операций pandas. В этом руководстве мы рассмотрим: Настройка базы данных SQLite Создание функции для выполнения SQL-запросов Создание агента для запросов к базам данных SQL Запуск агента с различными типами запросов Реализуя эти методы, мы расширим нашу агентскую RAG-систему для обработки структурированных данных в базах данных SQL, дополняя нашу предыдущую работу с табличными данными в pandas. Давайте начнем с настройки нашей среды и создания нашей базы данных SQLite. Для начала нам необходимо установить библиотеку cohere и создать клиент Cohere. Далее мы создадим базу данных SQLite для хранения результатов нашей оценки. SQLite — это легковесный, бессерверный движок базы данных, который идеально подходит для приложений малого и среднего размера. Вот что мы собираемся сделать: Создать новый файл базы данных SQLite с именем evaluation_results.db. Создать таблицу с именем evaluation_results со столбцами для usecase, run, score, temperature, tokens и latency. Вставить образцы данных в таблицу для имитации наших результатов оценки. Важно: данные можно найти здесь. Убедитесь, что файл находится в том же каталоге, что и этот ноутбук, для корректной работы импортов. Далее мы определим функцию с именем sql_table_query, которая позволит нам выполнять SQL-запросы к нашей базе данных evaluation_results.

План урока

1. Настройка
2. Создание базы данных SQLite
3. Создание функции для запросов к базе данных SQL
4. Настройка инструмента для взаимодействия с базой данных
5. Создание агента для запросов к данным SQL
6. Запуск агента
7. Резюме