add mcp server

Author: alexcheng1982

.gitignore

@@ -32,4 +32,6 @@ build/
 .vscode/
 
 ### Mac OS ###
-.DS_Store
+.DS_Store
+
+*.log

core/pom.xml

@@ -17,6 +17,7 @@
     <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
     <kotlin.code.style>official</kotlin.code.style>
     <kotlin.compiler.jvmTarget>1.8</kotlin.compiler.jvmTarget>
+    <docker-java.version>3.7.0</docker-java.version>
   </properties>
 
   <build>
@@ -35,25 +36,20 @@
   </build>
 
   <dependencies>
-    <dependency>
-      <groupId>com.javaaidev.llmcodeexecutor</groupId>
-      <artifactId>code-executor-model</artifactId>
-      <version>0.3.0</version>
-    </dependency>
     <dependency>
       <groupId>com.github.docker-java</groupId>
       <artifactId>docker-java-core</artifactId>
-      <version>3.4.2</version>
+      <version>${docker-java.version}</version>
     </dependency>
     <dependency>
       <groupId>com.github.docker-java</groupId>
       <artifactId>docker-java-transport-httpclient5</artifactId>
-      <version>3.4.2</version>
+      <version>${docker-java.version}</version>
     </dependency>
     <dependency>
       <groupId>ch.qos.logback</groupId>
       <artifactId>logback-classic</artifactId>
-      <version>1.5.16</version>
+      <version>1.5.21</version>
     </dependency>
   </dependencies>
 

core/src/main/kotlin/com/javaaidev/llmcodeexecutor/executor/core/CodeExecutionConfig.java

@@ -0,0 +1,21 @@
+package com.javaaidev.llmcodeexecutor.executor.core;
+
+public class CodeExecutionConfig {
+
+  private String containerImage;
+
+  public CodeExecutionConfig() {
+  }
+
+  public CodeExecutionConfig(String containerImage) {
+    this.containerImage = containerImage;
+  }
+
+  public String getContainerImage() {
+    return containerImage;
+  }
+
+  public void setContainerImage(String containerImage) {
+    this.containerImage = containerImage;
+  }
+}

core/src/main/kotlin/com/javaaidev/llmcodeexecutor/executor/core/CodeExecutor.kt

@@ -0,0 +1,7 @@
+package com.javaaidev.llmcodeexecutor.executor.core
+
+interface CodeExecutor {
+    fun execute(
+        request: ExecuteCodeParameters
+    ): ExecuteCodeResult
+}

core/src/main/kotlin/com/javaaidev/llmcodeexecutor/executor/core/CopiedFile.java

@@ -0,0 +1,21 @@
+package com.javaaidev.llmcodeexecutor.executor.core;
+
+public class CopiedFile {
+
+  private String path;
+
+  public CopiedFile() {
+  }
+
+  public CopiedFile(String path) {
+    this.path = path;
+  }
+
+  public String getPath() {
+    return path;
+  }
+
+  public void setPath(String path) {
+    this.path = path;
+  }
+}

core/src/main/kotlin/com/javaaidev/llmcodeexecutor/executor/core/ExecuteCodeParameters.java

@@ -0,0 +1,47 @@
+package com.javaaidev.llmcodeexecutor.executor.core;
+
+public class ExecuteCodeParameters {
+
+  private String code;
+  private String codeFileName;
+  private OutputFileCollectionConfig outputFileCollectionConfig;
+
+  public ExecuteCodeParameters() {
+  }
+
+  public ExecuteCodeParameters(String code) {
+    this.code = code;
+  }
+
+  public ExecuteCodeParameters(String code, String codeFileName,
+      OutputFileCollectionConfig outputFileCollectionConfig) {
+    this.code = code;
+    this.codeFileName = codeFileName;
+    this.outputFileCollectionConfig = outputFileCollectionConfig;
+  }
+
+  public String getCode() {
+    return code;
+  }
+
+  public void setCode(String code) {
+    this.code = code;
+  }
+
+  public String getCodeFileName() {
+    return codeFileName;
+  }
+
+  public void setCodeFileName(String codeFileName) {
+    this.codeFileName = codeFileName;
+  }
+
+  public OutputFileCollectionConfig getOutputFileCollectionConfig() {
+    return outputFileCollectionConfig;
+  }
+
+  public void setOutputFileCollectionConfig(
+      OutputFileCollectionConfig outputFileCollectionConfig) {
+    this.outputFileCollectionConfig = outputFileCollectionConfig;
+  }
+}

core/src/main/kotlin/com/javaaidev/llmcodeexecutor/executor/core/ExecuteCodeResult.java

@@ -0,0 +1,56 @@
+package com.javaaidev.llmcodeexecutor.executor.core;
+
+import java.util.List;
+
+public class ExecuteCodeResult {
+
+  private String output;
+  private String error;
+  private List<LoadedFile> loadedFiles;
+  private List<CopiedFile> copiedFiles;
+
+  public ExecuteCodeResult() {
+  }
+
+  public ExecuteCodeResult(String output, String error, List<LoadedFile> loadedFiles,
+      List<CopiedFile> copiedFiles) {
+    this.output = output;
+    this.error = error;
+    this.loadedFiles = loadedFiles;
+    this.copiedFiles = copiedFiles;
+  }
+
+  public String getOutput() {
+    return output;
+  }
+
+  public void setOutput(String output) {
+    this.output = output;
+  }
+
+  public String getError() {
+    return error;
+  }
+
+  public void setError(String error) {
+    this.error = error;
+  }
+
+  public List<LoadedFile> getLoadedFiles() {
+    return loadedFiles;
+  }
+
+  public void setLoadedFiles(
+      List<LoadedFile> loadedFiles) {
+    this.loadedFiles = loadedFiles;
+  }
+
+  public List<CopiedFile> getCopiedFiles() {
+    return copiedFiles;
+  }
+
+  public void setCopiedFiles(
+      List<CopiedFile> copiedFiles) {
+    this.copiedFiles = copiedFiles;
+  }
+}

core/src/main/kotlin/com/javaaidev/llmcodeexecutor/executor/core/LLMCodeExecutor.kt

@@ -7,10 +7,6 @@ import com.github.dockerjava.core.DefaultDockerClientConfig
 import com.github.dockerjava.core.DockerClientConfig
 import com.github.dockerjava.core.DockerClientImpl
 import com.github.dockerjava.httpclient5.ApacheDockerHttpClient
-import com.javaaidev.llmcodeexecutor.executor.model.CopiedFile
-import com.javaaidev.llmcodeexecutor.executor.model.ExecuteCodeParameters
-import com.javaaidev.llmcodeexecutor.executor.model.ExecuteCodeReturnType
-import com.javaaidev.llmcodeexecutor.executor.model.LoadedFile
 import org.slf4j.LoggerFactory
 import java.io.Closeable
 import java.nio.file.Files
@@ -38,12 +34,13 @@ data class CodeExecutorConfig(
     val containerOutputDirectory: String? = null,
 )
 
-fun ExecuteCodeParameters.withDefaultIncludedFilePattern(includedFilePattern: String?) {
+fun ExecuteCodeParameters.withDefaultIncludedFilePattern(includedFilePattern: String?): ExecuteCodeParameters {
     this.outputFileCollectionConfig?.let { config ->
         if (config.includedFilePattern == null) {
             config.includedFilePattern = includedFilePattern
         }
     }
+    return this
 }
 
 
@@ -70,7 +67,7 @@ class LLMCodeExecutor(
         logger.info("Use docker client connecting to {}", dockerClientConfig.dockerHost)
     }
 
-    fun execute(request: ExecuteCodeParameters): ExecuteCodeReturnType {
+    fun execute(request: ExecuteCodeParameters): ExecuteCodeResult {
         logger.info("Starting to run code: {}", request)
         pullImage()
         val cmd = dockerClient.createContainerCmd(config.containerImage)
@@ -190,7 +187,7 @@ class LLMCodeExecutor(
 
         dockerClient.removeContainerCmd(containerId).exec()
         logger.info("Container removed {}", containerId)
-        return ExecuteCodeReturnType(
+        return ExecuteCodeResult(
             outputBuilder.toString(),
             errorBuilder.toString(),
             loadedFiles,

core/src/main/kotlin/com/javaaidev/llmcodeexecutor/executor/core/LoadedFile.java

@@ -0,0 +1,31 @@
+package com.javaaidev.llmcodeexecutor.executor.core;
+
+public class LoadedFile {
+
+  private String mimeType;
+  private String data;
+
+  public LoadedFile() {
+  }
+
+  public LoadedFile(String mimeType, String data) {
+    this.mimeType = mimeType;
+    this.data = data;
+  }
+
+  public String getMimeType() {
+    return mimeType;
+  }
+
+  public void setMimeType(String mimeType) {
+    this.mimeType = mimeType;
+  }
+
+  public String getData() {
+    return data;
+  }
+
+  public void setData(String data) {
+    this.data = data;
+  }
+}

core/src/main/kotlin/com/javaaidev/llmcodeexecutor/executor/core/OutputFileCollectionConfig.java

@@ -0,0 +1,52 @@
+package com.javaaidev.llmcodeexecutor.executor.core;
+
+public class OutputFileCollectionConfig {
+
+  private Boolean loadFiles;
+  private Boolean copyFiles;
+  private String copiedFilesPath;
+  private String includedFilePattern;
+
+  public OutputFileCollectionConfig() {
+  }
+
+  public OutputFileCollectionConfig(Boolean loadFiles, Boolean copyFiles, String copiedFilesPath,
+      String includedFilePattern) {
+    this.loadFiles = loadFiles;
+    this.copyFiles = copyFiles;
+    this.copiedFilesPath = copiedFilesPath;
+    this.includedFilePattern = includedFilePattern;
+  }
+
+  public Boolean getLoadFiles() {
+    return loadFiles;
+  }
+
+  public void setLoadFiles(Boolean loadFiles) {
+    this.loadFiles = loadFiles;
+  }
+
+  public Boolean getCopyFiles() {
+    return copyFiles;
+  }
+
+  public void setCopyFiles(Boolean copyFiles) {
+    this.copyFiles = copyFiles;
+  }
+
+  public String getCopiedFilesPath() {
+    return copiedFilesPath;
+  }
+
+  public void setCopiedFilesPath(String copiedFilesPath) {
+    this.copiedFilesPath = copiedFilesPath;
+  }
+
+  public String getIncludedFilePattern() {
+    return includedFilePattern;
+  }
+
+  public void setIncludedFilePattern(String includedFilePattern) {
+    this.includedFilePattern = includedFilePattern;
+  }
+}