@jambonz/llm 0.6.1 → 0.6.3

package/dist/test-kit/index.cjs

@@ -271,10 +271,11 @@ async function* streamFromOpenAI(client, req, options) {
   }
   yield endEvent;
 }
-function appendOpenAIAssistantToolCall(history, toolCalls) {
+function appendOpenAIAssistantToolCall(history, toolCalls, assistantText) {
+  const text = assistantText?.trim() ? assistantText : null;
   const wireMessage = {
     role: "assistant",
-    content: null,
+    content: text,
     tool_calls: toolCalls.map((tc) => ({
       id: tc.id,
       type: "function",
@@ -288,7 +289,7 @@ function appendOpenAIAssistantToolCall(history, toolCalls) {
     ...history,
     {
       role: "assistant",
-      content: "",
+      content: text ?? "",
       vendorRaw: wireMessage
     }
   ];
@@ -559,8 +560,8 @@ var OpenAIAdapter = class {
       includeCacheKey: this.forwardCacheKey()
     });
   }
-  appendAssistantToolCall(history, toolCalls) {
-    return appendOpenAIAssistantToolCall(history, toolCalls);
+  appendAssistantToolCall(history, toolCalls, assistantText) {
+    return appendOpenAIAssistantToolCall(history, toolCalls, assistantText);
   }
   appendToolResult(history, toolCallId, result) {
     return appendOpenAIToolResult(history, toolCallId, result);
@@ -1121,6 +1122,116 @@ var xaiFactory = {
   manifest: xaiManifest,
   create: () => new XaiAdapter()
 };
+var BEDROCK_MANTLE_DEFAULT_BASE_URL = "https://bedrock-mantle.us-east-1.api.aws/v1";
+var bedrockMantleManifest = {
+  vendor: "bedrock-mantle",
+  displayName: "AWS Bedrock (OpenAI-compatible)",
+  authKinds: [
+    {
+      kind: "apiKey",
+      displayName: "API Key",
+      fields: [
+        {
+          name: "apiKey",
+          label: "Bedrock API Key",
+          type: "password",
+          required: true,
+          help: "Generate a long-term Bedrock API key in the AWS console (Amazon Bedrock \u2192 API keys)."
+        },
+        {
+          name: "baseURL",
+          label: "Base URL",
+          type: "url",
+          required: false,
+          default: BEDROCK_MANTLE_DEFAULT_BASE_URL,
+          help: "Change the region to match your key. xAI models are served only on the /openai/v1 path."
+        }
+      ]
+    }
+  ],
+  /* A convenience seed, NOT the source of truth: the live `/v1/models`
+   * endpoint is authoritative and grows continuously, which is why
+   * `supportsModelListing` is true. Any model the endpoint lists can be used
+   * by id without appearing here.
+   *
+   * AWS publishes no context-window metadata (neither `/v1/models` nor
+   * `GetFoundationModel` returns it), so these limits are carried over from
+   * the native-vendor manifests above for the same underlying models. Vision
+   * follows Bedrock's `inputModalities`. */
+  knownModels: [
+    {
+      id: "zai.glm-5",
+      displayName: "GLM 5",
+      capabilities: {
+        streaming: true,
+        tools: true,
+        vision: false,
+        systemPrompt: true,
+        maxContextTokens: 202800
+      }
+    },
+    {
+      id: "zai.glm-4.7",
+      displayName: "GLM 4.7",
+      capabilities: {
+        streaming: true,
+        tools: true,
+        vision: false,
+        systemPrompt: true,
+        maxContextTokens: 2e5
+      }
+    },
+    {
+      id: "moonshotai.kimi-k2.5",
+      displayName: "Kimi K2.5",
+      capabilities: {
+        streaming: true,
+        tools: true,
+        vision: true,
+        systemPrompt: true,
+        maxContextTokens: 262144
+      }
+    },
+    {
+      id: "xai.grok-4.3",
+      displayName: "Grok 4.3",
+      capabilities: {
+        streaming: true,
+        tools: true,
+        vision: true,
+        systemPrompt: true,
+        maxContextTokens: 1e6
+      }
+    }
+  ],
+  supportsModelListing: true,
+  docsUrl: "https://docs.aws.amazon.com/bedrock/latest/userguide/inference-chat-completions.html"
+};
+var BedrockMantleAdapter = class extends OpenAIAdapter {
+  vendor = bedrockMantleManifest.vendor;
+  init(auth, client) {
+    super.init(
+      {
+        ...auth,
+        baseURL: auth.baseURL ?? BEDROCK_MANTLE_DEFAULT_BASE_URL
+      },
+      client
+    );
+  }
+  knownModels() {
+    return bedrockMantleManifest.knownModels;
+  }
+  // Mantle fronts many third-party runtimes with differing param strictness,
+  // and prompt_cache_key is unverified against all of them — do not forward.
+  forwardCacheKey() {
+    return false;
+  }
+};
+var bedrockMantleFactory = {
+  vendor: bedrockMantleManifest.vendor,
+  manifest: bedrockMantleManifest,
+  create: () => new BedrockMantleAdapter()
+};
 
 // src/adapters/anthropic/adapter.ts
 var import_sdk = __toESM(require("@anthropic-ai/sdk"), 1);
@@ -1416,21 +1527,25 @@ var AnthropicAdapter = class {
     }
     yield endEvent;
   }
-  appendAssistantToolCall(history, toolCalls) {
+  appendAssistantToolCall(history, toolCalls, assistantText) {
+    const text = assistantText?.trim() ? assistantText : null;
     const wireMessage = {
       role: "assistant",
-      content: toolCalls.map((tc) => ({
-        type: "tool_use",
-        id: tc.id,
-        name: tc.name,
-        input: tc.arguments ?? {}
-      }))
+      content: [
+        ...text ? [{ type: "text", text }] : [],
+        ...toolCalls.map((tc) => ({
+          type: "tool_use",
+          id: tc.id,
+          name: tc.name,
+          input: tc.arguments ?? {}
+        }))
+      ]
     };
     return [
       ...history,
       {
         role: "assistant",
-        content: "",
+        content: text ?? "",
         vendorRaw: wireMessage
       }
     ];
@@ -1682,21 +1797,25 @@ async function* streamFromGemini(client, req) {
   };
   yield endEvent;
 }
-function appendGeminiAssistantToolCall(history, toolCalls) {
+function appendGeminiAssistantToolCall(history, toolCalls, assistantText) {
+  const text = assistantText?.trim() ? assistantText : null;
   const wireMessage = {
     role: "model",
-    parts: toolCalls.map((tc) => ({
-      functionCall: {
-        name: parseToolName(tc.id),
-        args: tc.arguments ?? {}
-      }
-    }))
+    parts: [
+      ...text ? [{ text }] : [],
+      ...toolCalls.map((tc) => ({
+        functionCall: {
+          name: parseToolName(tc.id),
+          args: tc.arguments ?? {}
+        }
+      }))
+    ]
   };
   return [
     ...history,
     {
       role: "assistant",
-      content: "",
+      content: text ?? "",
       vendorRaw: wireMessage
     }
   ];
@@ -1956,8 +2075,8 @@ var GoogleAdapter = class {
   stream(req) {
     return streamFromGemini(this.ensureClient(), req);
   }
-  appendAssistantToolCall(history, toolCalls) {
-    return appendGeminiAssistantToolCall(history, toolCalls);
+  appendAssistantToolCall(history, toolCalls, assistantText) {
+    return appendGeminiAssistantToolCall(history, toolCalls, assistantText);
   }
   appendToolResult(history, toolCallId, result) {
     return appendGeminiToolResult(history, toolCallId, result);
@@ -2162,8 +2281,8 @@ var VertexGeminiAdapter = class {
   stream(req) {
     return streamFromGemini(this.ensureClient(), req);
   }
-  appendAssistantToolCall(history, toolCalls) {
-    return appendGeminiAssistantToolCall(history, toolCalls);
+  appendAssistantToolCall(history, toolCalls, assistantText) {
+    return appendGeminiAssistantToolCall(history, toolCalls, assistantText);
   }
   appendToolResult(history, toolCallId, result) {
     return appendGeminiToolResult(history, toolCallId, result);
@@ -2360,8 +2479,8 @@ var VertexOpenAIAdapter = class {
       vendorMetadataExtractor: VERTEX_OPENAI_METADATA_EXTRACTOR
     });
   }
-  appendAssistantToolCall(history, toolCalls) {
-    return appendOpenAIAssistantToolCall(history, toolCalls);
+  appendAssistantToolCall(history, toolCalls, assistantText) {
+    return appendOpenAIAssistantToolCall(history, toolCalls, assistantText);
   }
   appendToolResult(history, toolCallId, result) {
     return appendOpenAIToolResult(history, toolCallId, result);
@@ -2847,22 +2966,26 @@ var BedrockAdapter = class {
     }
     yield endEvent;
   }
-  appendAssistantToolCall(history, toolCalls) {
+  appendAssistantToolCall(history, toolCalls, assistantText) {
+    const text = assistantText?.trim() ? assistantText : null;
     const wireMessage = {
       role: "assistant",
-      content: toolCalls.map((tc) => ({
-        toolUse: {
-          toolUseId: tc.id,
-          name: tc.name,
-          input: tc.arguments ?? {}
-        }
-      }))
+      content: [
+        ...text ? [{ text }] : [],
+        ...toolCalls.map((tc) => ({
+          toolUse: {
+            toolUseId: tc.id,
+            name: tc.name,
+            input: tc.arguments ?? {}
+          }
+        }))
+      ]
     };
     return [
       ...history,
       {
         role: "assistant",
-        content: "",
+        content: text ?? "",
         vendorRaw: wireMessage
       }
     ];
@@ -3152,8 +3275,8 @@ var AzureOpenAIAdapter = class {
       vendorMetadataExtractor: AZURE_OPENAI_METADATA_EXTRACTOR
     });
   }
-  appendAssistantToolCall(history, toolCalls) {
-    return appendOpenAIAssistantToolCall(history, toolCalls);
+  appendAssistantToolCall(history, toolCalls, assistantText) {
+    return appendOpenAIAssistantToolCall(history, toolCalls, assistantText);
   }
   appendToolResult(history, toolCallId, result) {
     return appendOpenAIToolResult(history, toolCallId, result);
@@ -3441,6 +3564,7 @@ registerAdapter(basetenFactory);
 registerAdapter(moonshotFactory);
 registerAdapter(zaiFactory);
 registerAdapter(xaiFactory);
+registerAdapter(bedrockMantleFactory);
 registerAdapter(anthropicFactory);
 registerAdapter(googleFactory);
 registerAdapter(vertexGeminiFactory);
@@ -3807,6 +3931,55 @@ function runContractTests(harness) {
       (0, import_vitest.expect)(appended.role).toBe("assistant");
       (0, import_vitest.expect)(appended.vendorRaw).toBeDefined();
     });
+    (0, import_vitest.it)("[22] appendAssistantToolCall carries pre-tool assistant text onto the wire", async () => {
+      await harness.mockScenario("tool-call");
+      const adapter = await init(harness);
+      const first = await drainStream(adapter, {
+        model: harness.toolCapableModel,
+        messages: [{ role: "user", content: "go" }],
+        tools: [
+          { name: "test_tool", description: "x", parameters: { type: "object" } }
+        ]
+      });
+      const tc = first.find(
+        (e) => e.type === "toolCall"
+      );
+      (0, import_vitest.expect)(tc).toBeDefined();
+      const preToolText = "Checking that now.";
+      const history = adapter.appendAssistantToolCall(
+        [{ role: "user", content: "go" }],
+        [tc],
+        preToolText
+      );
+      const appended = history[history.length - 1];
+      (0, import_vitest.expect)(appended.content).toBe(preToolText);
+      (0, import_vitest.expect)(JSON.stringify(appended.vendorRaw)).toContain(preToolText);
+    });
+    (0, import_vitest.it)("[23] appendAssistantToolCall omits empty pre-tool text from the wire", async () => {
+      await harness.mockScenario("tool-call");
+      const adapter = await init(harness);
+      const first = await drainStream(adapter, {
+        model: harness.toolCapableModel,
+        messages: [{ role: "user", content: "go" }],
+        tools: [
+          { name: "test_tool", description: "x", parameters: { type: "object" } }
+        ]
+      });
+      const tc = first.find(
+        (e) => e.type === "toolCall"
+      );
+      (0, import_vitest.expect)(tc).toBeDefined();
+      for (const blank of [void 0, "", "   "]) {
+        const history = adapter.appendAssistantToolCall(
+          [{ role: "user", content: "go" }],
+          [tc],
+          blank
+        );
+        const appended = history[history.length - 1];
+        (0, import_vitest.expect)(appended.content).toBe("");
+        (0, import_vitest.expect)(JSON.stringify(appended.vendorRaw)).not.toContain('"text":""');
+      }
+    });
   });
 }
 async function init(harness) {
@@ -3871,13 +4044,15 @@ var FakeAdapter = class {
         throw new Error(`FakeAdapter: unknown scenario '${scenario}'`);
     }
   }
-  appendAssistantToolCall(history, toolCalls) {
+  appendAssistantToolCall(history, toolCalls, assistantText) {
+    const text = assistantText?.trim() ? assistantText : null;
     return [
       ...history,
       {
         role: "assistant",
-        content: "",
+        content: text ?? "",
         vendorRaw: {
+          ...text ? { text } : {},
           role: "assistant",
           toolCalls: toolCalls.map((tc) => ({
             id: tc.id,