Skip to main content

A lightweight AI Agent framework

Project description

ThinAgents

A lightweight, pluggable AI Agent framework for Python.
Build LLM-powered agents that can use tools, remember conversations, and connect to external resources with minimal code. ThinAgents leverages litellm under the hood for its language model interactions.

Docs


Installation

pip install thinagents

Basic Usage

Create an agent and interact with an LLM in just a few lines:

from thinagents import Agent

agent = Agent(
    name="Greeting Agent",
    model="openai/gpt-4o-mini",
)

response = await agent.arun("Hello, how are you?")
print(response.content)

Tools

Agents can use Python functions as tools to perform actions or fetch data.

from thinagents import Agent

def get_weather(city: str) -> str:
    return f"The weather in {city} is sunny."

agent = Agent(
    name="Weather Agent",
    model="openai/gpt-4o-mini",
    tools=[get_weather],
)

response = await agent.arun("What is the weather in Tokyo?")
print(response.content)

Tools with Decorator

For richer metadata and parameter validation, use the @tool decorator:

from thinagents import Agent, tool

@tool(name="get_weather")
def get_weather(city: str) -> str:
    """Get the weather for a city."""
    return f"The weather in {city} is sunny."

agent = Agent(
    name="Weather Pro",
    model="openai/gpt-4o-mini",
    tools=[get_weather],
)

You can also use Pydantic models for parameter schemas:

from pydantic import BaseModel, Field
from thinagents import tool

class MultiplyInputSchema(BaseModel):
    a: int = Field(description="First operand")
    b: int = Field(description="Second operand")

@tool(name="multiply_tool", pydantic_schema=MultiplyInputSchema)
def multiply(a: int, b: int) -> int:
    return a * b

Returning Content and Artifact

Sometimes, a tool should return both a summary (for the LLM) and a large artifact (for downstream use):

from thinagents import tool

@tool(return_type="content_and_artifact")
def summarize_and_return_data(query: str) -> tuple[str, dict]:
    data = {"rows": list(range(10000))}
    summary = f"Found {len(data['rows'])} rows for query: {query}"
    return summary, data

response = await agent.arun("Summarize the data for X")
print(response.content)      # Sent to LLM
print(response.artifact)     # Available for downstream use

Async Usage

ThinAgents is async by design. You can stream responses or await the full result:

# Streaming
async for chunk in agent.astream("List files and get weather", conversation_id="1"):
    print(chunk.content, end="", flush=True)

# Or get the full response at once (non-streaming)
response = await agent.arun("List files and get weather", conversation_id="1")
print(response.content)

Memory

Agents can remember previous messages and tool results by attaching a memory backend.

from thinagents.memory import InMemoryStore

agent = Agent(
    name="Memory Demo",
    model="openai/gpt-4o-mini",
    memory=InMemoryStore(),  # Fast, in-memory storage
)

conv_id = "demo-1"
print(await agent.arun("Hi, I'm Alice!", conversation_id=conv_id))
print(await agent.arun("What is my name?", conversation_id=conv_id))
# → "Your name is Alice."

Persistent Memory

from thinagents.memory import FileMemory, SQLiteMemory

file_agent = Agent(
    name="File Mem Agent",
    model="openai/gpt-4o-mini",
    memory=FileMemory(storage_dir="./agent_mem"),
)

db_agent = Agent(
    name="SQLite Mem Agent",
    model="openai/gpt-4o-mini",
    memory=SQLiteMemory(db_path="./agent_mem.db"),
)

Storing Tool Artifacts

Enable artifact storage in memory:

agent = Agent(
    ...,
    memory=InMemoryStore(store_tool_artifacts=True),
)

Model Context Protocol (MCP) Integration

Connect your agent to external resources (files, APIs, etc.) using MCP.

agent = Agent(
    name="MCP Agent",
    model="openai/gpt-4o-mini",
    mcp_servers=[
        {
            "transport": "sse",
            "url": "http://localhost:8100/sse"
        },
        {
            "transport": "stdio",
            "command": "npx",
            "args": [
                "-y",
                "@modelcontextprotocol/server-filesystem",
                "/path/to/dir"
            ]
        },
    ],
)

License

MIT

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

thinagents-0.0.19.tar.gz (3.6 MB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

thinagents-0.0.19-py3-none-any.whl (3.7 MB view details)

Uploaded Python 3

File details

Details for the file thinagents-0.0.19.tar.gz.

File metadata

  • Download URL: thinagents-0.0.19.tar.gz
  • Upload date:
  • Size: 3.6 MB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.13.7

File hashes

Hashes for thinagents-0.0.19.tar.gz
Algorithm Hash digest
SHA256 2ded6235dc7ca42b8b35433ea22be0c449b4585d44c61703df73412c14d89ac8
MD5 7ea8ab594f6d1d95c1deedc403570721
BLAKE2b-256 e71b905d01e52107460f8bdec824956ef46ac4d8b6d36ae8dd3383102ff6557e

See more details on using hashes here.

File details

Details for the file thinagents-0.0.19-py3-none-any.whl.

File metadata

  • Download URL: thinagents-0.0.19-py3-none-any.whl
  • Upload date:
  • Size: 3.7 MB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.13.7

File hashes

Hashes for thinagents-0.0.19-py3-none-any.whl
Algorithm Hash digest
SHA256 4acf3517e5b12cab54ad9b8cda11b79a0b34ce629a05c73ae0de38a99e98c345
MD5 fb7b0ce982229bd037cf7c448b591bdc
BLAKE2b-256 3eb8dc2d4537cae05e523a57d7b56414285b8a27943d63ccd49194824464823c

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page