UNPKG

@langchain/community

Version:
18 lines (17 loc) 858 B
/* eslint-disable no-process-env */ /* eslint-disable @typescript-eslint/no-non-null-assertion */ import { test, expect } from "@jest/globals"; import { MotorheadMemory } from "../motorhead_memory.js"; test("Test managed motörhead memory", async () => { const memory = new MotorheadMemory({ sessionId: new Date().toISOString(), apiKey: process.env.METAL_API_KEY, clientId: process.env.METAL_CLIENT_ID, }); const result1 = await memory.loadMemoryVariables({}); expect(result1).toStrictEqual({ history: "" }); await memory.saveContext({ input: "Who is the best vocalist?" }, { response: "Ozzy Osbourne" }); const expectedString = "Human: Who is the best vocalist?\nAI: Ozzy Osbourne"; const result2 = await memory.loadMemoryVariables({}); expect(result2).toStrictEqual({ history: expectedString }); });