package com.zy.ai.mcp.dto;


import lombok.Data;

import java.util.*;

@Data
public class ToolDefinition {
    private String name;
    private String description;
    private Map<String, Object> inputSchema;   // JSON Schema as Map
    private Map<String, Object> outputSchema;  // JSON Schema as Map
    private McpToolHandler handler;

    public Map<String, Object> toMcpToolJson() {
        Map<String, Object> m = new LinkedHashMap<String, Object>();
        m.put("name", name);
        m.put("description", description);
        m.put("inputSchema", inputSchema);
        m.put("outputSchema", outputSchema);
        return m;
    }
}