diff --git a/pom.xml b/pom.xml index f04d4ac..7b4a7c7 100644 --- a/pom.xml +++ b/pom.xml @@ -39,6 +39,12 @@ 1.0-SNAPSHOT compile + + de.shahondin1624 + rag + 1.0-SNAPSHOT + compile + diff --git a/src/main/java/mcp/app/McpLauncher.java b/src/main/java/mcp/app/McpLauncher.java index dfc3548..deaa153 100644 --- a/src/main/java/mcp/app/McpLauncher.java +++ b/src/main/java/mcp/app/McpLauncher.java @@ -1,14 +1,15 @@ package mcp.app; -import de.shahondin1624.knowledgegraph.DatabaseLauncher; +import de.shahondin1624.rag.LocalRagService; import mcp.server.McpServlet; import mcp.server.ToolRegistrationServlet; +import org.eclipse.jetty.server.Server; import org.eclipse.jetty.servlet.ServletContextHandler; import org.eclipse.jetty.servlet.ServletHolder; -import org.eclipse.jetty.server.Server; import org.slf4j.Logger; import org.slf4j.LoggerFactory; +import java.nio.file.Paths; import java.util.Set; import java.util.stream.Collectors; import java.util.stream.Stream; @@ -18,11 +19,15 @@ public class McpLauncher { public static void main(String[] args) throws Exception { logger.info("Starting MCP App Launcher..."); - new DatabaseLauncher().start(); + // new DatabaseLauncher().start(); + + // Initialize RAG Service and index documentation + LocalRagService ragService = LocalRagService.getInstance(); + ragService.indexDirectory(Paths.get("/home/shahondin1624/Projects/rag/test-data")); // Scan both library tools and app tools Set classpaths = Stream.concat( - Stream.of("mcp.tools", "mcp.app.tools", "de.shahondin1624.knowledgegraph.tooling"), + Stream.of("mcp.tools", "mcp.app.tools", /*"de.shahondin1624.knowledgegraph.tooling", */ "de.shahondin1624.rag.tooling"), args.length > 0 ? Stream.of(args[0].split(",")) : Stream.empty() ).collect(Collectors.toSet()); diff --git a/src/main/java/mcp/app/tools/CalculatorTool.java b/src/main/java/mcp/app/tools/CalculatorTool.java index 204bc44..103b0cc 100644 --- a/src/main/java/mcp/app/tools/CalculatorTool.java +++ b/src/main/java/mcp/app/tools/CalculatorTool.java @@ -52,6 +52,6 @@ public class CalculatorTool extends McpValidatedTool { Expression expression = new ExpressionBuilder(expressionStr).build(); double result = expression.evaluate(); - return success(String.valueOf(result)); + return successResult(result); } } diff --git a/src/main/java/mcp/app/tools/CurrentDateTimeTool.java b/src/main/java/mcp/app/tools/CurrentDateTimeTool.java index 9bf02eb..e0479c4 100644 --- a/src/main/java/mcp/app/tools/CurrentDateTimeTool.java +++ b/src/main/java/mcp/app/tools/CurrentDateTimeTool.java @@ -55,6 +55,6 @@ public class CurrentDateTimeTool extends McpValidatedTool { public McpSchema.CallToolResult callValidated(McpSchema.CallToolRequest request, Map arguments) { String now = LocalDateTime.now().format(DateTimeFormatter.ISO_LOCAL_DATE_TIME); logger.info("CurrentDateTimeTool called, returning: {}", now); - return success(now); + return successResult(now); } } diff --git a/src/main/java/mcp/app/tools/EchoTool.java b/src/main/java/mcp/app/tools/EchoTool.java index d061a6f..8d6f458 100644 --- a/src/main/java/mcp/app/tools/EchoTool.java +++ b/src/main/java/mcp/app/tools/EchoTool.java @@ -50,6 +50,6 @@ public class EchoTool extends McpValidatedTool { public McpSchema.CallToolResult callValidated(McpSchema.CallToolRequest request, Map arguments) { String message = (String) arguments.get("message"); logger.debug("EchoTool called with message: {}", message); - return success(message); + return successResult(message); } }