22 lines
552 B
JavaScript
22 lines
552 B
JavaScript
/**
|
|
* @license
|
|
* Copyright 2025 Google LLC
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
import { test } from 'node:test';
|
|
import { strict as assert } from 'assert';
|
|
import { TestRig } from './test-helper.js';
|
|
|
|
test('should be able to save to memory', async (t) => {
|
|
const rig = new TestRig();
|
|
rig.setup(t.name);
|
|
|
|
const prompt = `remember that my favorite color is blue.
|
|
|
|
what is my favorite color? tell me that and surround it with $ symbol`;
|
|
const result = await rig.run(prompt);
|
|
|
|
assert.ok(result.toLowerCase().includes('$blue$'));
|
|
});
|