Quick Start

This guide will walk you through setting up Aita and start coding with Aita.

Prerequisites

Before you start, make sure you have the following installed:

  • Python 3.9 or higher

Installation

To install Aita, run the following command:

bash
pip install aita

Getting Started

To start using Aita, you can run the following code:

python

from aita.agent.pandas import PandasAgent
import pandas as pd

df = pd.DataFrame({
    'customer_id': [1, 2, 3, 4, 5],
    'purchase_amount': [100, 200, 300, 400, 500]
})
pandas_agent = PandasAgent("gpt-3.5-turbo") \
    .add_dataframe(df)

print(pandas_agent.stream("I want to get the top customer which making the most purchases"))

This code will create a new agent that uses the GPT-3.5-turbo model and has a confidence threshold of 0.8.

Next Steps

Now that you have successfully set up Aita, you can start exploring more features and functionalities. Check out the documentation for more information.

Last updated