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,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]

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"

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,

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)