aixyz 0.16.0 → 0.17.0

package/model.test.ts

@@ -6,12 +6,12 @@ const makeUserPrompt = (text: string): Prompt => [{ role: "user", content: [{ ty
 const identity = (input: string) => input;
 
 describe("fake model", () => {
-  test("has correct metadata", () => {
+  test("has correct metadata", async () => {
     const model = fake(identity);
     expect(model.specificationVersion).toBe("v3");
     expect(model.provider).toBe("aixyz/fake");
     expect(model.modelId).toBe("aixyz/fake");
-    expect(model.supportedUrls).toEqual({});
+    expect(await model.supportedUrls).toEqual({});
   });
 
   describe("doGenerate", () => {
@@ -107,4 +107,29 @@ describe("fake model", () => {
       expect(finish.usage.inputTokens.total).toBe(0);
     });
   });
+
+  describe("call recording", () => {
+    test("records doGenerate calls", async () => {
+      const model = fake(identity);
+      const prompt = makeUserPrompt("hi");
+      await model.doGenerate({ prompt });
+      expect(model.doGenerateCalls).toHaveLength(1);
+      expect(model.doGenerateCalls[0].prompt).toEqual(prompt);
+    });
+
+    test("records multiple doGenerate calls", async () => {
+      const model = fake(identity);
+      await model.doGenerate({ prompt: makeUserPrompt("first") });
+      await model.doGenerate({ prompt: makeUserPrompt("second") });
+      expect(model.doGenerateCalls).toHaveLength(2);
+    });
+
+    test("records doStream calls", async () => {
+      const model = fake(identity);
+      const prompt = makeUserPrompt("stream me");
+      await model.doStream({ prompt });
+      expect(model.doStreamCalls).toHaveLength(1);
+      expect(model.doStreamCalls[0].prompt).toEqual(prompt);
+    });
+  });
 });

package/model.ts

@@ -1,4 +1,5 @@
-import type { LanguageModelV3, LanguageModelV3CallOptions, LanguageModelV3Prompt } from "@ai-sdk/provider";
+import { MockLanguageModelV3 } from "ai/test";
+import type { LanguageModelV3CallOptions, LanguageModelV3Prompt } from "@ai-sdk/provider";
 
 export type Prompt = LanguageModelV3Prompt;
 
@@ -22,30 +23,34 @@ function lastUserText(prompt: Prompt): string {
 }
 
 /**
- * Creates a fake language model for testing and development.
+ * Creates a fake language model for testing and development, backed by `MockLanguageModelV3`.
  * Conforms to the LanguageModelV3 specification with zero token usage.
  *
  * The `transform` function receives the last user message text and the full prompt,
  * and returns the model output string.
  *
+ * All calls are recorded in `.doGenerateCalls` and `.doStreamCalls` for test assertions.
+ *
  * @example
  * import { fake } from "aixyz/model";
  *
  * // custom transform using last message only
  * const helloModel = fake((input) => `hello, ${input}`);
  *
+ * // inspect recorded calls after use
+ * await helloModel.doGenerate({ prompt });
+ * console.log(helloModel.doGenerateCalls.length); // 1
+ *
  * // custom transform using full prompt context
  * const contextModel = fake((input, prompt) => `${prompt.length} turns: ${input}`);
  */
-export function fake(transform: (lastMessage: string, prompt: Prompt) => string): LanguageModelV3 {
-  return {
-    specificationVersion: "v3" as const,
+export function fake(transform: (lastMessage: string, prompt: Prompt) => string): MockLanguageModelV3 {
+  return new MockLanguageModelV3({
     provider: "aixyz/fake",
     modelId: "aixyz/fake",
-    supportedUrls: {},
-    doGenerate(options: LanguageModelV3CallOptions) {
+    doGenerate: async (options: LanguageModelV3CallOptions) => {
       const text = transform(lastUserText(options.prompt), options.prompt);
-      return Promise.resolve({
+      return {
         content: [{ type: "text" as const, text }],
         finishReason: { unified: "stop" as const, raw: undefined },
         usage: {
@@ -53,9 +58,9 @@ export function fake(transform: (lastMessage: string, prompt: Prompt) => string)
           outputTokens: { total: 0, text: 0, reasoning: 0 },
         },
         warnings: [],
-      });
+      };
     },
-    doStream(options: LanguageModelV3CallOptions) {
+    doStream: async (options: LanguageModelV3CallOptions) => {
       const text = transform(lastUserText(options.prompt), options.prompt);
       const stream = new ReadableStream({
         start(controller) {
@@ -74,7 +79,7 @@ export function fake(transform: (lastMessage: string, prompt: Prompt) => string)
           controller.close();
         },
       });
-      return Promise.resolve({ stream });
+      return { stream };
     },
-  };
+  });
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aixyz",
-  "version": "0.16.0",
+  "version": "0.17.0",
   "description": "Payment-native SDK for AI Agent",
   "keywords": [
     "ai",
@@ -23,9 +23,9 @@
   ],
   "dependencies": {
     "@a2a-js/sdk": "^0.3.10",
-    "@aixyz/cli": "0.16.0",
-    "@aixyz/config": "0.16.0",
-    "@aixyz/erc-8004": "0.16.0",
+    "@aixyz/cli": "0.17.0",
+    "@aixyz/config": "0.17.0",
+    "@aixyz/erc-8004": "0.17.0",
     "@modelcontextprotocol/sdk": "^1.26.0",
     "@next/env": "^16.1.6",
     "@x402/core": "^2.3.1",