Refactor OpenVikingMemorySystemClient to use OpenVikingCredential and update session handling; increase EverOS timeout to 180 seconds; enhance tests for credential handling and vector removal in search results.

tests/test_memory_system_service.py

@@ -51,6 +51,62 @@ class FakeEverOS:
         return {"items": [{"source": f"everos-{method}"}]}
 
 
+class FakeEverOSWithVector(FakeEverOS):
+    async def search(self, user_id: str, session_id: str | None, query: str, method: str, limit: int) -> dict:
+        self.calls.append(("search", user_id, session_id, query, method, limit))
+        return {
+            "data": {
+                "episodes": [{"id": "episode-1", "vector": [0.1, 0.2]}],
+                "original_data": {
+                    "episodes": {
+                        "episode-1": {
+                            "summary": "喜欢拿铁",
+                            "vector": [0.1, 0.2],
+                            "nested": {"vector": [0.3]},
+                        }
+                    }
+                },
+            }
+        }
+
+
+def test_capture_includes_exception_type_when_message_is_empty():
+    service = MemorySystemService(openviking=FakeOpenViking(), everos=FakeEverOS())
+
+    class EmptyError(Exception):
+        pass
+
+    async def fail():
+        raise EmptyError()
+
+    response = asyncio.run(service._capture(fail))
+
+    assert response.status == "failed"
+    assert response.error == "EmptyError"
+
+
+def test_search_removes_vectors_from_items_and_backend_results():
+    service = MemorySystemService(openviking=FakeOpenViking(), everos=FakeEverOSWithVector())
+
+    response = asyncio.run(service.search(
+        SearchRequest(user_id="tom", session_id="sess-1", query="咖啡偏好", use_llm=False, limit=5)
+    ))
+
+    assert response.items == [
+        {"source_backend": "openviking", "source": "openviking-find"},
+        {"source_backend": "everos", "id": "episode-1"},
+    ]
+    assert not _has_key(response.backends["everos"].result, "vector")
+
+
+def _has_key(value, key: str) -> bool:
+    if isinstance(value, dict):
+        return key in value or any(_has_key(item, key) for item in value.values())
+    if isinstance(value, list):
+        return any(_has_key(item, key) for item in value)
+    return False
+
+
 def test_ingest_splits_user_and_assistant_messages():
     openviking = FakeOpenViking()
     everos = FakeEverOS()