package com.example.springaidemo;

import com.example.springaidemo.tool.McpDemoTool;
import org.springframework.ai.tool.ToolCallbackProvider;
import org.springframework.ai.tool.method.MethodToolCallbackProvider;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.context.annotation.Bean;

@SpringBootApplication
public class SpringAiDemoApplication {

    public static void main(String[] args) {
        String proxy = "127.0.0.1";
        int port = 7897;
        System.setProperty("proxyType", "4");
        System.setProperty("proxyPort", String.valueOf(port));
        System.setProperty("proxyHost", proxy);
        System.setProperty("proxySet", "true");
        SpringApplication.run(SpringAiDemoApplication.class, args);
    }

    @Bean
    public ToolCallbackProvider heroInfoTool(McpDemoTool mcpDemoTool) {
        return MethodToolCallbackProvider.builder().toolObjects(mcpDemoTool).build();
    }
}