summaryrefslogtreecommitdiff
path: root/integration-tests/save_memory.test.js
diff options
context:
space:
mode:
Diffstat (limited to 'integration-tests/save_memory.test.js')
-rw-r--r--integration-tests/save_memory.test.js22
1 files changed, 22 insertions, 0 deletions
diff --git a/integration-tests/save_memory.test.js b/integration-tests/save_memory.test.js
new file mode 100644
index 00000000..db2370d7
--- /dev/null
+++ b/integration-tests/save_memory.test.js
@@ -0,0 +1,22 @@
+/**
+ * @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`;
+ await rig.run(prompt);
+ const result = await rig.run(
+ 'what is my favorite color? tell me that and surround it with $ symbol',
+ );
+
+ assert.ok(result.toLowerCase().includes('$blue$'));
+});