diff --git a/src/main/java/com/olympus/hermione/tools/DocAndSourceCodeTool.java b/src/main/java/com/olympus/hermione/tools/DocAndSourceCodeTool.java
index 3413c18..8806f6d 100644
--- a/src/main/java/com/olympus/hermione/tools/DocAndSourceCodeTool.java
+++ b/src/main/java/com/olympus/hermione/tools/DocAndSourceCodeTool.java
@@ -22,7 +22,7 @@ public class DocAndSourceCodeTool {
 
 
     private static final Logger logger = Logger.getLogger(DocAndSourceCodeTool.class.getName());
-    private static final int DEFAULT_TOP_K = 5;
+    private static final int DEFAULT_TOP_K = 3;
     private static final double DEFAULT_SIMILARITY_THRESHOLD = 0.7;
 
     private final String application;
@@ -80,7 +80,7 @@ public class DocAndSourceCodeTool {
         logger.info("[TOOL]LLM SourceCodeTool getCodeBySimilarity source code for class: " + query);
        SimilaritySearchCodeInput similaritySearchCodeInput = new SimilaritySearchCodeInput();
         similaritySearchCodeInput.setQuery(query);
-        similaritySearchCodeInput.setTopK("3");
+        similaritySearchCodeInput.setTopK("2");
         similaritySearchCodeInput.setSimilarityThreshold("0.7");
         similaritySearchCodeInput.setFilterExpression("ksApplicationName == '" + this.application+"'");