SqlAgent
The SqlAgent can be used to ask questions in natural language and interact with SQL databases.
Example
Pass Data Source to SqlAgent and analyze the data.
from aita.agent.sql import SqlAgent
from aita.datasource.postgresql import PostgreSqlDataSource
from aita.prompt.base import BasicContextPromptTemplate
postgres_datasource = PostgreSqlDataSource("postgresql://localhost:5432/db")
sql_agent = SqlAgent("gpt-3.5-turbo")
.set_context_prompt(BasicContextPromptTemplate)
.add_datasource(postgres_datasource)
sql_agent.stream("Get the top 10 customers by purchase amount")
{% endcode}
Last updated