AWS操作AI代理

┌─────────────────────────────────────────────────────────────────┐
│                     Zarathustra AWS Ops                         │
├─────────────────────────────────────────────────────────────────┤
│                                                                 │
│  ┌─────────────┐    ┌──────────────┐    ┌─────────────────┐    │
│  │ SQS Queue   │───▶│  SQS         │───▶│ Workflow        │    │
│  │ (Requests)  │    │  Listener    │    │ Manager         │    │
│  └─────────────┘    └──────────────┘    └────────┬────────┘    │
│                                                   │             │
│                                         ┌─────────▼─────────┐   │
│                                         │   LangChain       │   │
│                                         │   AWS Ops Agent   │   │
│                                         └─────────┬─────────┘   │
│                                                   │             │
│                                         ┌─────────▼─────────┐   │
│                                         │   MCP Client      │   │
│                                         └─────────┬─────────┘   │
│                                                   │             │
└───────────────────────────────────────────────────┼─────────────┘
                                                    │
                    ┌───────────────────────────────┼───────────────────────────────┐
                    │                               │                               │
           ┌────────▼────────┐            ┌────────▼────────┐            ┌────────▼────────┐
           │  AWS Ops MCP    │            │  Route53 MCP    │            │  S3 MCP         │
           │  (Secrets Mgr)  │            │  (Coming Soon)  │            │  (Coming Soon)  │
           └────────┬────────┘            └────────┬────────┘            └────────┬────────┘
                    │                               │                               │
           ┌────────▼────────┐            ┌────────▼────────┐            ┌────────▼────────┐
           │ AWS Secrets     │            │ AWS Route53     │            │ AWS S3          │
           │ Manager         │            │                 │            │                 │
           └─────────────────┘            └─────────────────┘            └─────────────────┘

zarathustra-aws-ops/
├── main.py                    # Entry point
├── src/
│   ├── core/
│   │   ├── agent.py           # LangChain agent with MCP tools
│   │   └── workflow_manager.py
│   ├── integrations/
│   │   └── mcp_client.py      # MCP client for tool execution
│   ├── listeners/
│   │   └── sqs_listener.py    # SQS message processor
│   └── resources/
│       └── prompt.md          # Agent system prompt
├── config/
│   └── mcp_servers.yaml       # MCP server configuration
├── Dockerfile
├── docker-compose.yml
├── requirements.txt
└── README.md

git clone https://github.com/DepStacks/zarathustra-aws-ops.git
cd zarathustra-aws-ops
python -m venv .venv
source .venv/bin/activate
pip install -r requirements.txt

cp .env.example .env
# Edit .env with your settings

python main.py

# With MCP server
docker-compose up -d

# Agent only (external MCP server)
docker build -t zarathustra-aws-ops .
docker run -d --env-file .env zarathustra-aws-ops

{
  "prompt": "Create a secret called prod/myapp/database with value {\"host\": \"db.example.com\"}",
  "profile": "production",
  "region": "us-east-1",
  "callback_url": "https://your-app.com/webhook/response",
  "metadata": {
    "request_id": "abc123",
    "user": "john@example.com"
  }
}

{
  "prompt": "Create a secret called prod/api/keys with value {\"stripe_key\": \"sk_live_xxx\", \"sendgrid_key\": \"SG.xxx\"}",
  "profile": "production"
}

{
  "prompt": "List all secrets starting with 'prod/' in the staging account",
  "profile": "staging"
}

{
  "prompt": "Delete the secret staging/old-app/config with 7 day recovery window",
  "profile": "staging"
}

Slack/Telegram/Jira → zarathustra-api → SQS → zarathustra-aws-ops → MCP → AWS

def _create_new_service_tools(self) -> List[BaseTool]:
    class NewToolArgs(BaseModel):
        param: str = Field(description="Parameter description")
    
    return [self._create_mcp_tool(
        "aws-ops", "new_tool",
        "Tool description",
        NewToolArgs
    )]

tools.extend(self._create_new_service_tools())

pytest tests/