aixyz 0.14.0 → 0.16.0

package/model.test.ts

@@ -1,7 +1,7 @@
 import { describe, expect, test } from "bun:test";
 import { fake, type Prompt } from "./model";
 
-const makeUserPrompt = (text: string) => [{ role: "user", content: [{ type: "text", text }] }];
+const makeUserPrompt = (text: string): Prompt => [{ role: "user", content: [{ type: "text", text }] }];
 
 const identity = (input: string) => input;
 
@@ -19,36 +19,36 @@ describe("fake model", () => {
       const model = fake(identity);
       const result = await model.doGenerate({ prompt: makeUserPrompt("hello world") });
       expect(result.content).toEqual([{ type: "text", text: "hello world" }]);
-      expect(result.finishReason).toBe("stop");
+      expect(result.finishReason).toEqual({ unified: "stop", raw: undefined });
       expect(result.warnings).toEqual([]);
     });
 
     test("applies custom transform to the last user text", async () => {
       const model = fake((input) => `hello, ${input}`);
       const result = await model.doGenerate({ prompt: makeUserPrompt("world") });
-      expect(result.content[0].text).toBe("hello, world");
+      expect(result.content[0]).toEqual({ type: "text", text: "hello, world" });
     });
 
     test("applies custom transform to the last user message when multiple messages exist", async () => {
       const model = fake((input) => `hello, ${input}`);
-      const prompt = [
+      const prompt: Prompt = [
         { role: "user", content: [{ type: "text", text: "first" }] },
         { role: "assistant", content: [{ type: "text", text: "reply" }] },
         { role: "user", content: [{ type: "text", text: "you" }] },
       ];
       const result = await model.doGenerate({ prompt });
-      expect(result.content[0].text).toBe("hello, you");
+      expect(result.content[0]).toEqual({ type: "text", text: "hello, you" });
     });
 
     test("passes full prompt as second argument to transform", async () => {
       const model = fake((_input: string, prompt: Prompt) => `${prompt.length} turns`);
-      const prompt = [
+      const prompt: Prompt = [
         { role: "user", content: [{ type: "text", text: "first" }] },
         { role: "assistant", content: [{ type: "text", text: "reply" }] },
         { role: "user", content: [{ type: "text", text: "second" }] },
       ];
       const result = await model.doGenerate({ prompt });
-      expect(result.content[0].text).toBe("3 turns");
+      expect(result.content[0]).toEqual({ type: "text", text: "3 turns" });
     });
 
     test("returns zero usage", async () => {
@@ -61,42 +61,47 @@ describe("fake model", () => {
     test("returns empty string when no user message", async () => {
       const model = fake(identity);
       const result = await model.doGenerate({ prompt: [] });
-      expect(result.content[0].text).toBe("");
+      expect(result.content[0]).toEqual({ type: "text", text: "" });
     });
   });
 
   describe("doStream", () => {
+    async function collectStream(stream: ReadableStream): Promise<unknown[]> {
+      const reader = stream.getReader();
+      const chunks: unknown[] = [];
+      while (true) {
+        const { done, value } = await reader.read();
+        if (done) break;
+        chunks.push(value);
+      }
+      return chunks;
+    }
+
     test("streams back the last user text with identity transform", async () => {
       const model = fake(identity);
       const { stream } = await model.doStream({ prompt: makeUserPrompt("stream me") });
-      const chunks: unknown[] = [];
-      for await (const chunk of stream) {
-        chunks.push(chunk);
-      }
+      const chunks = await collectStream(stream);
       expect(chunks[0]).toEqual({ type: "stream-start", warnings: [] });
       expect(chunks[1]).toEqual({ type: "text-start", id: "1" });
       expect(chunks[2]).toEqual({ type: "text-delta", id: "1", delta: "stream me" });
       expect(chunks[3]).toEqual({ type: "text-end", id: "1" });
-      expect((chunks[4] as { type: string; finishReason: string }).finishReason).toBe("stop");
+      expect((chunks[4] as { type: string; finishReason: unknown }).finishReason).toEqual({
+        unified: "stop",
+        raw: undefined,
+      });
     });
 
     test("applies custom transform when streaming", async () => {
       const model = fake((input) => `hello, ${input}`);
       const { stream } = await model.doStream({ prompt: makeUserPrompt("you") });
-      const chunks: unknown[] = [];
-      for await (const chunk of stream) {
-        chunks.push(chunk);
-      }
+      const chunks = await collectStream(stream);
       expect(chunks[2]).toEqual({ type: "text-delta", id: "1", delta: "hello, you" });
     });
 
     test("stream contains finish chunk with zero usage", async () => {
       const model = fake(identity);
       const { stream } = await model.doStream({ prompt: makeUserPrompt("hi") });
-      const chunks: unknown[] = [];
-      for await (const chunk of stream) {
-        chunks.push(chunk);
-      }
+      const chunks = await collectStream(stream);
       const finish = chunks[4] as { type: string; usage: { inputTokens: { total: number } } };
       expect(finish.type).toBe("finish");
       expect(finish.usage.inputTokens.total).toBe(0);

package/model.ts

@@ -1,4 +1,6 @@
-export type Prompt = Array<{ role: string; content: Array<{ type: string; text?: string }> }>;
+import type { LanguageModelV3, LanguageModelV3CallOptions, LanguageModelV3Prompt } from "@ai-sdk/provider";
+
+export type Prompt = LanguageModelV3Prompt;
 
 /**
  * Returns the text of the last user message in the prompt, or an empty string if none exists.
@@ -7,8 +9,9 @@ export type Prompt = Array<{ role: string; content: Array<{ type: string; text?:
  */
 function lastUserText(prompt: Prompt): string {
   for (let i = prompt.length - 1; i >= 0; i--) {
-    if (prompt[i].role === "user") {
-      for (const part of prompt[i].content) {
+    const message = prompt[i];
+    if (message.role === "user") {
+      for (const part of message.content) {
         if (part.type === "text" && part.text) {
           return part.text;
         }
@@ -20,7 +23,7 @@ function lastUserText(prompt: Prompt): string {
 
 /**
  * Creates a fake language model for testing and development.
- * Conforms to the LanguageModelV2 v3 specification with zero token usage.
+ * 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.
@@ -34,17 +37,17 @@ function lastUserText(prompt: Prompt): string {
  * // custom transform using full prompt context
  * const contextModel = fake((input, prompt) => `${prompt.length} turns: ${input}`);
  */
-export function fake(transform: (lastMessage: string, prompt: Prompt) => string) {
+export function fake(transform: (lastMessage: string, prompt: Prompt) => string): LanguageModelV3 {
   return {
     specificationVersion: "v3" as const,
     provider: "aixyz/fake",
     modelId: "aixyz/fake",
     supportedUrls: {},
-    doGenerate(options: { prompt: Prompt }) {
+    doGenerate(options: LanguageModelV3CallOptions) {
       const text = transform(lastUserText(options.prompt), options.prompt);
       return Promise.resolve({
         content: [{ type: "text" as const, text }],
-        finishReason: "stop" as const,
+        finishReason: { unified: "stop" as const, raw: undefined },
         usage: {
           inputTokens: { total: 0, noCache: 0, cacheRead: 0, cacheWrite: 0 },
           outputTokens: { total: 0, text: 0, reasoning: 0 },
@@ -52,7 +55,7 @@ export function fake(transform: (lastMessage: string, prompt: Prompt) => string)
         warnings: [],
       });
     },
-    doStream(options: { prompt: Prompt }) {
+    doStream(options: LanguageModelV3CallOptions) {
       const text = transform(lastUserText(options.prompt), options.prompt);
       const stream = new ReadableStream({
         start(controller) {
@@ -62,7 +65,7 @@ export function fake(transform: (lastMessage: string, prompt: Prompt) => string)
           controller.enqueue({ type: "text-end" as const, id: "1" });
           controller.enqueue({
             type: "finish" as const,
-            finishReason: "stop" as const,
+            finishReason: { unified: "stop" as const, raw: undefined },
             usage: {
               inputTokens: { total: 0, noCache: 0, cacheRead: 0, cacheWrite: 0 },
               outputTokens: { total: 0, text: 0, reasoning: 0 },

package/package.json

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