From f93ae5c3e51d71441c50328e8fe7cd77fbd47b30 Mon Sep 17 00:00:00 2001
From: "jinli.yl" <jinli.yl@alibaba-inc.com>
Date: Sun, 7 Sep 2025 18:26:10 +0800
Subject: [PATCH] refactor(memory): update message handling and bump version

- Update message variable from 'messages' to 'new_messages' in memory services
- Bump package version from 0.1.6 to 0.1.7
- Update FlowLLM dependency version from >=0.1.5 to >=0.1.6
---
 pyproject.toml                             | 4 ++--
 reme_ai/__init__.py                        | 2 +-
 reme_ai/service/personal_memory_service.py | 4 ++--
 reme_ai/service/task_memory_service.py     | 6 +++---
 4 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index b292df5d..9fc134e6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "reme_ai"
-version = "0.1.6"
+version = "0.1.7"
 description = "Remember me"
 authors = [
     { name = "jinli.yl", email = "jinli.yl@alibaba-inc.com" },
@@ -24,7 +24,7 @@ classifiers = [
 keywords = ["llm", "memory", "experience", "memoryscope", "ai", "mcp", "http"]
 
 dependencies = [
-    "flowllm>=0.1.5",
+    "flowllm>=0.1.6",
 ]
 
 [tool.setuptools.packages.find]
diff --git a/reme_ai/__init__.py b/reme_ai/__init__.py
index 4212844c..4be01267 100644
--- a/reme_ai/__init__.py
+++ b/reme_ai/__init__.py
@@ -3,4 +3,4 @@ from reme_ai import retrieve
 from reme_ai import summary
 from reme_ai import vector_store
 
-__version__ = "0.1.6"
+__version__ = "0.1.7"
diff --git a/reme_ai/service/personal_memory_service.py b/reme_ai/service/personal_memory_service.py
index 0bc2b1b2..60a85a27 100644
--- a/reme_ai/service/personal_memory_service.py
+++ b/reme_ai/service/personal_memory_service.py
@@ -38,7 +38,7 @@ class PersonalMemoryService(BaseMemoryService):
         kwargs = {
             "workspace_id": user_id,
             "trajectories": [
-                {"messages": messages, "score": 1.0}
+                {"messages": new_messages, "score": 1.0}
             ]
         }
 
@@ -67,7 +67,7 @@ class PersonalMemoryService(BaseMemoryService):
                 raise ValueError(f"Invalid message type={type(message)}")
 
         # Extract query from the last message
-        query = messages[-1]["content"] if messages else ""
+        query = new_messages[-1]["content"] if messages else ""
 
         kwargs = {
             "workspace_id": user_id,
diff --git a/reme_ai/service/task_memory_service.py b/reme_ai/service/task_memory_service.py
index 3735daaa..32499d05 100644
--- a/reme_ai/service/task_memory_service.py
+++ b/reme_ai/service/task_memory_service.py
@@ -38,7 +38,7 @@ class TaskMemoryService(BaseMemoryService):
         kwargs = {
             "workspace_id": user_id,
             "trajectories": [
-                {"messages": messages, "score": 1.0}
+                {"messages": new_messages, "score": 1.0}
             ]
         }
 
@@ -68,8 +68,8 @@ class TaskMemoryService(BaseMemoryService):
 
         kwargs = {
             "workspace_id": user_id,
-            "messages": messages,
-            "top_k": filters.get("top_k", 1)
+            "messages": new_messages,
+            "top_k": filters.get("top_k", 1) if filters else 1
         }
 
         result: FlowResponse = await retrieve_flow(**kwargs)