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);
}
}