fix index error

Author: yaozheng-fang

tests/test_knowledgebase.py

@@ -33,5 +33,3 @@ async def test_knowledgebase():
     )
     res = "".join(res_list)
     assert key in res, f"Test failed for backend local res is {res}"
-    assert key in res, f"Test failed for backend local res is {res}"
-    assert key in res, f"Test failed for backend local res is {res}"

tests/test_long_term_memory.py

@@ -49,7 +49,7 @@ async def test_long_term_memory():
         events=[
             Event(
                 invocation_id="test_invocation_id",
-                author=agent.name,
+                author="user",
                 branch=None,
                 content=types.Content(
                     parts=[types.Part(text="My name is Alice.")],

veadk/knowledgebase/knowledgebase.py

@@ -59,7 +59,7 @@ def add(
         """
         kwargs.pop("session_id", None)  # remove session_id
         logger.info(f"Adding documents to knowledgebase: app_name={app_name}")
-        self.adapter.add(data, app_name, **kwargs)
+        self.adapter.add(data, app_name=app_name, **kwargs)
 
     def search(self, query: str, app_name: str, top_k: int = None) -> list[str]:
         """Retrieve documents similar to the query text in the vector database.

veadk/knowledgebase/knowledgebase_database_adapter.py

@@ -200,10 +200,10 @@ class KnowledgebaseLocalDatabaseAdapter(BaseModel):
 
     database_client: BaseDatabase
 
-    def add(self, content: list[str], **kwargs):
-        self.database_client.add(content)
+    def add(self, data: list[str], **kwargs):
+        self.database_client.add(data)
 
-    def query(self, query: str, app_name: str, user_id: str, **kwargs):
+    def query(self, query: str, **kwargs):
         return self.database_client.query(query, **kwargs)
 
     def delete(self, app_name: str, **kwargs):