fix: Adjust kmeans threshold for tool search

inferablehq · Dec 17, 2024 · 91e5318 · 91e5318
1 parent 388658e
commit 91e5318
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/control-plane/src/modules/workflows/agent/run.ts b/control-plane/src/modules/workflows/agent/run.ts
@@ -278,7 +278,7 @@ async function findRelatedFunctionTools(workflow: Run, search: string) {
 
   let relatedTools = relatedToolsSearchResults;
 
-  if (useKmeans && relatedToolsSearchResults?.length > 1) {
+  if (useKmeans && relatedToolsSearchResults?.length > 30) {
     logger.info("Using kmeans clustering for related tool search");
     relatedTools = mostRelevantKMeansCluster(relatedToolsSearchResults);
   }