# Aita
AI Powered Data Agent, a comprehensive solution for data analysis, engineering, and visualization.
[](https://github.com/project-aita/aita/actions/workflows/build.yml?query=workflow%3Abuild)
[](https://pypi.org/project/aita/)
[](https://github.com/project-aita/aita/pulls?utf8=%E2%9C%93&q=is%3Apr%20author%3Aapp%2Fdependabot)
[](https://github.com/psf/black)
[](https://github.com/PyCQA/bandit)
[](https://github.com/project-aita/aita/blob/main/.pre-commit-config.yaml)
[](https://github.com/project-aita/aita/blob/main/LICENSE)
[](https://github.com/project-aita/aita/blob/main/assets/images/coverage.svg)
## Tech Stack
Our platform leverages a combination of cutting-edge technologies and frameworks:
- **[Langchain](https://www.langchain.com/)**: Facilitates the seamless integration of language models into application workflows, significantly enhancing AI interaction capabilities.
- **[Reflex](https://reflex.dev/)**: An open-source framework for quickly building beautiful, interactive web applications in pure Python
- **[Apache Arrow](https://arrow.apache.org/)**: A cross-language development platform for in-memory data that specifies a standardized language-independent columnar memory format for flat and hierarchical data, organized for efficient analytic operations on modern hardware like CPUs and GPUs.
- **[Jupyter Ai Magics](https://github.com/jupyterlab/jupyter-ai)**: A JupyterLab extension that provides a set of magics for working with AI models.
## Installation
Simply install the package using pip:
```shell
pip install aita
```
Or with extra dependencies:
```shell
pip install aita[snowflake]
```
## Example
Below is an example of using SqlAgent to connect to a PostgreSQL database and ask a question.
You can read more details in the [documentation](https://aita-1.gitbook.io/aita/).
```python
from aita.agent.sql import SqlAgent
from aita.datasource.postgresql import PostgreSqlDataSource
# Connect to a PostgreSQL database
datasource = PostgreSqlDataSource("postgresql://user:password@localhost:5432/dbname")
# Create a SQL agent
sql_agent = SqlAgent("gpt-3.5-turbo").add_datasource(datasource)
# ask question to the agent
sql_agent.stream("What is the total number of orders?")
```
## Supported Models
Model provider are supported by jupyter ai magics. Ensure the corresponding environment variables are set before using the Aita agent.
| Provider | Provider ID | Environment variable(s) | Python package(s) |
|---------------------|----------------------|----------------------------|---------------------------------|
| AI21 | `ai21` | `AI21_API_KEY` | `ai21` |
| Anthropic | `anthropic` | `ANTHROPIC_API_KEY` | `langchain-anthropic` |
| Anthropic (chat) | `anthropic-chat` | `ANTHROPIC_API_KEY` | `langchain-anthropic` |
| Bedrock | `bedrock` | N/A | `boto3` |
| Bedrock (chat) | `bedrock-chat` | N/A | `boto3` |
| Cohere | `cohere` | `COHERE_API_KEY` | `cohere` |
| ERNIE-Bot | `qianfan` | `QIANFAN_AK`, `QIANFAN_SK` | `qianfan` |
| Gemini | `gemini` | `GOOGLE_API_KEY` | `langchain-google-genai` |
| GPT4All | `gpt4all` | N/A | `gpt4all` |
| Hugging Face Hub | `huggingface_hub` | `HUGGINGFACEHUB_API_TOKEN` | `huggingface_hub`, `ipywidgets`, `pillow` |
| NVIDIA | `nvidia-chat` | `NVIDIA_API_KEY` | `langchain_nvidia_ai_endpoints` |
| OpenAI | `openai` | `OPENAI_API_KEY` | `langchain-openai` |
| OpenAI (chat) | `openai-chat` | `OPENAI_API_KEY` | `langchain-openai` |
| SageMaker | `sagemaker-endpoint` | N/A | `boto3` |
## Supported Data Sources
- [x] Snowflake
- [x] Sqlite
- [ ] BigQuery
- [x] Postgres
- [ ] MySQL
- [ ] Redshift
- [ ] DynamoDB
- [x] File (CSV, Excel, Parquet, etc.)
## Supported Engines
- [x] Apache Spark
- [ ] Apache Flink
- [ ] Presto
- [ ] **[Ray.io](https://ray.io/)**: A distributed computing framework that efficiently scales AI tasks and data processing across clusters, improving performance and resource utilization.
## 🛡 License
[](https://github.com/project-aita/aita/blob/main/LICENSE)
This project is licensed under the terms of the `Apache Software License 2.0` license. See [LICENSE](https://github.com/aita/aita/blob/master/LICENSE) for more details.
Raw data
{
"_id": null,
"home_page": "https://www.project-aita.com",
"name": "aita",
"maintainer": null,
"docs_url": null,
"requires_python": "<4.0,>=3.9",
"maintainer_email": null,
"keywords": "Artificial Intelligence, Data Analysis, Code Generation",
"author": "aita",
"author_email": "contact@project-aita.com",
"download_url": "https://files.pythonhosted.org/packages/c1/18/8e4300b6bc60afd7918a432a515f99fd604617c19d378b9a4a8254c4f08a/aita-0.1.5.tar.gz",
"platform": null,
"description": "# Aita\nAI Powered Data Agent, a comprehensive solution for data analysis, engineering, and visualization. \n\n\n[](https://github.com/project-aita/aita/actions/workflows/build.yml?query=workflow%3Abuild)\n[](https://pypi.org/project/aita/)\n[](https://github.com/project-aita/aita/pulls?utf8=%E2%9C%93&q=is%3Apr%20author%3Aapp%2Fdependabot)\n\n[](https://github.com/psf/black)\n[](https://github.com/PyCQA/bandit)\n[](https://github.com/project-aita/aita/blob/main/.pre-commit-config.yaml)\n[](https://github.com/project-aita/aita/blob/main/LICENSE)\n[](https://github.com/project-aita/aita/blob/main/assets/images/coverage.svg)\n\n\n## Tech Stack\n\nOur platform leverages a combination of cutting-edge technologies and frameworks:\n\n- **[Langchain](https://www.langchain.com/)**: Facilitates the seamless integration of language models into application workflows, significantly enhancing AI interaction capabilities.\n- **[Reflex](https://reflex.dev/)**: An open-source framework for quickly building beautiful, interactive web applications in pure Python\n- **[Apache Arrow](https://arrow.apache.org/)**: A cross-language development platform for in-memory data that specifies a standardized language-independent columnar memory format for flat and hierarchical data, organized for efficient analytic operations on modern hardware like CPUs and GPUs.\n- **[Jupyter Ai Magics](https://github.com/jupyterlab/jupyter-ai)**: A JupyterLab extension that provides a set of magics for working with AI models.\n\n## Installation\nSimply install the package using pip:\n\n```shell\npip install aita\n\n```\nOr with extra dependencies:\n\n```shell\npip install aita[snowflake]\n```\n\n## Example\nBelow is an example of using SqlAgent to connect to a PostgreSQL database and ask a question.\nYou can read more details in the [documentation](https://aita-1.gitbook.io/aita/).\n\n```python\nfrom aita.agent.sql import SqlAgent\nfrom aita.datasource.postgresql import PostgreSqlDataSource\n\n# Connect to a PostgreSQL database\ndatasource = PostgreSqlDataSource(\"postgresql://user:password@localhost:5432/dbname\")\n\n# Create a SQL agent\nsql_agent = SqlAgent(\"gpt-3.5-turbo\").add_datasource(datasource)\n\n# ask question to the agent\nsql_agent.stream(\"What is the total number of orders?\")\n```\n\n## Supported Models\nModel provider are supported by jupyter ai magics. Ensure the corresponding environment variables are set before using the Aita agent.\n\n| Provider | Provider ID | Environment variable(s) | Python package(s) |\n|---------------------|----------------------|----------------------------|---------------------------------|\n| AI21 | `ai21` | `AI21_API_KEY` | `ai21` |\n| Anthropic | `anthropic` | `ANTHROPIC_API_KEY` | `langchain-anthropic` |\n| Anthropic (chat) | `anthropic-chat` | `ANTHROPIC_API_KEY` | `langchain-anthropic` |\n| Bedrock | `bedrock` | N/A | `boto3` |\n| Bedrock (chat) | `bedrock-chat` | N/A | `boto3` |\n| Cohere | `cohere` | `COHERE_API_KEY` | `cohere` |\n| ERNIE-Bot | `qianfan` | `QIANFAN_AK`, `QIANFAN_SK` | `qianfan` |\n| Gemini | `gemini` | `GOOGLE_API_KEY` | `langchain-google-genai` |\n| GPT4All | `gpt4all` | N/A | `gpt4all` |\n| Hugging Face Hub | `huggingface_hub` | `HUGGINGFACEHUB_API_TOKEN` | `huggingface_hub`, `ipywidgets`, `pillow` |\n| NVIDIA | `nvidia-chat` | `NVIDIA_API_KEY` | `langchain_nvidia_ai_endpoints` |\n| OpenAI | `openai` | `OPENAI_API_KEY` | `langchain-openai` |\n| OpenAI (chat) | `openai-chat` | `OPENAI_API_KEY` | `langchain-openai` |\n| SageMaker | `sagemaker-endpoint` | N/A | `boto3` |\n\n## Supported Data Sources\n- [x] Snowflake\n- [x] Sqlite\n- [ ] BigQuery\n- [x] Postgres\n- [ ] MySQL\n- [ ] Redshift\n- [ ] DynamoDB\n- [x] File (CSV, Excel, Parquet, etc.)\n\n## Supported Engines\n- [x] Apache Spark\n- [ ] Apache Flink\n- [ ] Presto\n- [ ] **[Ray.io](https://ray.io/)**: A distributed computing framework that efficiently scales AI tasks and data processing across clusters, improving performance and resource utilization.\n\n## \ud83d\udee1 License\n\n[](https://github.com/project-aita/aita/blob/main/LICENSE)\n\nThis project is licensed under the terms of the `Apache Software License 2.0` license. See [LICENSE](https://github.com/aita/aita/blob/master/LICENSE) for more details.\n",
"bugtrack_url": null,
"license": "Apache Software License 2.0",
"summary": "AI Powered Data Platform",
"version": "0.1.5",
"project_urls": {
"Homepage": "https://www.project-aita.com",
"Repository": "https://github.com/project-aita/aita"
},
"split_keywords": [
"artificial intelligence",
" data analysis",
" code generation"
],
"urls": [
{
"comment_text": "",
"digests": {
"blake2b_256": "269775135e1a192a0187359d68f1a76eb71b62d895dc4be6c6009a575132a075",
"md5": "03a56a62aca255b528c381bc4d07fa9a",
"sha256": "b5760d473c4243b74b1357ecfa0c31c08f0104bdc1adef7006346e8cf920f625"
},
"downloads": -1,
"filename": "aita-0.1.5-py3-none-any.whl",
"has_sig": false,
"md5_digest": "03a56a62aca255b528c381bc4d07fa9a",
"packagetype": "bdist_wheel",
"python_version": "py3",
"requires_python": "<4.0,>=3.9",
"size": 88430,
"upload_time": "2024-07-23T22:32:07",
"upload_time_iso_8601": "2024-07-23T22:32:07.115173Z",
"url": "https://files.pythonhosted.org/packages/26/97/75135e1a192a0187359d68f1a76eb71b62d895dc4be6c6009a575132a075/aita-0.1.5-py3-none-any.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "c1188e4300b6bc60afd7918a432a515f99fd604617c19d378b9a4a8254c4f08a",
"md5": "96730dbd25f23bac76f034d4ce8967b9",
"sha256": "5c3ce3b890d2534976643d81cf0f48e0463698392a123700d8c98a3ada481dcf"
},
"downloads": -1,
"filename": "aita-0.1.5.tar.gz",
"has_sig": false,
"md5_digest": "96730dbd25f23bac76f034d4ce8967b9",
"packagetype": "sdist",
"python_version": "source",
"requires_python": "<4.0,>=3.9",
"size": 64808,
"upload_time": "2024-07-23T22:32:08",
"upload_time_iso_8601": "2024-07-23T22:32:08.482115Z",
"url": "https://files.pythonhosted.org/packages/c1/18/8e4300b6bc60afd7918a432a515f99fd604617c19d378b9a4a8254c4f08a/aita-0.1.5.tar.gz",
"yanked": false,
"yanked_reason": null
}
],
"upload_time": "2024-07-23 22:32:08",
"github": true,
"gitlab": false,
"bitbucket": false,
"codeberg": false,
"github_user": "project-aita",
"github_project": "aita",
"travis_ci": false,
"coveralls": false,
"github_actions": true,
"lcname": "aita"
}
JSON
