aios-kernel-sdk 0.1.0

Python SDK for the AIOS AI Kernel — the operating system layer for AI agents

aios-kernel-sdk

Python SDK for the AIOS AI Kernel — the operating system layer for AI agents.

Installation

pip install aios-kernel-sdk

# With LangGraph adapter:
pip install "aios-kernel-sdk[langgraph]"

Quickstart

import asyncio
from aios_sdk import AIKernel, LLMQuery

async def main():
    async with AIKernel("http://localhost:8000", "my-agent") as kernel:
        response = await kernel.llm(LLMQuery(
            model="anthropic/claude-haiku-4-5-20251001",
            messages=[{"role": "user", "content": "Summarise the Q4 report"}],
        ))
        print(response["content"])

asyncio.run(main())

Start the kernel first:

git clone https://github.com/stagsz/aios && cd aios
docker compose up -d

Features

• Async-first client with graceful fallback to litellm on kernel unavailability
• Typed Pydantic v2 query models
• LangGraph checkpointer adapter (M3 stub — full implementation post-M1)
• OpenAI-compatible adapter stub
• Prompt-injection sanitization on all tool output (M6)

Error handling

from aios_sdk import KernelError

try:
    response = await kernel.llm(query)
except KernelError as e:
    print(f"Kernel error {e.status_code}: {e.detail}")

When the kernel is unreachable and fallback_to_litellm=True (default), the SDK calls litellm directly — no code change required.

Links

• Main repo
• Docs
• Issue tracker