From c9096f306c909f1b04029f6bdf267d8a8f6fa7ea Mon Sep 17 00:00:00 2001 From: "fangyaozheng@bytedance.com" Date: Thu, 29 Jan 2026 13:37:32 +0800 Subject: [PATCH 1/2] fix(viking-mem): change partial response to full response when load_memory --- .../vikingdb_memory_backend.py | 44 +++---------------- 1 file changed, 6 insertions(+), 38 deletions(-) diff --git a/veadk/memory/long_term_memory_backends/vikingdb_memory_backend.py b/veadk/memory/long_term_memory_backends/vikingdb_memory_backend.py index f2aa5c41..5c37fbcc 100644 --- a/veadk/memory/long_term_memory_backends/vikingdb_memory_backend.py +++ b/veadk/memory/long_term_memory_backends/vikingdb_memory_backend.py @@ -240,44 +240,12 @@ def search_memory( result = response.get("data", {}).get("result_list", []) - memory_results = [] - for r in result: - if r.get("memory_info", {}): - content = "" - # extract user profile - if r.get("memory_info").get("user_profile"): - logger.debug( - f"Detect user profile: {r.get('memory_info').get('user_profile')}" - ) - content += ( - f"User profile: {r.get('memory_info').get('user_profile')} " - ) - - # # extract original messages - # if r.get("memory_info").get("original_message"): - # logger.debug( - # f"Detect original message: {r.get('memory_info').get('original_message')}" - # ) - # content += f"Original message: {r.get('memory_info').get('original_message')} " - - # extract summary - if r.get("memory_info").get("summary"): - logger.debug( - f"Detect summary: {r.get('memory_info').get('summary')}" - ) - content += f"Message summary: {r.get('memory_info').get('summary')}" - - memory_results.append( - json.dumps( - { - "role": "user", - "parts": [{"text": content}], - }, - ensure_ascii=False, - ) - ) - - return memory_results + return [ + json.dumps( + {"role": "user", "parts": [{"text": str(result)}]}, + ensure_ascii=False, + ) + ] def get_user_profile(self, user_id: str) -> str: from veadk.utils.volcengine_sign import ve_request From 8191cd1fee5ee0c8a9b39ac902eefc2f3b00b9bf Mon Sep 17 00:00:00 2001 From: "fangyaozheng@bytedance.com" Date: Thu, 29 Jan 2026 13:38:11 +0800 Subject: [PATCH 2/2] fix empty response --- .../vikingdb_memory_backend.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/veadk/memory/long_term_memory_backends/vikingdb_memory_backend.py b/veadk/memory/long_term_memory_backends/vikingdb_memory_backend.py index 5c37fbcc..00f47391 100644 --- a/veadk/memory/long_term_memory_backends/vikingdb_memory_backend.py +++ b/veadk/memory/long_term_memory_backends/vikingdb_memory_backend.py @@ -240,12 +240,16 @@ def search_memory( result = response.get("data", {}).get("result_list", []) - return [ - json.dumps( - {"role": "user", "parts": [{"text": str(result)}]}, - ensure_ascii=False, - ) - ] + return ( + [ + json.dumps( + {"role": "user", "parts": [{"text": str(result)}]}, + ensure_ascii=False, + ) + ] + if result + else [] + ) def get_user_profile(self, user_id: str) -> str: from veadk.utils.volcengine_sign import ve_request