mem0ai--mem0/integrations/vercel-ai-sdk/tests/memory-core.test.ts

import { addMemories, retrieveMemories } from "../src";
import { LanguageModelV3Prompt } from '@ai-sdk/provider';
import { testConfig } from "../config/test-config";

describe("Memory Core Functions", () => {
  const { userId } = testConfig;
  jest.setTimeout(20000);

  describe("addMemories", () => {
    it("should successfully add memories and return correct format", async () => {
      const messages: LanguageModelV3Prompt = [
        {
          role: "user",
          content: [
            { type: "text", text: "I love red cars." },
            { type: "text", text: "I like Toyota Cars." },
            { type: "text", text: "I prefer SUVs." },
          ],
        }
      ];

      const response = await addMemories(messages, { user_id: userId });
      
      expect(Array.isArray(response)).toBe(true);
      response.forEach((memory: { event: any; }) => {
        expect(memory).toHaveProperty('id');
        expect(memory).toHaveProperty('data');
        expect(memory).toHaveProperty('event');
        expect(memory.event).toBe('ADD');
      });
    });
  });

  describe("retrieveMemories", () => {
    beforeEach(async () => {
      // Add some test memories before each retrieval test
      const messages: LanguageModelV3Prompt = [
        {
          role: "user",
          content: [
            { type: "text", text: "I love red cars." },
            { type: "text", text: "I like Toyota Cars." },
            { type: "text", text: "I prefer SUVs." },
          ],
        }
      ];
      await addMemories(messages, { user_id: userId });
    });

    it("should retrieve memories with string prompt", async () => {
      const prompt = "Which car would I prefer?";
      const response = await retrieveMemories(prompt, { user_id: userId });
      
      expect(typeof response).toBe('string');
      expect(response.match(/Memory:/g)?.length).toBeGreaterThan(2);
    });

    it("should retrieve memories with array of prompts", async () => {
      const messages: LanguageModelV3Prompt = [
        {
          role: "user",
          content: [
            { type: "text", text: "Which car would I prefer?" },
            { type: "text", text: "Suggest me some cars" },
          ],
        }
      ];

      const response = await retrieveMemories(messages, { user_id: userId });
      
      expect(typeof response).toBe('string');
      expect(response.match(/Memory:/g)?.length).toBeGreaterThan(2);
    });
  });
});