Merge branch 'dev' into dev

Author: CaralHsi

.github/workflows/python-tests.yml

@@ -41,7 +41,12 @@ jobs:
     steps:
     - uses: actions/checkout@v4
     - name: Install poetry
-      run: pipx install poetry
+      # This is a temporary fix to ensure compatibility with Poetry & virtualenv
+      # Revert to the original installation method once the poetry==2.1.4 is released
+      run: |
+        echo "virtualenv==20.32.0" > constraints.txt
+        pipx install poetry==2.1.3 --pip-args="--constraint=constraints.txt"
+        rm constraints.txt
     - name: Set up Python ${{ matrix.python-version }}
       uses: actions/setup-python@v5
       with:

docs/openapi.json

@@ -884,7 +884,7 @@
             "type": "string",
             "title": "Session Id",
             "description": "Session ID for the MOS. This is used to distinguish between different dialogue",
-            "default": "842877f4-c3f7-4c22-ad38-5950026870fe"
+            "default": "0ce84b9c-0615-4b9d-83dd-fba50537d5d3"
           },
           "chat_model": {
             "$ref": "#/components/schemas/LLMConfigFactory",

src/memos/api/context/dependencies.py

@@ -1,4 +1,5 @@
 import logging
+import os
 
 from fastapi import Depends, Header, Request
 
@@ -24,6 +25,13 @@ def get_trace_id_from_header(
     return g_trace_id or x_trace_id or trace_id
 
 
+def generate_trace_id() -> str:
+    """
+    Get a random trace_id.
+    """
+    return os.urandom(16).hex()
+
+
 def get_request_context(
     request: Request, trace_id: str | None = Depends(get_trace_id_from_header)
 ) -> RequestContext:
@@ -57,6 +65,9 @@ def get_g_object(trace_id: str | None = Depends(get_trace_id_from_header)) -> G:
     This creates a RequestContext and sets it globally for access
     throughout the request lifecycle.
     """
+    if trace_id is None:
+        trace_id = generate_trace_id()
+
     g = RequestContext(trace_id=trace_id)
     set_request_context(g)
     logger.info(f"Request g object created with trace_id: {g.trace_id}")

src/memos/graph_dbs/base.py

@@ -70,15 +70,27 @@ def edge_exists(self, source_id: str, target_id: str, type: str) -> bool:
 
     # Graph Query & Reasoning
     @abstractmethod
-    def get_node(self, id: str) -> dict[str, Any] | None:
+    def get_node(self, id: str, include_embedding: bool = False) -> dict[str, Any] | None:
         """
         Retrieve the metadata and content of a node.
         Args:
             id: Node identifier.
+            include_embedding: with/without embedding
         Returns:
             Dictionary of node fields, or None if not found.
         """
 
+    @abstractmethod
+    def get_nodes(self, id: str, include_embedding: bool = False) -> dict[str, Any] | None:
+        """
+        Retrieve the metadata and memory of a list of nodes.
+        Args:
+            ids: List of Node identifier.
+            include_embedding: with/without embedding
+        Returns:
+        list[dict]: Parsed node records containing 'id', 'memory', and 'metadata'.
+        """
+
     @abstractmethod
     def get_neighbors(
         self, id: str, type: str, direction: Literal["in", "out", "both"] = "out"
@@ -163,7 +175,9 @@ def get_by_metadata(self, filters: list[dict[str, Any]]) -> list[str]:
         """
 
     @abstractmethod
-    def get_structure_optimization_candidates(self, scope: str) -> list[dict]:
+    def get_structure_optimization_candidates(
+        self, scope: str, include_embedding: bool = False
+    ) -> list[dict]:
         """
         Find nodes that are likely candidates for structure optimization:
         - Isolated nodes, nodes with empty background, or nodes with exactly one child.
@@ -205,7 +219,7 @@ def clear(self) -> None:
         """
 
     @abstractmethod
-    def export_graph(self) -> dict[str, Any]:
+    def export_graph(self, include_embedding: bool = False) -> dict[str, Any]:
         """
         Export the entire graph as a serializable dictionary.
 
@@ -221,3 +235,16 @@ def import_graph(self, data: dict[str, Any]) -> None:
         Args:
             data: A dictionary containing all nodes and edges to be loaded.
         """
+
+    @abstractmethod
+    def get_all_memory_items(self, scope: str, include_embedding: bool = False) -> list[dict]:
+        """
+        Retrieve all memory items of a specific memory_type.
+
+        Args:
+            scope (str): Must be one of 'WorkingMemory', 'LongTermMemory', or 'UserMemory'.
+            include_embedding: with/without embedding
+
+        Returns:
+            list[dict]: Full list of memory items under this scope.
+        """