pip install praisonaiagents

from praisonaiagents import Agent
from praisonaiagents.thinking import ThinkingBudget

# Create a thinking budget
budget = ThinkingBudget(
    max_tokens=16000,
    max_time_seconds=60,
    adaptive=True
)

# Apply to agent
agent = Agent(
    name="Reasoner",
    instructions="You are a reasoning assistant.",
)
# Set thinking budget via property
agent.thinking_budget = budget

from praisonaiagents.thinking import ThinkingBudget

budget = ThinkingBudget(
    max_tokens=16000,
    max_time_seconds=60,
    adaptive=True
)

from praisonaiagents.thinking import ThinkingConfig

config = ThinkingConfig(
    enabled=True,
    budget=budget,
    show_thinking=False
)

from praisonaiagents.thinking import ThinkingUsage

usage = ThinkingUsage(
    tokens_used=5000,
    time_seconds=15.5
)

from praisonaiagents.thinking import ThinkingTracker

tracker = ThinkingTracker()
tracker.record(usage)
print(tracker.total_tokens)
print(tracker.total_time)

from praisonaiagents import Agent
from praisonaiagents.thinking import ThinkingBudget

agent = Agent(
    name="Analyst",
    instructions="You are an analytical assistant.",
)
agent.thinking_budget = ThinkingBudget(max_tokens=16000)

# Agent will use extended thinking for complex tasks
result = agent.chat("Analyze the implications of quantum computing on cryptography")

from praisonaiagents.thinking import ThinkingBudget

# Budget adapts based on task complexity
budget = ThinkingBudget(
    max_tokens=32000,
    adaptive=True  # Automatically adjusts based on task
)

agent = Agent(
    name="Researcher",
    instructions="You are a research assistant.",
)
agent.thinking_budget = budget

from praisonaiagents.thinking import ThinkingBudget

# Limit thinking time for faster responses
budget = ThinkingBudget(
    max_tokens=16000,
    max_time_seconds=30  # Stop thinking after 30 seconds
)

from praisonaiagents.thinking import ThinkingTracker

tracker = ThinkingTracker()

# After agent runs
usage = agent.get_thinking_usage()
tracker.record(usage)

print(f"Total thinking tokens: {tracker.total_tokens}")
print(f"Total thinking time: {tracker.total_time}s")