diff --git a/src/main/java/com/olympus/hermione/tools/DocAndSourceCodeTool.java b/src/main/java/com/olympus/hermione/tools/DocAndSourceCodeTool.java index 222126d..3413c18 100644 --- a/src/main/java/com/olympus/hermione/tools/DocAndSourceCodeTool.java +++ b/src/main/java/com/olympus/hermione/tools/DocAndSourceCodeTool.java @@ -82,7 +82,7 @@ public class DocAndSourceCodeTool { similaritySearchCodeInput.setQuery(query); similaritySearchCodeInput.setTopK("3"); similaritySearchCodeInput.setSimilarityThreshold("0.7"); - similaritySearchCodeInput.setFilterExpression("'ksApplicationName' == '" + this.application+"'"); + similaritySearchCodeInput.setFilterExpression("ksApplicationName == '" + this.application+"'"); if(this.application == null || this.application.isEmpty()){ return " Cannot retrieve any source code without an application. Avoid using this tool again."; diff --git a/src/main/java/com/olympus/hermione/tools/SourceCodeTool.java b/src/main/java/com/olympus/hermione/tools/SourceCodeTool.java index ee3b89a..5333c38 100644 --- a/src/main/java/com/olympus/hermione/tools/SourceCodeTool.java +++ b/src/main/java/com/olympus/hermione/tools/SourceCodeTool.java @@ -1,14 +1,12 @@ package com.olympus.hermione.tools; -import java.time.LocalDateTime; import java.util.logging.Logger; import org.springframework.ai.tool.annotation.Tool; import org.springframework.cloud.client.ServiceInstance; -import org.springframework.context.i18n.LocaleContextHolder; +import org.springframework.cloud.client.discovery.DiscoveryClient; import org.springframework.web.client.RestTemplate; -import org.springframework.cloud.client.discovery.DiscoveryClient; import com.olympus.dto.CodeRagResponse; import com.olympus.dto.SimilaritySearchCodeInput;