Skip to content

Commit c9096f3

Browse files
committed
fix(viking-mem): change partial response to full response when load_memory
1 parent e7dc2f5 commit c9096f3

1 file changed

Lines changed: 6 additions & 38 deletions

File tree

veadk/memory/long_term_memory_backends/vikingdb_memory_backend.py

Lines changed: 6 additions & 38 deletions
Original file line numberDiff line numberDiff line change
@@ -240,44 +240,12 @@ 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-
)
278-
)
279-
280-
return memory_results
243+
return [
244+
json.dumps(
245+
{"role": "user", "parts": [{"text": str(result)}]},
246+
ensure_ascii=False,
247+
)
248+
]
281249

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

0 commit comments

Comments
 (0)