fix(ts-oss): return snake_case entity ids from the redis and valkey stores (#6902)

mem0-ts/src/oss/tests/valkey.test.ts

@@ -166,7 +166,7 @@ describe("Valkey – mocked iovalkey client", () => {
     const result = await store.get("mem-1");
     expect(result?.id).toBe("mem-1");
     expect(result?.payload.data).toBe("hello valkey");
-    expect(result?.payload.userId).toBe("alice");
+    expect(result?.payload.user_id).toBe("alice");
     // created_at is persisted as unix seconds and rendered back to its ISO instant.
     expect(result?.payload.createdAt).toBe("2024-01-01T00:00:00.000Z");
   });
@@ -244,6 +244,91 @@ describe("Valkey – mocked iovalkey client", () => {
     expect(searchCall[2]).toContain("@user_id:{a\\|b\\ c}");
   });
 
+  it("get returns entity ids as snake_case", async () => {
+    const store = new ValkeyDB({
+      collectionName: "test",
+      embeddingModelDims: 4,
+      valkeyUrl: "valkey://localhost:6379",
+    });
+    await store.initialize();
+
+    await store.insert(
+      [[0.1, 0.2, 0.3, 0.4]],
+      ["mem-ids"],
+      [
+        {
+          data: "likes tea",
+          user_id: "user-1",
+          agent_id: "agent-1",
+          run_id: "run-1",
+        },
+      ],
+    );
+
+    const result = await store.get("mem-ids");
+    expect(result?.payload.user_id).toBe("user-1");
+    expect(result?.payload.agent_id).toBe("agent-1");
+    expect(result?.payload.run_id).toBe("run-1");
+    expect(result?.payload).not.toHaveProperty("userId");
+    expect(result?.payload).not.toHaveProperty("agentId");
+    expect(result?.payload).not.toHaveProperty("runId");
+  });
+
+  it("search and list return entity ids as snake_case", async () => {
+    const store = new ValkeyDB({
+      collectionName: "test",
+      embeddingModelDims: 4,
+      valkeyUrl: "valkey://localhost:6379",
+    });
+    await store.initialize();
+
+    const iovalkey = require("iovalkey");
+    const mockClient = iovalkey.__mockClient;
+    mockClient.call.mockImplementation(async (...args: any[]) => {
+      if (args[0] === "FT.SEARCH") {
+        return [
+          1,
+          "mem0:test:mem-search",
+          [
+            "memory_id",
+            "mem-search",
+            "hash",
+            "h1",
+            "memory",
+            "likes tea",
+            "created_at",
+            "1700000000",
+            "agent_id",
+            "agent-1",
+            "run_id",
+            "run-1",
+            "user_id",
+            "user-1",
+            "metadata",
+            "{}",
+            "vector_score",
+            "0.1",
+          ],
+        ];
+      }
+      return "OK";
+    });
+
+    const results = await store.search([0.1, 0.2, 0.3, 0.4], 5);
+    expect(results[0].payload.user_id).toBe("user-1");
+    expect(results[0].payload.agent_id).toBe("agent-1");
+    expect(results[0].payload.run_id).toBe("run-1");
+    expect(results[0].payload).not.toHaveProperty("userId");
+    expect(results[0].payload).not.toHaveProperty("agentId");
+    expect(results[0].payload).not.toHaveProperty("runId");
+
+    const [listed] = await store.list();
+    expect(listed[0].payload.user_id).toBe("user-1");
+    expect(listed[0].payload.agent_id).toBe("agent-1");
+    expect(listed[0].payload.run_id).toBe("run-1");
+    expect(listed[0].payload).not.toHaveProperty("userId");
+  });
+
   it("does not raise an unhandled rejection when initialization fails", async () => {
     const iovalkey = require("iovalkey");
     iovalkey.__mockClient.call.mockImplementationOnce(async () => {