fix: support more languages

Angular2Guy · Jul 12, 2024 · 733726c · 733726c
1 parent 3c30742
commit 733726c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/backend/src/main/java/ch/xxx/aidoclibchat/adapter/controller/CodeGenerationController.java b/backend/src/main/java/ch/xxx/aidoclibchat/adapter/controller/CodeGenerationController.java
@@ -45,7 +45,7 @@ public String getGenerateTests(@RequestParam("url") String url,
 	}
 
 	@GetMapping("/sources")
-	public GithubSources getSources(@RequestParam("url") String url, @RequestParam(required = false) String testUrl) {
+	public GithubSources getSources(@RequestParam("url") String url, @RequestParam(name="testUrl", required = false) String testUrl) {
 		var sources = this.codeGenerationService.createTestSources(URLDecoder.decode(url, StandardCharsets.UTF_8),
 				true);
 		var test = Optional.ofNullable(testUrl)

diff --git a/backend/src/main/java/ch/xxx/aidoclibchat/usecase/service/CodeGenerationService.java b/backend/src/main/java/ch/xxx/aidoclibchat/usecase/service/CodeGenerationService.java
@@ -35,7 +35,7 @@ public class CodeGenerationService {
 	private final String ollamaPrompt = """
 			You are an assistant to generate spring tests for the class under test. 
 			Analyse the classes provided and generate tests for all methods. Base your tests on the test example.
-			Generate and implement the test methods. 	
+			Generate and implement the test methods. Generate and implement complete tests methods.
 					 
 			Generate tests for this class:
 			{classToTest}
@@ -95,7 +95,7 @@ private List<GithubSource> createDependencies(final boolean referencedSources, f
 		return sourceLines.stream().filter(x -> referencedSources).filter(myLine -> myLine.contains("import"))
 				.filter(myLine -> myLine.contains(basePackage))
 				.map(myLine -> String.format("%s%s%s", myUrl.split(basePackage.replace(".", "/"))[0].trim(),
-						myLine.split("import")[1].split(";")[0].replaceAll("\\.", "/").trim(), ".java"))
+						myLine.split("import")[1].split(";")[0].replaceAll("\\.", "/").trim(), myUrl.substring(myUrl.lastIndexOf('.'))))
 				.map(myLine -> this.createTestSources(myLine, false)).toList();
 	}