add tests

Author: alexcheng1982

.github/workflows/build.yaml

@@ -0,0 +1,24 @@
+name: build
+
+on:
+  push:
+    branches: [ main ]
+  pull_request:
+    branches: [ main ]
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+
+    steps:
+      - uses: actions/checkout@v4
+      - name: Set up JDK
+        uses: actions/setup-java@v3
+        with:
+          java-version: "21"
+          distribution: "temurin"
+          cache: maven
+      - name: Build with Maven
+        env:
+          OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
+        run: mvn --batch-mode -ntp package

core/pom.xml

@@ -41,6 +41,18 @@
           </execution>
         </executions>
       </plugin>
+      <plugin>
+        <groupId>org.apache.maven.plugins</groupId>
+        <artifactId>maven-jar-plugin</artifactId>
+        <version>3.0.2</version>
+        <executions>
+          <execution>
+            <goals>
+              <goal>test-jar</goal>
+            </goals>
+          </execution>
+        </executions>
+      </plugin>
       <plugin>
         <artifactId>maven-surefire-plugin</artifactId>
       </plugin>

core/src/test/kotlin/com/javaaidev/llmcodeexecutor/core/Manual.java

@@ -0,0 +1,14 @@
+package com.javaaidev.llmcodeexecutor.core;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+import org.junit.jupiter.api.Tag;
+
+@Target({ElementType.TYPE, ElementType.METHOD})
+@Retention(RetentionPolicy.RUNTIME)
+@Tag("manual")
+public @interface Manual {
+
+}

executors/python/pom.xml

@@ -79,6 +79,14 @@
       <artifactId>kotlin-stdlib</artifactId>
       <version>2.1.10</version>
     </dependency>
+    <dependency>
+      <groupId>com.javaaidev.llmcodeexecutor</groupId>
+      <artifactId>core</artifactId>
+      <version>${project.version}</version>
+      <classifier>tests</classifier>
+      <type>test-jar</type>
+      <scope>test</scope>
+    </dependency>
   </dependencies>
 
 </project>

executors/python/src/main/kotlin/com/javaaidev/llmcodeexecutor/executor/python/PythonCodeExecutor.kt

@@ -2,16 +2,24 @@ package com.javaaidev.llmcodeexecutor.executor.python
 
 import com.javaaidev.llmcodeexecutor.core.*
 import java.nio.file.Files
+import java.time.Duration
+
+data class PythonCodeExecutorConfig(
+    val containerImage: String? = null,
+)
 
 object PythonCodeExecutor {
 
-    fun execute(request: CodeExecutionRequest): CodeExecutionResponse {
+    fun execute(
+        request: CodeExecutionRequest,
+        config: PythonCodeExecutorConfig? = null
+    ): CodeExecutionResponse {
         val codeDir = Files.createTempDirectory("code_executor")
         val codeFile = "app.py"
         Files.writeString(codeDir.resolve(codeFile), request.code)
         val codeExecutor = LLMCodeExecutor(
             CodeExecutorConfig(
-                "python-executor",
+                config?.containerImage ?: "ghcr.io/javaaidev/llm-code-executor-python:base-3.12",
                 listOf("uv", "run", codeFile),
                 listOf("/app/.venv"),
                 listOf(
@@ -21,10 +29,18 @@ object PythonCodeExecutor {
                     )
                 ),
                 "/app",
-                containerOutputDirectory = "/app",
+                Duration.ofMinutes(3),
+                "/app",
+            )
+        )
+        return codeExecutor.execute(
+            request.copy(
+                outputFileCollectionConfig = request.outputFileCollectionConfig?.copy(
+                    includedFilePattern = request.outputFileCollectionConfig?.includedFilePattern
+                        ?: "!${codeFile}"
+                )
             )
         )
-        return codeExecutor.execute(request)
     }
 }
 

executors/python/src/test/kotlin/com/javaaidev/llmcodeexecutor/executor/python/PythonCodeExecutorTest.kt

@@ -0,0 +1,46 @@
+package com.javaaidev.llmcodeexecutor.executor.python
+
+import com.javaaidev.llmcodeexecutor.core.CodeExecutionRequest
+import com.javaaidev.llmcodeexecutor.core.Manual
+import com.javaaidev.llmcodeexecutor.core.OutputFileCollectionConfig
+import org.junit.jupiter.api.Test
+import kotlin.test.assertEquals
+
+@Manual
+class PythonCodeExecutorTest {
+    @Test
+    fun basic() {
+        val result = PythonCodeExecutor.execute(
+            CodeExecutionRequest(
+                """
+        print("Hello")
+    """.trimIndent()
+            )
+        )
+        assertEquals("Hello", result.output.trim())
+    }
+
+    @Test
+    fun collectFiles() {
+        val result = PythonCodeExecutor.execute(
+            CodeExecutionRequest(
+                """
+        import numpy as np
+        import matplotlib.pyplot as plt
+        x = np.random.randint(0, 100, 100)
+        y = np.random.randint(0, 100, 100)
+        plt.scatter(x, y)
+        plt.savefig('scatter.png')
+        print('Scatter plot saved to scatter.png')
+        
+    """.trimIndent(),
+                OutputFileCollectionConfig(
+                    copyFiles = true,
+                    copiedFilesPath = "./target",
+                    includedFilePattern = "*.png"
+                )
+            )
+        )
+        assertEquals(1, result.copiedFiles?.size)
+    }
+}

pom.xml

@@ -26,6 +26,18 @@
   <build>
     <sourceDirectory>src/main/kotlin</sourceDirectory>
     <testSourceDirectory>src/test/kotlin</testSourceDirectory>
+    <pluginManagement>
+      <plugins>
+        <plugin>
+          <groupId>org.apache.maven.plugins</groupId>
+          <artifactId>maven-surefire-plugin</artifactId>
+          <version>3.2.5</version>
+          <configuration>
+            <excludedGroups>manual</excludedGroups>
+          </configuration>
+        </plugin>
+      </plugins>
+    </pluginManagement>
     <plugins>
       <plugin>
         <groupId>org.jetbrains.kotlin</groupId>
@@ -50,7 +62,6 @@
       </plugin>
       <plugin>
         <artifactId>maven-surefire-plugin</artifactId>
-        <version>2.22.2</version>
       </plugin>
       <plugin>
         <artifactId>maven-failsafe-plugin</artifactId>