@langchain/community
Version:
Third-party integrations for LangChain.js
18 lines (17 loc) • 858 B
JavaScript
/* 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 });
});