Rename the Picocli command, AssistantService and add a Tool able to read/write file

Signed-off-by: Charles Moulliard <cmoulliard@ibm.com>

Author: cmoulliard

ai/src/main/java/dev/snowdrop/langchain4j/vertexai/CodeAssistantCommand.java

@@ -0,0 +1,22 @@
+package dev.snowdrop.langchain4j.vertexai;
+
+import io.quarkus.picocli.runtime.annotations.TopCommand;
+import jakarta.inject.Inject;
+import picocli.CommandLine.Command;
+import picocli.CommandLine.Parameters;
+
+@TopCommand
+@Command(name = "code", mixinStandardHelpOptions = true)
+public class CodeAssistantCommand implements Runnable {
+
+	@Parameters(defaultValue = "Java Hello World class", description = "The tasks to be executed by the AI coding assistant")
+	String task;
+
+	@Inject
+	CodeAssistantService codeAssistantService;
+
+	@Override
+	public void run() {
+		System.out.println(codeAssistantService.writeCode(task));
+	}
+}

…op/langchain4j/vertexai/MyAiService.java …ain4j/vertexai/CodeAssistantService.javaai/src/main/java/dev/snowdrop/langchain4j/vertexai/MyAiService.java renamed to ai/src/main/java/dev/snowdrop/langchain4j/vertexai/CodeAssistantService.java ai/src/main/java/dev/snowdrop/langchain4j/vertexai/MyAiService.java renamed to ai/src/main/java/dev/snowdrop/langchain4j/vertexai/CodeAssistantService.java

@@ -3,20 +3,18 @@
 import dev.langchain4j.service.SystemMessage;
 import dev.langchain4j.service.UserMessage;
 import io.quarkiverse.langchain4j.RegisterAiService;
+import io.quarkiverse.langchain4j.ToolBox;
 import jakarta.enterprise.context.ApplicationScoped;
 
 @RegisterAiService
 @ApplicationScoped
-public interface MyAiService {
+public interface CodeAssistantService {
 
 	// We got an error HTTP 400: Unexpected role "system". The Messages API accepts a top-level `system` parameter,
 	// not "system" as an input message role."
 	// See: https://platform.claude.com/docs/en/api/messages#message_param
 	//@SystemMessage("You are a professional poet")
 
-	@UserMessage("""
-			    Write a poem about {topic}.
-			    The poem should be {lines} lines long.
-			""")
-	String writeAPoem(String topic, int lines);
+	@ToolBox(FileSystemTool.class)
+	String writeCode(@UserMessage String task);
 }

ai/src/main/java/dev/snowdrop/langchain4j/vertexai/FileSystemTool.java

@@ -0,0 +1,59 @@
+package dev.snowdrop.langchain4j.vertexai;
+
+import dev.langchain4j.agent.tool.P;
+import dev.langchain4j.agent.tool.Tool;
+import jakarta.enterprise.context.ApplicationScoped;
+import org.jboss.logging.Logger;
+
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+
+@ApplicationScoped
+public class FileSystemTool {
+	private static final Logger logger = Logger.getLogger(FileSystemTool.class.getName());
+
+	@Tool("Reads the full content of a specified file")
+	public String readFile(@P("Path to the file to analyze") String path) {
+		try {
+			logger.info("Reading file: " + path);
+			return Files.readString(Paths.get(path));
+		} catch (IOException e) {
+			return "Error reading file: " + e.getMessage();
+		}
+	}
+
+	@Tool("Writes the content to the specified file, overwriting it if it exists")
+	public String writeFile(@P("Path to the file where the content must be changed") String path,
+			@P("The content that you AI proposes to change") String content) {
+
+		logger.debugf("Writing file: %s", path);
+		logger.debugf("String content : %s", content);
+
+		if (path == null || path.isBlank() || content == null) {
+			return "Error: File path and content cannot be null or empty. Please provide both.";
+		}
+
+		/*
+		 * Console console = System.console(); if (console == null) { return
+		 * "Confirmation failed: Console not available."; }
+		 *
+		 * logger.info("\n--------------------------------------------------");
+		 * logger.info("AI is requesting to WRITE to the file: " + path); logger.info("Content to be written:");
+		 * logger.info(content); logger.info("--------------------------------------------------"); String confirmation
+		 * = console.readLine("Do you want to proceed? (y/n): ");
+		 *
+		 * if (!"y".equalsIgnoreCase(confirmation)) { return "Write operation cancelled by user."; }
+		 */
+
+		try {
+			Path filePath = Paths.get(path);
+			logger.debugf("File path is: %s", filePath);
+			Files.writeString(filePath, content);
+			return "File '" + path + "' written successfully.";
+		} catch (IOException e) {
+			return "Error writing file: " + e.getMessage();
+		}
+	}
+}