search knowledgebase with agent name only

Author: cuericlee

veadk/tools/load_knowledgebase_tool.py

@@ -51,11 +51,11 @@ class SearchKnowledgebaseResponse(BaseModel):
 
 
 async def search_knowledgebase(
-    self, query: str, app_name: str, user_id: str
+    self, query: str, app_name: str
 ) -> SearchKnowledgebaseResponse:
     """Searches the knowledgebase of the current user."""
     if isinstance(knowledgebase, KnowledgeBase):
-        res = knowledgebase.search(query, app_name=app_name, user_id=user_id)
+        res = knowledgebase.search(query, app_name=app_name)
         entry_list = []
         for r in res:
             entry_list.append(KnowledgebaseEntry(content=r))
@@ -87,8 +87,7 @@ async def load_knowledgebase(
     """
     search_knowledgebase_response = await tool_context.search_knowledgebase(
         query,
-        tool_context._invocation_context.app_name,
-        tool_context._invocation_context.user_id,
+        tool_context._invocation_context.app_name
     )
     return LoadKnowledgebaseResponse(
         knowledges=search_knowledgebase_response.knowledges