diff --git a/axhub-tool-core/src/main/java/io/shinhanlife/axhub/biz/mcp/tool/presentation/BusinessToolController.java b/axhub-tool-core/src/main/java/io/shinhanlife/axhub/biz/mcp/tool/presentation/BusinessToolController.java index 830d025..b251991 100644 --- a/axhub-tool-core/src/main/java/io/shinhanlife/axhub/biz/mcp/tool/presentation/BusinessToolController.java +++ b/axhub-tool-core/src/main/java/io/shinhanlife/axhub/biz/mcp/tool/presentation/BusinessToolController.java @@ -25,6 +25,7 @@ import io.shinhanlife.axhub.biz.mcp.tool.annotation.McpFunction; import io.shinhanlife.axhub.biz.mcp.tool.config.McpProperties; import java.lang.reflect.Method; import io.shinhanlife.axhub.biz.mcp.tool.util.JsonSchemaGenerator; +import org.springframework.util.ClassUtils; import lombok.extern.slf4j.Slf4j; @@ -76,7 +77,7 @@ public class BusinessToolController { Map toolBeans = applicationContext.getBeansWithAnnotation(McpTool.class); outerLoop: for (Object bean : toolBeans.values()) { - Class userClass = org.springframework.util.ClassUtils.getUserClass(bean); + Class userClass = ClassUtils.getUserClass(bean); for (Method method : userClass.getDeclaredMethods()) { McpFunction mcpFunc = method.getAnnotation(McpFunction.class); if (mcpFunc != null) { diff --git a/axhub-tool-core/src/main/java/io/shinhanlife/axhub/biz/mcp/tool/service/ToolRegistryHeartbeatSender.java b/axhub-tool-core/src/main/java/io/shinhanlife/axhub/biz/mcp/tool/service/ToolRegistryHeartbeatSender.java index f2ed605..c126806 100644 --- a/axhub-tool-core/src/main/java/io/shinhanlife/axhub/biz/mcp/tool/service/ToolRegistryHeartbeatSender.java +++ b/axhub-tool-core/src/main/java/io/shinhanlife/axhub/biz/mcp/tool/service/ToolRegistryHeartbeatSender.java @@ -23,6 +23,7 @@ import java.util.ArrayList; import java.util.HashMap; import java.util.List; import java.util.Map; +import org.springframework.util.ClassUtils; /** * @package io.shinhanlife.axhub.biz.mcp.tool.service @@ -67,7 +68,7 @@ public class ToolRegistryHeartbeatSender { private void scanAndBuildMetadata() { Map toolBeans = applicationContext.getBeansWithAnnotation(McpTool.class); for (Object bean : toolBeans.values()) { - Class userClass = org.springframework.util.ClassUtils.getUserClass(bean); + Class userClass = ClassUtils.getUserClass(bean); McpTool toolAnnotation = userClass.getAnnotation(McpTool.class); for (Method method : userClass.getDeclaredMethods()) {