A lightweight Python SDK for building AI agents with unified LLM providers, ReAct orchestration, and tool calling
Project description
miiflow-agent
A lightweight, unified Python SDK for LLM providers with built-in agentic patterns
miiflow-agent gives you a unified API across LLM providers, with built-in support for ReAct agents, tool calling, and streaming — all in ~15K lines of focused code.
from miiflow_agent import LLMClient, Message
# Same interface for any provider
client = LLMClient.create("openai", model="gpt-4o-mini")
response = client.chat([Message.user("Hello!")])
# Switch providers with one line
client = LLMClient.create("anthropic", model="claude-sonnet-4-20250514")
Demo of an Agentic Run
https://github.com/user-attachments/assets/0b5c870a-f9b2-4d55-a829-9d7c000be907
Why miiflow-agent?
| miiflow-agent | LangChain | LiteLLM | |
|---|---|---|---|
| Codebase size | ~15K lines | ~500K lines | ~50K lines |
| Dependencies | 8 core | 50+ | 20+ |
| Built-in agents | ReAct + sub-agent hand-off | Requires setup | None |
| Tool system | @tool decorator | Chains | None |
| Learning curve | Hours | Weeks | Hours |
| Type safety | Full generics | Partial | Basic |
The LangChain Problem
LangChain is powerful but complex. For production apps, you often fight its abstractions more than use them. miiflow-agent gives you what you actually need:
- Unified provider interface — swap OpenAI → Claude → Gemini with one line
- Agentic patterns built-in — a single ReAct loop with emergent planning and multi-agent hand-off, not bolted on
- Simple tool system — decorate any function with
@tool - Real streaming — event-based, not just token callbacks
- Type-safe — full generics, proper error types
The LiteLLM Gap
LiteLLM unifies provider APIs but stops there. miiflow-agent adds:
- ReAct agents with multi-hop reasoning
- Sub-agent hand-off for complex multi-step tasks
- Tool calling with automatic schema generation
- Context injection (Pydantic AI compatible)
Installation
pip install miiflow-agent
# With optional providers
pip install miiflow-agent[groq,google]
# Everything
pip install miiflow-agent[all]
Quick Start
Basic Chat
from miiflow_agent import LLMClient, Message
client = LLMClient.create("openai", model="gpt-4o-mini")
response = client.chat([
Message.system("You are a helpful assistant."),
Message.user("What is Python?")
])
print(response.message.content)
Streaming
async for chunk in client.astream_chat([Message.user("Tell me a story")]):
print(chunk.delta, end="", flush=True)
ReAct Agent with Tools
from miiflow_agent import Agent, AgentType, LLMClient, tool
@tool("calculate", "Evaluate mathematical expressions")
def calculate(expression: str) -> str:
return str(eval(expression))
@tool("search", "Search for information")
def search(query: str) -> str:
return f"Results for '{query}': ..."
# Create agent
agent = Agent(
LLMClient.create("openai", model="gpt-4o"),
agent_type=AgentType.REACT,
max_iterations=10
)
agent.add_tool(calculate)
agent.add_tool(search)
# Run with automatic reasoning
result = await agent.run("What is 25 * 4 + the population of France?")
print(result.data) # Agent reasons, calls tools, synthesizes answer
Context Injection (Pydantic AI Style)
from dataclasses import dataclass
from miiflow_agent import Agent, RunContext, tool
@dataclass
class UserContext:
user_id: str
permissions: list[str]
@tool("get_user_data")
def get_user_data(ctx: RunContext[UserContext], field: str) -> str:
"""Fetch data for the current user."""
if "read" not in ctx.deps.permissions:
return "Permission denied"
return f"User {ctx.deps.user_id} data for {field}"
agent = Agent(client, deps_type=UserContext)
agent.add_tool(get_user_data)
result = await agent.run(
"What's my account status?",
deps=UserContext(user_id="alice", permissions=["read"])
)
Architecture
┌─────────────────────────────────────────────────────────────────┐
│ Your Application │
└─────────────────────────────┬───────────────────────────────────┘
│
┌─────────────────────────────▼───────────────────────────────────┐
│ LLMClient │
│ • Unified interface for all providers │
│ • Automatic tool schema generation │
│ • Metrics collection & observability │
└─────────────────────────────┬───────────────────────────────────┘
│
┌─────────────────────┼─────────────────────┐
│ │ │
▼ ▼ ▼
┌───────────────┐ ┌───────────────┐ ┌───────────────┐
│ Agent │ │ Provider │ │ Tools │
│ │ │ Clients │ │ │
│ • SINGLE_HOP │ │ │ │ • @tool │
│ • REACT │ │ • OpenAI │ │ • FunctionTool│
│ • SubAgents │ │ • Anthropic │ │ • HTTPTool │
│ │ │ • Gemini │ │ • Registry │
│ ┌───────────┐ │ │ • More... │ │ │
│ │Orchestrator│ │ │ │ │ ┌───────────┐ │
│ │ • plan │ │ │ │ │ │ Schemas │ │
│ │ • handoff │ │ │ │ │ │ • Auto-gen│ │
│ └───────────┘ │ │ │ │ │ • Validate│ │
└───────────────┘ │ │ │ └───────────┘ │
└───────────────┘ └───────────────┘
│
▼
┌───────────────┐
│ Message │
│ Unified │
│ Format │
│ │
│ • Text │
│ • Images │
│ • Tool calls │
└───────────────┘
Supported Providers
| Provider | Streaming | Tool Calling | Vision | Status |
|---|---|---|---|---|
| OpenAI | ✅ | ✅ | ✅ | Stable |
| Anthropic | ✅ | ✅ | ✅ | Stable |
| Google Gemini | ✅ | ✅ | ✅ | Stable |
| Groq | ✅ | ✅ | - | Beta |
| Amazon Bedrock | ✅ | ✅ | ✅ | Beta |
| Mistral | ✅ | ✅ | - | Beta |
| OpenRouter | ✅ | ✅ | ✅ | Beta |
| Ollama | ✅ | ✅ | - | Beta |
| xAI | ✅ | ✅ | - | Beta |
Stable providers are production-tested with full feature support. Beta providers are functional but may have edge cases.
Agentic Patterns
miiflow-agent runs a single, unified ReAct loop. Each turn the model emits exactly one of: tool calls (the loop continues) or a text answer (the loop exits). Planning and parallel multi-agent execution are emergent behaviors inside this one loop — there are no separate "plan & execute" or "multi-agent" orchestrators to choose between. Complex tasks are handled by letting the model plan over multiple turns and dispatch work to sub-agents as normal tool calls.
Migrating from a pre-1.8 release? The standalone
PlanAndExecuteOrchestrator,MultiAgentOrchestrator, and theAgentType.PLAN_AND_EXECUTE/PARALLEL_PLAN/MULTI_AGENTenum values have been removed. Express the same outcomes with a singleAgentplussub_agents=[SubAgent(...)]— see Sub-agent hand-off below.AgentType.REACT(the default) andAgentType.SINGLE_HOPare the only modes.
ReAct (Reasoning + Acting)
The agent thinks step-by-step, deciding when to use tools, and plans across turns for multi-step tasks:
agent = Agent(client, agent_type=AgentType.REACT)
# Agent internally:
# Thought: I need to search for this information
# Action: search("topic")
# Observation: Results...
# Thought: Now I can answer
# Final Answer: ...
Sub-agent hand-off
For complex work that benefits from specialization, give the agent one or more
sub-agents. The parent's ReAct loop decides when to hand off, and the dispatch
happens as an ordinary tool call — no separate orchestrator. Sub-agents are wired
in through AgentConfig; the framework synthesizes a dispatch_assistant-shaped
tool that routes calls through the dispatch lifecycle (depth, cycle, budget, and
event bubbling):
from miiflow_agent import Agent
from miiflow_agent.core.config import AgentConfig
# `sub_agents` holds objects implementing the SubAgent protocol — each one a
# parent-side "edge" to a specialist child, carrying per-edge policy such as
# `when_to_use`, `handoff_schema`, and `clarification_policy`.
lead = Agent(config=AgentConfig(
client=client,
system_prompt="Coordinate research and writing.",
sub_agents=[researcher_subagent, writer_subagent],
))
result = await lead.run(
"Research Python web frameworks, compare them, and write a summary"
)
# The ReAct loop plans, dispatches to each specialist, and synthesizes the result.
See
examples/subagents.pyfor a complete, runnable hand-off example, andmiiflow_agent/core/subagent.pyfor theSubAgentprotocol and its per-edge policy fields.
Event Streaming
Stream real-time events during agent execution:
from miiflow_agent import Agent, AgentType, RunContext
from miiflow_agent.core.react import ReActEventType
agent = Agent(client, agent_type=AgentType.REACT)
context = RunContext(deps=None)
async for event in agent.stream_react("What is 2+2?", context):
match event.event_type:
case ReActEventType.THINKING_CHUNK:
print(event.data.get("delta", ""), end="")
case ReActEventType.TOOL_START:
print(f"\nCalling: {event.data['tool_name']}")
case ReActEventType.OBSERVATION:
print(f"Result: {event.data['observation']}")
case ReActEventType.FINAL_ANSWER:
print(f"\nAnswer: {event.data['answer']}")
Observability
Built-in Phoenix tracing support:
from miiflow_agent.core import setup_tracing
setup_tracing(phoenix_endpoint="http://localhost:6006")
# All LLM calls are now traced
Error Handling
Comprehensive error hierarchy:
from miiflow_agent import (
MiiflowLLMError, # Base
ProviderError, # Provider-specific
RateLimitError, # Rate limited
AuthenticationError, # Invalid API key
TimeoutError, # Request timeout
ToolError, # Tool execution failed
)
try:
response = client.chat(messages)
except RateLimitError as e:
print(f"Rate limited, retry after {e.retry_after}s")
except AuthenticationError:
print("Check your API key")
except ProviderError as e:
print(f"{e.provider} error: {e.message}")
Error Handling
- Quickstart Guide — Get started in 5 minutes
- API Reference — Complete API documentation
- Tool Tutorial — Build custom tools
- Agent Tutorial — Build ReAct agents
- Provider Guide — Provider-specific configuration
- Observability — Tracing and debugging
Contributing
We welcome contributions! Here's how to get started:
# Clone and install
git clone https://github.com/Miiflow/miiflow-agent.git
cd miiflow-agent
pip install -e ".[all]"
# Run tests
pytest tests/
# Format code
black miiflow_agent/ tests/
isort miiflow_agent/ tests/
Ways to Contribute
- Report bugs — Open an issue with reproduction steps
- Request features — Describe your use case
- Add providers — See CONTRIBUTING.md for the provider guide
- Improve docs — Fix typos, add examples
- Write tests — Increase coverage
See CONTRIBUTING.md for detailed guidelines.
License
MIT License - see LICENSE for details.
Project details
Release history Release notifications | RSS feed
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 miiflow_agent-1.11.0.tar.gz.
File metadata
- Download URL: miiflow_agent-1.11.0.tar.gz
- Upload date:
- Size: 293.5 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: poetry/2.4.1 CPython/3.11.15 Linux/6.17.0-1018-azure
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
31bc0ccf1e2a6e19030265e0800ac78ac34f23b91c8b35979b227d9c3897ee60
|
|
| MD5 |
da8513f485f1f63afc687fc471a78160
|
|
| BLAKE2b-256 |
01a3be5f42c5bb18b2840cb0a2438bc6041cb6f8ef08147c86fa007d0052b6cc
|
File details
Details for the file miiflow_agent-1.11.0-py3-none-any.whl.
File metadata
- Download URL: miiflow_agent-1.11.0-py3-none-any.whl
- Upload date:
- Size: 349.1 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: poetry/2.4.1 CPython/3.11.15 Linux/6.17.0-1018-azure
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
21fdb28df677cfaec261a3b1b86ab3e7e6678d8cab74daff6e2d636e5f528e5d
|
|
| MD5 |
366a26b05e30ee0b296c1c9263345ff9
|
|
| BLAKE2b-256 |
1ab1d74186cf7db01892c674049ed7a5f19eddb33a759972afdae799d5169cc5
|