Skip to content

Commit ccd59d1

Browse files
fix(viking-mem): change partial response to full response when load_memory (#479)
* fix(viking-mem): change partial response to full response when load_memory * fix empty response
1 parent e7dc2f5 commit ccd59d1

File tree

1 file changed

+9
-37
lines changed

1 file changed

+9
-37
lines changed

veadk/memory/long_term_memory_backends/vikingdb_memory_backend.py

Lines changed: 9 additions & 37 deletions
Original file line numberDiff line numberDiff line change
@@ -240,44 +240,16 @@ def search_memory(
240240

241241
result = response.get("data", {}).get("result_list", [])
242242

243-
memory_results = []
244-
for r in result:
245-
if r.get("memory_info", {}):
246-
content = ""
247-
# extract user profile
248-
if r.get("memory_info").get("user_profile"):
249-
logger.debug(
250-
f"Detect user profile: {r.get('memory_info').get('user_profile')}"
251-
)
252-
content += (
253-
f"User profile: {r.get('memory_info').get('user_profile')} "
254-
)
255-
256-
# # extract original messages
257-
# if r.get("memory_info").get("original_message"):
258-
# logger.debug(
259-
# f"Detect original message: {r.get('memory_info').get('original_message')}"
260-
# )
261-
# content += f"Original message: {r.get('memory_info').get('original_message')} "
262-
263-
# extract summary
264-
if r.get("memory_info").get("summary"):
265-
logger.debug(
266-
f"Detect summary: {r.get('memory_info').get('summary')}"
267-
)
268-
content += f"Message summary: {r.get('memory_info').get('summary')}"
269-
270-
memory_results.append(
271-
json.dumps(
272-
{
273-
"role": "user",
274-
"parts": [{"text": content}],
275-
},
276-
ensure_ascii=False,
277-
)
243+
return (
244+
[
245+
json.dumps(
246+
{"role": "user", "parts": [{"text": str(result)}]},
247+
ensure_ascii=False,
278248
)
279-
280-
return memory_results
249+
]
250+
if result
251+
else []
252+
)
281253

282254
def get_user_profile(self, user_id: str) -> str:
283255
from veadk.utils.volcengine_sign import ve_request

0 commit comments

Comments
 (0)