supermemory/packages/tools/src/shared/memory-client.test.ts

import { afterEach, describe, expect, it, vi } from "vitest"
import { buildMemoriesText } from "./memory-client"
import { createLogger } from "./logger"

const API_KEY = "sm_test_key"
const BASE_URL = "https://api.supermemory.ai"
const CONTAINER_TAG = "user-123"

const logger = createLogger(false)

/** Stubs `/v4/profile` so the injected prompt can be asserted without network access. */
function mockProfileResponse(body: unknown) {
	const fetchMock = vi.fn().mockResolvedValue({
		ok: true,
		json: async () => body,
	})
	vi.stubGlobal("fetch", fetchMock)
	return fetchMock
}

afterEach(() => {
	vi.unstubAllGlobals()
})

describe("buildMemoriesText", () => {
	it.each([
		"profile",
		"query",
		"full",
	] as const)("returns no prompt when %s mode finds no memories", async (mode) => {
		mockProfileResponse({
			profile: { static: [], dynamic: [] },
			searchResults: { results: [] },
		})

		const memories = await buildMemoriesText({
			containerTag: CONTAINER_TAG,
			queryText: mode === "profile" ? "" : "what do you know about me?",
			mode,
			baseUrl: BASE_URL,
			apiKey: API_KEY,
			logger,
		})

		expect(memories).toBe("")
	})

	it("does not invoke a custom template when no memories are found", async () => {
		mockProfileResponse({
			profile: { static: [], dynamic: [] },
			searchResults: { results: [] },
		})
		const promptTemplate = vi.fn(() => "<user_memories></user_memories>")

		const memories = await buildMemoriesText({
			containerTag: CONTAINER_TAG,
			queryText: "what do you know about me?",
			mode: "full",
			baseUrl: BASE_URL,
			apiKey: API_KEY,
			logger,
			promptTemplate,
		})

		expect(memories).toBe("")
		expect(promptTemplate).not.toHaveBeenCalled()
	})

	// The profile is not injected in "query" mode. Deduplicating the search
	// results against it would drop a fact present in both, leaving the model
	// with nothing.
	it("injects a search result that also exists in the profile in query mode", async () => {
		mockProfileResponse({
			profile: {
				static: [{ memory: "User is allergic to peanuts" }],
				dynamic: [],
			},
			searchResults: { results: [{ memory: "User is allergic to peanuts" }] },
		})

		const memories = await buildMemoriesText({
			containerTag: CONTAINER_TAG,
			queryText: "what should I avoid eating?",
			mode: "query",
			baseUrl: BASE_URL,
			apiKey: API_KEY,
			logger,
		})

		expect(memories).toContain("User is allergic to peanuts")
	})

	it("does not repeat a profile memory in the search results in full mode", async () => {
		mockProfileResponse({
			profile: {
				static: [{ memory: "User is allergic to peanuts" }],
				dynamic: [],
			},
			searchResults: {
				results: [
					{ memory: "User is allergic to peanuts" },
					{ memory: "User prefers async/await" },
				],
			},
		})

		const memories = await buildMemoriesText({
			containerTag: CONTAINER_TAG,
			queryText: "what should I avoid eating?",
			mode: "full",
			baseUrl: BASE_URL,
			apiKey: API_KEY,
			logger,
		})

		expect(memories).toContain("## Static Profile")
		expect(memories).toContain("User prefers async/await")
		// Present once, under the profile — not duplicated into the search results.
		expect(memories.match(/User is allergic to peanuts/g)).toHaveLength(1)
	})
})