Prebuilt Agents

from shipit_agent import Agent
from shipit_agent.agents import AgentRegistry

registry = AgentRegistry.default()
agent_def = registry.get("security-auditor")

agent = Agent.with_builtins(
    llm=llm,
    prompt=agent_def.system_prompt(),
    max_iterations=agent_def.max_iterations,
)
result = agent.run("Audit my authentication module")

from shipit_agent import Agent
from shipit_agent.agents import AgentRegistry

registry = AgentRegistry.default()
agent_def = registry.get("security-auditor")

# Point the agent at your project directory
agent = Agent.with_builtins(
    llm=llm,
    prompt=agent_def.system_prompt(),
    project_root="/path/to/my-project",     # agent works inside this directory
    max_iterations=agent_def.max_iterations,
)

result = agent.run("Perform a full security audit of this project. Check for OWASP Top 10 vulnerabilities.")
print(result.output)

agent_def = registry.get("code-reviewer")

agent = Agent.with_builtins(
    llm=llm,
    prompt=agent_def.system_prompt(),
    project_root="/path/to/my-project",
)

result = agent.run("Review the recent changes in src/auth/ for security and code quality issues")
print(result.output)

agent_def = registry.get("architect")

agent = Agent.with_builtins(
    llm=llm,
    prompt=agent_def.system_prompt(),
    project_root="/path/to/my-project",
)

result = agent.run("Review the project architecture. Identify scalability bottlenecks and suggest improvements.")
print(result.output)

agent_def = registry.get("docker-specialist")

agent = Agent.with_builtins(
    llm=llm,
    prompt=agent_def.system_prompt(),
    project_root="/path/to/my-project",
)

result = agent.run("Review the Dockerfiles and docker-compose.yml for optimization and security best practices")
print(result.output)

agent_def = registry.get("security-auditor")

agent = Agent.with_builtins(
    llm=llm,
    prompt=agent_def.system_prompt(),
    project_root="/path/to/my-project",
)

# Watch the agent work in real-time
for event in agent.stream("Scan the entire project for hardcoded secrets and API keys"):
    if event.type == "tool_called":
        tool = event.payload.get("tool_name", "")
        args = str(event.payload.get("arguments", ""))[:100]
        print(f"  🔧 {tool}: {args}")
    elif event.type == "tool_completed":
        print(f"  ✅ {event.payload.get('tool_name', '')} done")
    elif event.type == "run_completed":
        print(f"\n=== Audit Complete ===")
        print(event.payload.get("output", "")[:1000])

# Plain Agent — no file tools, just the LLM + prompt
agent = Agent(
    llm=llm,
    prompt=agent_def.system_prompt(),
    max_iterations=agent_def.max_iterations,
)

result = agent.run("Explain the OWASP Top 10 vulnerabilities and how to prevent each one")

registry = AgentRegistry.default()

# All categories
print(registry.categories())
# ['Architecture', 'Code Quality', 'Content', 'DevOps', 'Planning', 'Research', 'Security', 'Testing']

# List by category
for agent_def in registry.list_by_category("Security"):
    print(f"{agent_def.id}: {agent_def.role}")

results = registry.search("code review python")
for r in results[:5]:
    print(f"{r.id}: {r.role}")

agent_def = registry.get("architect")
print(agent_def.system_prompt())
# # Role
# You are a Software Architect specializing in scalable, maintainable system design.
# 
# # Goal
# Design robust, scalable architectures...
# 
# # Instructions
# When asked to design or review architecture:
# 1. UNDERSTAND THE CONTEXT...

from shipit_agent.agents import AgentDefinition

custom = AgentDefinition(
    id="my-compliance-checker",
    name="Compliance Checker",
    role="Regulatory compliance specialist",
    goal="Check code for GDPR, SOX, and PCI-DSS compliance",
    backstory="Former compliance officer at a Fortune 500 bank.",
    tools=["read_file", "grep_files", "glob_files"],
    prompt="Review code for regulatory compliance...",
    category="Compliance",
    tags=["compliance", "gdpr", "pci"],
)

# Project-local agents override built-in ones with the same ID
builtin = AgentRegistry.default()
project = AgentRegistry([custom])
merged = builtin.merge(project)

.shipit/
  agents/
    my-custom-agent.json
    researcher.json          # overrides built-in "researcher"

local = AgentRegistry.from_directory(".shipit/agents/")
full = AgentRegistry.default().merge(local)

from shipit_agent.deep.ship_crew import ShipAgent

# Load from registry and wrap as ShipAgent
agent = ShipAgent.from_registry("security-auditor", llm=llm)

from shipit_agent.deep import DeepAgent

agent_def = registry.get("researcher")
deep = DeepAgent.with_builtins(
    llm=llm,
    prompt=agent_def.system_prompt(),
    verify=True,
    reflect=True,
)
result = deep.run("Research AI agent security best practices")

agent_def = registry.get("planner")
agent = Agent.with_builtins(llm=llm, prompt=agent_def.system_prompt())

for event in agent.stream("Plan a microservices migration"):
    print(f"[{event.type}] {event.message}")

from shipit_agent import Agent

agent_def = registry.get("code-reviewer")

# Plain Agent — no built-in tools, just the system prompt
agent = Agent(
    llm=llm,
    prompt=agent_def.system_prompt(),
    max_iterations=agent_def.max_iterations,
)

result = agent.run("Review this function for bugs")
print(result.output)

agent_def = registry.get("security-auditor")
agent = Agent(llm=llm, prompt=agent_def.system_prompt())

for event in agent.stream("Check this code for SQL injection"):
    if event.type == "run_started":
        print(f"[START] Agent running...")
    elif event.type == "tool_called":
        print(f"[TOOL]  {event.payload.get('tool_name', 'llm')}")
    elif event.type == "tool_completed":
        print(f"[DONE]  {event.payload.get('tool_name', 'llm')}")
    elif event.type == "run_completed":
        print(f"[END]   Output: {event.payload.get('output', '')[:200]}")

from shipit_agent.deep import DeepAgent

agent_def = registry.get("researcher")

# DeepAgent with verification and reflection
deep = DeepAgent.with_builtins(
    llm=llm,
    prompt=agent_def.system_prompt(),
    verify=True,
    reflect=True,
    max_iterations=12,
)

# Run with full deep agent capabilities
result = deep.run("Research the latest trends in AI agent security")
print(result.output)

# Stream deep agent events
for event in deep.stream("Research quantum computing advances"):
    print(f"[{event.type}] {event.message[:100]}")

from shipit_agent import Agent
from shipit_agent.deep import DeepAgent

# Load prebuilt agents as sub-agents for delegation
researcher_def = registry.get("researcher")
writer_def = registry.get("blog-writer")
reviewer_def = registry.get("code-reviewer")

researcher_agent = Agent(llm=llm, prompt=researcher_def.system_prompt())
writer_agent = Agent(llm=llm, prompt=writer_def.system_prompt())
reviewer_agent = Agent(llm=llm, prompt=reviewer_def.system_prompt())

deep = DeepAgent.with_builtins(
    llm=llm,
    agents=[researcher_agent, writer_agent, reviewer_agent],
)

result = deep.run("Research AI trends, write a blog post, and review it")

# To JSON (camelCase keys)
d = agent_def.to_dict()

# From JSON (accepts camelCase or snake_case)
restored = AgentDefinition.from_dict(d)