feat: add topk

2026-06-11 08:16:03 +00:00 · 2025-07-23 17:29:36 +08:00
parent 4e95bc542c
commit 47e9ce96fc
3 changed files with 4 additions and 4 deletions
--- a/pkg/provider/runners/localagent.py
+++ b/pkg/provider/runners/localagent.py
@@ -56,7 +56,7 @@ class LocalAgentRunner(runner.RequestRunner):
                self.ap.logger.warning(f'Knowledge base {kb_uuid} not found')
                raise ValueError(f'Knowledge base {kb_uuid} not found')

-            result = await kb.retrieve(user_message_text)
+            result = await kb.retrieve(user_message_text,kb.knowledge_base_entity.top_k)

            final_user_message_text = ''