Skip to content

Commit 874dfa7

Browse files
authored
fix: search knowledgebase with agent name only (#34)
* search knowledgebase with agent name only * search knowledgebase with agent name only
1 parent ef59a10 commit 874dfa7

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

veadk/tools/load_knowledgebase_tool.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -51,11 +51,11 @@ class SearchKnowledgebaseResponse(BaseModel):
5151

5252

5353
async def search_knowledgebase(
54-
self, query: str, app_name: str, user_id: str
54+
self, query: str, app_name: str
5555
) -> SearchKnowledgebaseResponse:
5656
"""Searches the knowledgebase of the current user."""
5757
if isinstance(knowledgebase, KnowledgeBase):
58-
res = knowledgebase.search(query, app_name=app_name, user_id=user_id)
58+
res = knowledgebase.search(query, app_name=app_name)
5959
entry_list = []
6060
for r in res:
6161
entry_list.append(KnowledgebaseEntry(content=r))
@@ -86,9 +86,7 @@ async def load_knowledgebase(
8686
A list of knowledgebase results.
8787
"""
8888
search_knowledgebase_response = await tool_context.search_knowledgebase(
89-
query,
90-
tool_context._invocation_context.app_name,
91-
tool_context._invocation_context.user_id,
89+
query, tool_context._invocation_context.app_name
9290
)
9391
return LoadKnowledgebaseResponse(
9492
knowledges=search_knowledgebase_response.knowledges

0 commit comments

Comments
 (0)