@gajae-code/ai 0.15.4 → 0.15.6

package/src/providers/google-shared.ts

@@ -921,6 +921,7 @@ export function streamGoogleGenAI<T extends "google-generative-ai" | "google-ver
 				options.toolChoice !== "auto" &&
 				options.toolChoice !== "none";
 			const fetchImpl = plan.fetch ?? options?.fetch ?? (globalThis.fetch.bind(globalThis) as FetchImpl);
+			options?.onStreamCreated?.();
 			let response = await fetchImpl(plan.url, {
 				method: "POST",
 				headers: { ...plan.headers, "Content-Type": "application/json", Accept: "text/event-stream" },
@@ -935,6 +936,7 @@ export function streamGoogleGenAI<T extends "google-generative-ai" | "google-ver
 				);
 				if (
 					!options?.fallbackManaged &&
+					!options?.disableProviderRetries &&
 					firstTokenTime === undefined &&
 					isForcedToolChoiceUnsupportedError(error, true)
 				) {
@@ -957,6 +959,7 @@ export function streamGoogleGenAI<T extends "google-generative-ai" | "google-ver
 					params = retryParams;
 					rawRequestDump = { ...rawRequestDump, body: params };
 					wireBody = paramsToWireBody(params);
+					options?.onStreamCreated?.();
 					response = await fetchImpl(plan.url, {
 						method: "POST",
 						headers: { ...plan.headers, "Content-Type": "application/json", Accept: "text/event-stream" },

package/src/providers/kiro-api-key.d.ts

@@ -0,0 +1,50 @@
+import type { Model, StreamFunction } from "../types";
+export declare function isKiroApiKey(value: string | undefined): value is string;
+export declare function kiroApiRegion(options?: {
+    region?: string;
+}): string;
+export declare function kiroApiBaseUrl(region: string): string;
+export declare function toKiroModelId(modelId: string): string;
+export declare function kiroApiStaticModels(): Model<"kiro-codewhisperer-stream">[];
+/** Discover models this API key can use. Returns null when the key is missing. */
+export declare function fetchKiroApiModels(apiKey: string, region?: string): Promise<Model<"kiro-codewhisperer-stream">[]>;
+type KiroStreamEvent = {
+    type: "content";
+    data: string;
+} | {
+    type: "toolUse";
+    data: {
+        name: string;
+        toolUseId: string;
+        input: string;
+        stop?: boolean;
+    };
+} | {
+    type: "toolUseInput";
+    data: {
+        input: string;
+    };
+} | {
+    type: "toolUseStop";
+    data: {
+        stop: boolean;
+    };
+} | {
+    type: "usage";
+    data: {
+        inputTokens?: number;
+        outputTokens?: number;
+    };
+} | {
+    type: "error";
+    data: {
+        error: string;
+        message?: string;
+    };
+};
+export declare function parseKiroApiEvents(buffer: string): {
+    events: KiroStreamEvent[];
+    remaining: string;
+};
+export declare const streamKiroApiKey: StreamFunction<"kiro-codewhisperer-stream">;
+export {};