Aita
  • Introduction
  • Roadmap
  • Aita Lab
    • Chat
    • Data Source
  • Architecture
  • Contribution
  • Getting Started
    • Quick Start
  • Installation
  • Reference
    • Aita agent
      • Pandas Agent
      • Pyarrow Agent
      • Python Agent
      • Spark Agent
      • SqlAgent
    • Data Sources
      • sql
      • File Data Source
      • Postgres Data Source
      • Snowflake Data Source
    • Aita tool
      • PythonTool
      • PandasTool
      • PyArrow Tool
      • pyspark tool
      • Sql Tools
  • Tech specs
    • Aita tech spec v1
Powered by GitBook
On this page
  • Prerequisites
  • Installation
  • Getting Started
  • Next Steps
  1. Getting Started

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

PreviousGetting StartedNextInstallation

Last updated 11 months ago

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

documentation