From 91b15c1cfd808b3d071db960dd58b8014fe09260 Mon Sep 17 00:00:00 2001
From: dosco <832235+dosco@users.noreply.github.com>
Date: Mon, 7 Oct 2024 12:15:52 -0700
Subject: [PATCH] fix: test issue

---
 src/ax/package.json            | 2 +-
 src/ax/prompts/prompts.test.ts | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/ax/package.json b/src/ax/package.json
index 4135a56..3430755 100644
--- a/src/ax/package.json
+++ b/src/ax/package.json
@@ -24,7 +24,7 @@
     "build:module": "tsc -p tsconfig.build.module.json",
     "build:cjs": "tsc -p tsconfig.build.cjs.json",
     "watch:build": "tsc -p tsconfig.build.module.json -w",
-    "test:unit": "ava --verbose",
+    "test": "ava --verbose",
     "tsx": "node --env-file=.env --import=tsx",
     "release": "release-it",
     "publish": "npm run build && cd build && npm publish",
diff --git a/src/ax/prompts/prompts.test.ts b/src/ax/prompts/prompts.test.ts
index 3500bf2..ed6e2fd 100644
--- a/src/ax/prompts/prompts.test.ts
+++ b/src/ax/prompts/prompts.test.ts
@@ -59,7 +59,7 @@ test('generate prompt', async (t) => {
   );
   gen.setExamples(examples);
 
-  const res = await gen.forward({ someText });
+  const res = await gen.forward({ someText }, { stream: false });
 
   t.deepEqual(res, {
     reason: 'Blah blah blah',