fix:memory debuging

Author: cuericlee

veadk/knowledgebase/knowledgebase.py

@@ -26,7 +26,7 @@ class KnowledgeBase:
     def __init__(
         self,
         backend: Literal["local", "opensearch", "viking", "redis", "mysql"] = "local",
-        top_k: int = 5,
+        top_k: int = 10,
         db_config=None,
     ):
         logger.debug(f"Create knowledgebase, backend is {backend}")

veadk/memory/long_term_memory.py

@@ -79,6 +79,9 @@ async def add_session_to_memory(
 
     @override
     async def search_memory(self, *, app_name: str, user_id: str, query: str):
+        logger.info(
+            f"Long term memory query: {query} app_name: {app_name} user_id: {user_id}"
+        )
         memory_chunks = self.adapter.query(
             query=query,
             app_name=app_name,

veadk/memory/memory_database_adapter.py

@@ -74,7 +74,7 @@ def add(self, content: list[str], app_name: str, user_id: str, session_id: str):
 
     def query(self, query: str, app_name: str, user_id: str):
         key = f"{app_name}:{user_id}"
-        top_k = 10
+        top_k = 3
 
         try:
             result = self.database_client.query(key, query)
@@ -130,7 +130,7 @@ def add(self, content: list[str], app_name: str, user_id: str, session_id: str):
 
     def query(self, query: str, app_name: str, user_id: str):
         """Search content from table app_name_user_id."""
-        top_k = 10
+        top_k = 3
         table = f"{app_name}_{user_id}"
 
         if not self.database_client.table_exists(table):
@@ -162,11 +162,13 @@ class MemoryVectorDatabaseAdapter(BaseModel):
 
     def add(self, content: list[str], app_name: str, user_id: str, session_id: str):
         collection_name = format_collection_name(f"{app_name}_{user_id}")
+        logger.info(f"Adding {len(content)} texts to collection {collection_name}")
         self.database_client.add(content, collection_name=collection_name)
 
     def query(self, query: str, app_name: str, user_id: str):
         collection_name = format_collection_name(f"{app_name}_{user_id}")
-        top_k = 10
+        top_k = 3
+        logger.info(f"Querying {query} from collection {collection_name}")
         return self.database_client.query(
             query, collection_name=collection_name, top_k=top_k
         )