MCP Server Registration & Management

curl -X POST https://aim.example.com/api/v1/agents/agent_123/capabilities \
  -H "Authorization: Bearer {token}" \
  -H "Content-Type: application/json" \
  -d '{
    "capabilities": [
      {
        "name": "database_read",
        "resource": "production_db",
        "permissions": ["SELECT"],
        "mcp_server": "database-mcp",
        "conditions": {
          "tables": ["users", "orders"],
          "time_window": "09:00-18:00",
          "rate_limit": 1000
        }
      },
      {
        "name": "file_system_access",
        "resource": "/data/reports",
        "permissions": ["read", "write"],
        "mcp_server": "filesystem-mcp",
        "conditions": {
          "file_types": [".csv", ".json", ".txt"],
          "max_file_size": "10MB"
        }
      },
      {
        "name": "github_operations",
        "resource": "acme-corp/*",
        "permissions": ["read", "pr_create"],
        "mcp_server": "github-mcp",
        "requires_approval": true
      }
    ]
  }'

// Manual capability registration from Node.js
import axios from 'axios';

async function registerCapabilities(agentId: string, token: string) {
  const response = await axios.post(
    `https://aim.example.com/api/v1/agents/${agentId}/capabilities`,
    {
      capabilities: [
        {
          name: 'api_call',
          resource: 'openai_api',
          permissions: ['completion', 'embedding'],
          mcp_server: 'openai-mcp',
          conditions: {
            models: ['gpt-4', 'text-embedding-3'],
            max_tokens: 4000,
            rate_limit: 100
          }
        }
      ]
    },
    {
      headers: {
        'Authorization': `Bearer ${token}`,
        'Content-Type': 'application/json'
      }
    }
  );

  console.log('Capabilities registered:', response.data);
}

// Manual capability registration from Go
package main

import (
    "bytes"
    "encoding/json"
    "net/http"
)

type Capability struct {
    Name        string            `json:"name"`
    Resource    string            `json:"resource"`
    Permissions []string          `json:"permissions"`
    MCPServer   string            `json:"mcp_server"`
    Conditions  map[string]interface{} `json:"conditions"`
}

func registerCapabilities(agentID, token string) error {
    capabilities := []Capability{
        {
            Name:        "s3_access",
            Resource:    "s3://my-bucket",
            Permissions: []string{"read", "write"},
            MCPServer:   "aws-mcp",
            Conditions: map[string]interface{}{
                "prefix":    "data/",
                "max_size":  "100MB",
            },
        },
    }

    payload, _ := json.Marshal(map[string]interface{}{
        "capabilities": capabilities,
    })

    req, _ := http.NewRequest("POST",
        "https://aim.example.com/api/v1/agents/"+agentID+"/capabilities",
        bytes.NewBuffer(payload))
    req.Header.Set("Authorization", "Bearer "+token)
    req.Header.Set("Content-Type", "application/json")

    client := &http.Client{}
    resp, err := client.Do(req)
    return err
}