Création d'une plateforme de gestion d'outils d'agent : un guide d'architecture pratique

# Tool registration example
class ToolRegistry:
    def register_tool(self, tool_info: dict):
        """
        Register a new tool
        tool_info = {
            "name": "Text Translation Tool",
            "id": "translate_v1",
            "description": "Supports multi-language text translation",
            "version": "1.0.0",
            "api_schema": {...}
        }
        """
        # Validate required information
        self._validate_tool_info(tool_info)
        # Store in database
        self.db.save_tool(tool_info)

-- Core table structure
CREATE TABLE tools (
    id VARCHAR(50) PRIMARY KEY,
    name VARCHAR(100) NOT NULL,
    description TEXT,
    version VARCHAR(20),
    api_schema JSON,
    created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
);

class ToolLoader:
    def __init__(self):
        self._loaded_tools = {}

    def load_tool(self, tool_id: str):
        """Dynamically load a tool"""
        if tool_id in self._loaded_tools:
            return self._loaded_tools[tool_id]

        tool_info = self.registry.get_tool(tool_id)
        tool = self._create_tool_instance(tool_info)
        self._loaded_tools[tool_id] = tool
        return tool

class ToolAccessControl:
    def check_permission(self, user_id: str, tool_id: str) -> bool:
        """Check if user has permission to use a tool"""
        user_role = self.get_user_role(user_id)
        tool_permissions = self.get_tool_permissions(tool_id)

        return user_role in tool_permissions

class ToolTracer:
    def trace_call(self, tool_id: str, params: dict):
        span = self.tracer.start_span(
            name=f"tool_call_{tool_id}",
            attributes={
                "tool_id": tool_id,
                "params": json.dumps(params),
                "timestamp": time.time()
            }
        )
        return span

class ToolMonitor:
    def collect_metrics(self, tool_id: str):
        """Collect tool usage metrics"""
        metrics = {
            "qps": self._calculate_qps(tool_id),
            "latency": self._get_avg_latency(tool_id),
            "error_rate": self._get_error_rate(tool_id)
        }
        return metrics

    def check_alerts(self, metrics: dict):
        """Check if alerts need to be triggered"""
        if metrics["error_rate"] > 0.1:  # Error rate > 10%
            self.send_alert("High Error Rate Alert")

# Initialize platform
platform = ToolPlatform()

# Register new tool
platform.registry.register_tool({
    "id": "weather_v1",
    "name": "Weather Query Tool",
    "description": "Get weather information for major cities worldwide",
    "version": "1.0.0",
    "api_schema": {
        "input": {
            "city": "string",
            "country": "string"
        },
        "output": {
            "temperature": "float",
            "weather": "string"
        }
    }
})

# Use tool
async def use_weather_tool(city: str):
    # Permission check
    if not platform.access_control.check_permission(user_id, "weather_v1"):
        raise PermissionError("No permission to use this tool")

    # Load tool
    tool = platform.loader.load_tool("weather_v1")

    # Call tracing
    with platform.tracer.trace_call("weather_v1", {"city": city}):
        result = await tool.query_weather(city)

    # Collect metrics
    platform.monitor.collect_metrics("weather_v1")

    return result