Official Python SDK for building powerful AI agents with access to 50+ built-in tools, 300+ integrations, and all major LLMs—all through a single, unified platform.
Project description
BaseAI Python SDK
One SDK. Unlimited Tools. Build AI Agents in Minutes.
Official Python SDK for building powerful AI agents with access to 50+ built-in tools, 300+ integrations, and all major LLMs—all through a single, unified platform.
🎥 Watch the Demo
🚀 Why BaseAI?
Stop juggling multiple tools, APIs, and accounts. BaseAI consolidates everything you need into one simple SDK.
The Problem You're Solving
Building AI agents today means:
- ❌ Managing 10+ different API keys and accounts
- ❌ Integrating multiple libraries and SDKs
- ❌ Searching for the right tools and benchmarking
- ❌ Building custom integrations from scratch
- ❌ Dealing with authentication and security concerns
- ❌ Paying multiple subscription fees
The BaseAI Solution
One SDK. One API key. Everything you need.
- ✅ 50+ built-in tools - Everything from web search to image editing to web development
- ✅ 300+ MCP integrations - Access 200+ Composio integrations (Gmail, Slack, GitHub, etc.) plus custom MCP servers
- ✅ All LLMs in one place - Unified access to multiple language models
- ✅ Isolated sandboxes - Secure execution environments built-in
- ✅ Pay-as-you-go pricing - 80% cheaper than alternatives
- ✅ Type-safe SDK - Full IDE support for Python
Install one SDK. Access everything. Build a working prototype in minutes.
💰 Pricing
Pay-As-You-Go Model
BaseAI uses transparent, pay-as-you-go pricing with no hidden fees.
80% Cheaper Than Alternatives
Unlike traditional platforms that require:
- Multiple subscription fees for different services
- Per-user licensing costs
- Minimum monthly commitments
- Platform lock-in fees
BaseAI charges you only for what you use, when you use it. No minimums. Just simple, transparent pricing that scales with your needs.
What You Get
- Access to all tools - No per-tool pricing
- All LLMs - Unified access without separate accounts
- 300+ integrations - No per-integration fees
- Isolated sandboxes - Included at no extra cost
- Full SDK support - Python SDK included
Get started today: Get your API key
Installation
pip install a2abase-sdk
📦 Published on PyPI: https://pypi.org/project/a2abase-sdk/
Or install directly from GitHub:
pip install git+https://github.com/A2ABaseAI/sdks.git#subdirectory=python
Requirements
- Python 3.11+
- See requirements.txt for dependencies
🚀 Quick Start
import asyncio
import os
from baseai import BaseAI
from baseai.tools import BaseAITool
async def main():
api_key = os.getenv("BASEAI_API_KEY")
if not api_key:
raise ValueError("Please set BASEAI_API_KEY environment variable")
client = BaseAI(api_key=api_key, api_url="https://a2abase.ai/api")
thread = await client.Thread.create()
agent = await client.Agent.create(
name="My Assistant",
system_prompt="You are a helpful AI assistant.",
mcp_tools=[BaseAITool.WEB_SEARCH_TOOL],
)
run = await agent.run("Hello, how are you?", thread)
stream = await run.get_stream()
async for chunk in stream:
print(chunk, end="")
asyncio.run(main())
🔑 Getting Your API Key
- Sign up at BaseAI
- Get your API key from the dashboard
- Set it as an environment variable:
export BASEAI_API_KEY="pk_xxx:sk_xxx"
That's it! You're ready to build.
🛠️ Available Tools
The SDK provides access to the same comprehensive set of tools through the BaseAITool enum:
File Management
FILES_TOOL- Read, write, and edit files in the sandboxUPLOAD_FILE_TOOL- Upload files to the sandbox workspace
Development & Automation
SHELL_TOOL- Execute shell commands in isolated sandboxesWEB_DEV_TOOL- Create and manage modern web applications (Next.js, React, Vite, shadcn/ui)DEPLOY_TOOL- Deploy web applications to productionEXPOSE_TOOL- Expose local services to the internet- Project templates - Scaffold projects from predefined templates
Image & Vision
VISION_TOOL- Analyze and understand images with AIIMAGE_SEARCH_TOOL- Search for images on the webIMAGE_EDIT_TOOL- Edit and manipulate imagesDESIGN_TOOL- Design and create visual content
Content Creation
DOCS_TOOL- Create and manage documents (HTML, Markdown formats)SHEETS_TOOL- Create and manage spreadsheets with data analysis, visualization, and formattingPRESENTATION_TOOL- Create and manage presentationsPRESENTATION_OUTLINE_TOOL- Create presentation outlines
Search & Browser
WEB_SEARCH_TOOL- Search the web for informationBROWSER_TOOL- Browse websites and interact with web pages (navigate, click, fill forms, scroll)
Research & Intelligence
- People search - Search for people and enrich profiles
- Company search - Search for companies and business information
- Paper search - Search academic papers and research documents
- Web search - General web search capabilities
Knowledge & Data
KB_TOOL- Access and manage knowledge base (local and global sync)DATA_PROVIDERS_TOOL- Access structured data from providers:- LinkedIn - Professional network data
- Yahoo Finance - Financial market data
- Amazon - Product and marketplace data
- Zillow - Real estate data
- Twitter - Social media data
- ActiveJobs - Job listings data
Agent Management
- Agent creation - Create and configure AI agents
- Agent configuration - Manage agent settings and versions
- Workflow management - Create and manage agent workflows
- Trigger management - Set up automated triggers
- Credential profiles - Manage API credentials
Task Management
- Task list - Create, update, and manage tasks organized by sections
- Task tracking - Track completion status and progress
- Batch operations - Manage multiple tasks at once
Communication
- Message tool - Ask questions, inform users, and mark completion
- Message expansion - Expand truncated messages from previous conversations
Automation
- Computer use - Control sandbox browser and GUI (mouse, keyboard, screenshots)
- Document parser - Parse and extract data from documents
Templates & Scaffolding
- Project templates - Generate projects from predefined templates
- Template search - Search and discover available templates
🔌 MCP Integrations
BaseAI supports 300+ MCP (Model Context Protocol) integrations through Composio and custom MCP servers, allowing you to connect any MCP-compatible server. MCPs can be connected via:
- HTTP/HTTPS endpoints
- SSE (Server-Sent Events)
- stdio (standard input/output)
Composio MCP Integrations (200+)
BaseAI integrates with Composio.dev to provide access to 200+ pre-configured MCP servers including:
Productivity & Communication
- Gmail, Google Calendar, Slack, Microsoft Teams, Notion, Linear, Asana, Jira, Trello, Airtable
Code & Development
- GitHub, GitLab, Bitbucket, Docker Hub, AWS, Google Cloud Platform, Azure
CRM & Sales
- Salesforce, HubSpot, Pipedrive, Zoho CRM, Intercom
Data & Analytics
- Google Sheets, Google Drive, Dropbox, Airtable, MongoDB, PostgreSQL, MySQL
Marketing & Social
- Twitter/X, LinkedIn, Facebook, Instagram, Mailchimp, SendGrid
E-commerce & Payments
- Shopify, Stripe, PayPal, WooCommerce, Square
And 150+ more integrations including project management tools, databases, cloud services, APIs, and business applications.
Custom MCP Servers
You can also connect any custom MCP-compatible server by providing the MCP endpoint URL. This enables integration with any tool, API, or service that follows the Model Context Protocol standard.
Usage
Creating an Agent
from baseai import BaseAI
from baseai.tools import BaseAITool
client = BaseAI(api_key="your-api-key", api_url="https://a2abase.ai/api")
# Create an agent with tools
agent = await client.Agent.create(
name="My Agent",
system_prompt="You are a helpful assistant.",
mcp_tools=[BaseAITool.WEB_SEARCH_TOOL, BaseAITool.FILES_TOOL],
)
Running an Agent
# Create a thread
thread = await client.Thread.create()
# Run the agent
run = await agent.run("Your task here", thread)
# Stream the response
stream = await run.get_stream()
async for chunk in stream:
print(chunk, end="")
Finding an Existing Agent
# Find agent by name
agent = await client.Agent.find_by_name("My Agent")
if agent:
# Use the existing agent
pass
📚 Examples
Comprehensive examples are available in the example/ directory, demonstrating:
- Tool-Specific Examples: Each tool from
BaseAIToolenum with practical use cases - Common Use Cases: Real-world scenarios like research, content creation, automation, and more
See the examples README for a complete list of available examples.
Running Examples
Python:
- Run scripts:
cd python && PYTHONPATH=. python3 example/<name>.py - Run in Google Colab: See Running Examples in Google Colab for instructions
Running All Examples
cd python
PYTHONPATH=. python3 example/run_all_examples.py
Running Examples in Google Colab
To run examples in Google Colab:
Option 1: Open from GitHub
- Go to Google Colab
- Click "File" → "Open notebook"
- Select the "GitHub" tab
- Enter:
A2ABaseAI/sdks - Navigate to
python/example/quick_start.ipynbor any.pyfile
Option 2: Manual Setup
- Open Google Colab
- Create a new notebook
- Copy the code from any example file
- Adapt it for notebook use (remove
if __name__ == "__main__"and useawaitdirectly)
Quick Start in Colab:
# Install the SDK
!pip install a2abase-sdk
import os
from baseai import BaseAI
from baseai.tools import BaseAITool
# Set your API key (use Colab's secrets or environment variables)
os.environ['BASEAI_API_KEY'] = 'pk_xxx:sk_xxx'
# Create client
client = BaseAI(api_key=os.getenv("BASEAI_API_KEY"), api_url="https://a2abase.ai/api")
# Create thread and agent
thread = await client.Thread.create()
agent = await client.Agent.create(
name="My Assistant",
system_prompt="You are a helpful AI assistant.",
mcp_tools=[BaseAITool.WEB_SEARCH_TOOL],
)
# Run the agent
run = await agent.run("Hello, how are you?", thread)
stream = await run.get_stream()
# Stream and display results
async for chunk in stream:
print(chunk, end="")
Note: In Google Colab, you can use await directly in cells without asyncio.run().
📖 Documentation
- PyPI Package: https://pypi.org/project/a2abase-sdk/
- GitHub Repository: https://github.com/A2ABaseAI/sdks
- Full Documentation: See the repository for more examples and API reference.
💬 Support
Need help? Join our Discord community for support and discussions:
- Discord: https://discord.gg/qAncfHmYUm
🤝 Contributing
Contributions are welcome! Please see our contributing guidelines.
📄 License
MIT License - see LICENSE file for details.
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file a2abase_sdk-1.0.4.tar.gz.
File metadata
- Download URL: a2abase_sdk-1.0.4.tar.gz
- Upload date:
- Size: 15.6 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.14.0
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
50538cfb759f928ac610b179e032d6a12a0dff9d03c15366e51db5e3f4940939
|
|
| MD5 |
c301ef76c21129d447deb847d181f12a
|
|
| BLAKE2b-256 |
ba6240755b206a6f2a62452fd5e9d3fedccad4268f2f080f7b7b1f4320bcc99d
|
File details
Details for the file a2abase_sdk-1.0.4-py3-none-any.whl.
File metadata
- Download URL: a2abase_sdk-1.0.4-py3-none-any.whl
- Upload date:
- Size: 15.4 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.14.0
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
958c511e15b79aa077c682cfec881f59dc02b8cdbeac13ca80729e37476d6cb2
|
|
| MD5 |
76489af529671f5becbb5b4a97a2b066
|
|
| BLAKE2b-256 |
84a271868f8ff7f06bc8c33e0ace7dc5437e993a153822621d429397c68fba96
|