Skip to content

Commit df76d5c

Browse files
committed
update viking memory logs
1 parent 4c09753 commit df76d5c

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

veadk/memory/long_term_memory_backends/vikingdb_memory_backend.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -98,13 +98,20 @@ def save_memory(self, user_id: str, event_strings: list[str], **kwargs) -> bool:
9898
"default_assistant_id": "assistant",
9999
"time": int(time.time() * 1000),
100100
}
101+
102+
logger.debug(
103+
f"Request for add {len(messages)} memory to VikingDB: collection_name={self.index}, metadata={metadata}, session_id={session_id}"
104+
)
105+
101106
response = self._client.add_messages(
102107
collection_name=self.index,
103108
messages=messages,
104109
metadata=metadata,
105110
session_id=session_id,
106111
)
107112

113+
logger.debug(f"Response from add memory to VikingDB: {response}")
114+
108115
if not response.get("code") == 0:
109116
raise ValueError(f"Save VikingDB memory error: {response}")
110117

@@ -118,10 +125,18 @@ def search_memory(
118125
"user_id": user_id,
119126
"memory_type": ["sys_event_v1"],
120127
}
128+
129+
logger.debug(
130+
f"Request for search memory in VikingDB: filter={filter}, collection_name={self.index}, query={query}",
131+
limit={top_k},
132+
)
133+
121134
response = self._client.search_memory(
122135
collection_name=self.index, query=query, filter=filter, limit=top_k
123136
)
124137

138+
logger.debug(f"Response from search memory in VikingDB: {response}")
139+
125140
if not response.get("code") == 0:
126141
raise ValueError(f"Search VikingDB memory error: {response}")
127142

0 commit comments

Comments
 (0)