aita 0.1.2

AI Powered Data Platform

aita

AI Powered Data Agent, a comprehensive solution for data analysis, engineering, and visualization.

Tech Stack

Our platform leverages a combination of cutting-edge technologies and frameworks:

• Langchain: Facilitates the seamless integration of language models into application workflows, significantly enhancing AI interaction capabilities.
• Reflex: An open-source framework for quickly building beautiful, interactive web applications in pure Python
• Apache Arrow: 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: A JupyterLab extension that provides a set of magics for working with AI models.

Supported Models

Model provider are supported by jupyter ai magics.

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

• Snowflake
• Sqlite
• BigQuery
• Postgres
• MySQL
• Redshift
• DynamoDB
• File (CSV, Excel, Parquet, etc.)

Supported Engines

• Apache Spark
• Apache Flink
• Presto
• Ray.io: A distributed computing framework that efficiently scales AI tasks and data processing across clusters, improving performance and resource utilization.

🛡 License

This project is licensed under the terms of the Apache Software License 2.0 license. See LICENSE for more details.