Skip to content

Commit 3fc4822

Browse files
fix(kb): fix local knowledgebase (#186)
1 parent 1ddfca8 commit 3fc4822

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

veadk/memory/long_term_memory.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -181,17 +181,17 @@ async def add_session_to_memory(
181181

182182
@override
183183
async def search_memory(self, *, app_name: str, user_id: str, query: str):
184-
logger.info(
185-
f"Searching long term memory: query={query} index={self._index} top_k={self.top_k}"
186-
)
187-
188184
# prevent model invoke `load_memory` before add session to this memory
189185
if not self._backend:
190186
logger.error(
191187
"Long term memory backend is not initialized, cannot search memory."
192188
)
193189
return SearchMemoryResponse(memories=[])
194190

191+
logger.info(
192+
f"Searching long term memory: query={query} index={self._index} top_k={self.top_k}"
193+
)
194+
195195
memory_chunks = self._backend.search_memory(
196196
query=query, top_k=self.top_k, user_id=user_id
197197
)

veadk/tools/load_knowledgebase_tool.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ async def search_knowledgebase(
5555
) -> SearchKnowledgebaseResponse:
5656
"""Searches the knowledgebase of the current user."""
5757
if isinstance(knowledgebase, KnowledgeBase):
58-
res = knowledgebase.search(query, app_name=app_name)
58+
res = knowledgebase.search(query)
5959
entry_list = []
6060
for r in res:
6161
entry_list.append(KnowledgebaseEntry(content=r))

0 commit comments

Comments
 (0)