package com.vincent.rsf.server.ai.config;

import com.vincent.rsf.server.ai.entity.AiMcpMount;
import com.vincent.rsf.server.ai.service.BuiltinMcpToolRegistry;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.ai.tool.ToolCallbackProvider;

@Configuration
public class AiMcpServerConfig {

    @Bean
    public ToolCallbackProvider rsfBuiltinMcpToolCallbackProvider(BuiltinMcpToolRegistry builtinMcpToolRegistry) {
        AiMcpMount builtinMount = new AiMcpMount()
                .setBuiltinCode(AiDefaults.MCP_BUILTIN_RSF_WMS);
        return ToolCallbackProvider.from(builtinMcpToolRegistry.createToolCallbacks(builtinMount, null));
    }
}