@zhivex-ai/bedrock 0.1.0

package/dist/index.d.ts

@@ -0,0 +1,13 @@
+import { BedrockRuntimeClient } from "@aws-sdk/client-bedrock-runtime";
+import { type CallableProviderAdapter, type ProviderAdapter } from "@zhivex-ai/core";
+export interface BedrockProviderOptions {
+    client?: BedrockRuntimeClient;
+    region?: string;
+}
+export interface BedrockLanguageModelOptions {
+    additionalModelRequestFields?: Record<string, unknown>;
+    additionalModelResponseFieldPaths?: string[];
+    [key: string]: unknown;
+}
+export declare const createBedrock: (options?: BedrockProviderOptions) => CallableProviderAdapter & ProviderAdapter;
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,oBAAoB,EAKrB,MAAM,iCAAiC,CAAC;AAEzC,OAAO,EAOL,KAAK,uBAAuB,EAM5B,KAAK,eAAe,EACrB,MAAM,iBAAiB,CAAC;AAEzB,MAAM,WAAW,sBAAsB;IACrC,MAAM,CAAC,EAAE,oBAAoB,CAAC;IAC9B,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,2BAA2B;IAC1C,4BAA4B,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACvD,iCAAiC,CAAC,EAAE,MAAM,EAAE,CAAC;IAC7C,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB;AA6JD,eAAO,MAAM,aAAa,GAAI,UAAS,sBAA2B,KAAG,uBAAuB,GAAG,eAe9F,CAAC"}

package/dist/index.js

@@ -0,0 +1,152 @@
+import { BedrockRuntimeClient, ConverseCommand } from "@aws-sdk/client-bedrock-runtime";
+import { ConfigurationError, ValidationError, createProviderAdapter, normalizeFinishReason, withRetry, withTimeoutSignal } from "@zhivex-ai/core";
+const capabilities = {
+    streaming: false,
+    tools: false,
+    structuredOutput: false,
+    jsonMode: false,
+    toolChoice: false,
+    parallelToolCalls: false,
+    vision: true,
+    files: false,
+    audioInput: false,
+    audioOutput: false,
+    embeddings: false,
+    reasoning: false,
+    webSearch: false
+};
+const supportedImageFormats = new Set(["png", "jpeg", "gif", "webp"]);
+const parseDataUrl = (value) => {
+    const match = value.match(/^data:(image\/[a-zA-Z0-9.+-]+);base64,(.+)$/);
+    if (!match) {
+        throw new ValidationError("Bedrock image inputs must be provided as data URLs.");
+    }
+    return {
+        mediaType: match[1].toLowerCase(),
+        bytes: Buffer.from(match[2], "base64")
+    };
+};
+const toBedrockImageFormat = (mediaType) => {
+    const subtype = mediaType.split("/")[1]?.toLowerCase() ?? "";
+    if (!supportedImageFormats.has(subtype)) {
+        throw new ValidationError(`Unsupported Bedrock image media type "${mediaType}".`);
+    }
+    return subtype;
+};
+const mapMessagePart = (part) => {
+    switch (part.type) {
+        case "text":
+            return part.text ? [{ text: part.text }] : [];
+        case "image": {
+            const parsed = parseDataUrl(part.image);
+            return [
+                {
+                    image: {
+                        format: toBedrockImageFormat(part.mediaType ?? parsed.mediaType),
+                        source: {
+                            bytes: parsed.bytes
+                        }
+                    }
+                }
+            ];
+        }
+        default:
+            return [];
+    }
+};
+const systemBlocksFromMessages = (messages) => {
+    const text = messages
+        .filter((message) => message.role === "system")
+        .flatMap((message) => message.parts)
+        .filter((part) => part.type === "text")
+        .map((part) => part.text)
+        .join("\n");
+    return text ? [{ text }] : undefined;
+};
+const mapMessages = (messages) => messages
+    .filter((message) => message.role !== "system")
+    .map((message) => ({
+    role: message.role === "assistant" ? "assistant" : "user",
+    content: message.parts.flatMap(mapMessagePart)
+}));
+const normalizeBedrockError = (error) => {
+    const err = error;
+    const message = err.message || "Bedrock request failed.";
+    const lower = message.toLowerCase();
+    const status = err.$metadata?.httpStatusCode;
+    if (status === 401 || status === 403 || lower.includes("accessdenied") || lower.includes("credential")) {
+        return new ConfigurationError(message, { cause: error });
+    }
+    if (status === 400 || lower.includes("validation") || lower.includes("model")) {
+        return new ValidationError(message, { cause: error });
+    }
+    return error instanceof Error ? error : new Error(message);
+};
+class BedrockLanguageModel {
+    modelId;
+    client;
+    provider = "bedrock";
+    capabilities = capabilities;
+    constructor(modelId, client) {
+        this.modelId = modelId;
+        this.client = client;
+    }
+    async generate(input) {
+        const { cleanup } = withTimeoutSignal(input);
+        try {
+            const commandInput = {
+                modelId: this.modelId,
+                messages: mapMessages(input.messages),
+                system: systemBlocksFromMessages(input.messages),
+                inferenceConfig: {
+                    temperature: input.temperature,
+                    maxTokens: input.maxTokens
+                },
+                ...input.providerOptions
+            };
+            const response = await withRetry(() => this.client.send(new ConverseCommand(commandInput)), input).catch((error) => {
+                throw normalizeBedrockError(error);
+            });
+            const text = response.output?.message?.content
+                ?.map((chunk) => ("text" in chunk && chunk.text ? chunk.text : ""))
+                .join("") ?? "";
+            return {
+                messages: text
+                    ? [
+                        {
+                            role: "assistant",
+                            parts: [{ type: "text", text }]
+                        }
+                    ]
+                    : [],
+                text,
+                finishReason: normalizeFinishReason(response.stopReason),
+                providerFinishReason: response.stopReason,
+                usage: {
+                    inputTokens: response.usage?.inputTokens,
+                    outputTokens: response.usage?.outputTokens,
+                    totalTokens: response.usage?.totalTokens ?? ((response.usage?.inputTokens ?? 0) + (response.usage?.outputTokens ?? 0))
+                },
+                rawResponse: response
+            };
+        }
+        finally {
+            cleanup();
+        }
+    }
+}
+export const createBedrock = (options = {}) => {
+    const client = options.client ??
+        (() => {
+            const region = options.region ?? process.env.AWS_REGION;
+            if (!region) {
+                throw new ConfigurationError("Missing AWS region for Bedrock.");
+            }
+            return new BedrockRuntimeClient({ region });
+        })();
+    return createProviderAdapter({
+        name: "bedrock",
+        languageModel: (modelId) => new BedrockLanguageModel(modelId, client)
+    });
+};
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,oBAAoB,EACpB,eAAe,EAIhB,MAAM,iCAAiC,CAAC;AAEzC,OAAO,EACL,kBAAkB,EAClB,eAAe,EACf,qBAAqB,EACrB,qBAAqB,EACrB,SAAS,EACT,iBAAiB,EAQlB,MAAM,iBAAiB,CAAC;AAazB,MAAM,YAAY,GAAsB;IACtC,SAAS,EAAE,KAAK;IAChB,KAAK,EAAE,KAAK;IACZ,gBAAgB,EAAE,KAAK;IACvB,QAAQ,EAAE,KAAK;IACf,UAAU,EAAE,KAAK;IACjB,iBAAiB,EAAE,KAAK;IACxB,MAAM,EAAE,IAAI;IACZ,KAAK,EAAE,KAAK;IACZ,UAAU,EAAE,KAAK;IACjB,WAAW,EAAE,KAAK;IAClB,UAAU,EAAE,KAAK;IACjB,SAAS,EAAE,KAAK;IAChB,SAAS,EAAE,KAAK;CACjB,CAAC;AAEF,MAAM,qBAAqB,GAAG,IAAI,GAAG,CAAC,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;AAEtE,MAAM,YAAY,GAAG,CAAC,KAAa,EAAE,EAAE;IACrC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,6CAA6C,CAAC,CAAC;IACzE,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,IAAI,eAAe,CAAC,qDAAqD,CAAC,CAAC;IACnF,CAAC;IAED,OAAO;QACL,SAAS,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE;QACjC,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC;KACvC,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,oBAAoB,GAAG,CAAC,SAAiB,EAAE,EAAE;IACjD,MAAM,OAAO,GAAG,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;IAC7D,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;QACxC,MAAM,IAAI,eAAe,CAAC,yCAAyC,SAAS,IAAI,CAAC,CAAC;IACpF,CAAC;IACD,OAAO,OAA0C,CAAC;AACpD,CAAC,CAAC;AAEF,MAAM,cAAc,GAAG,CAAC,IAAmC,EAAkB,EAAE;IAC7E,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,KAAK,MAAM;YACT,OAAO,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAChD,KAAK,OAAO,CAAC,CAAC,CAAC;YACb,MAAM,MAAM,GAAG,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACxC,OAAO;gBACL;oBACE,KAAK,EAAE;wBACL,MAAM,EAAE,oBAAoB,CAAC,IAAI,CAAC,SAAS,IAAI,MAAM,CAAC,SAAS,CAAC;wBAChE,MAAM,EAAE;4BACN,KAAK,EAAE,MAAM,CAAC,KAAK;yBACpB;qBACF;iBACF;aACF,CAAC;QACJ,CAAC;QACD;YACE,OAAO,EAAE,CAAC;IACd,CAAC;AACH,CAAC,CAAC;AAEF,MAAM,wBAAwB,GAAG,CAAC,QAAwB,EAAE,EAAE;IAC5D,MAAM,IAAI,GAAG,QAAQ;SAClB,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,QAAQ,CAAC;SAC9C,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC;SACnC,MAAM,CAAC,CAAC,IAAI,EAAoE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;SACxG,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;SACxB,IAAI,CAAC,IAAI,CAAC,CAAC;IAEd,OAAO,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;AACvC,CAAC,CAAC;AAEF,MAAM,WAAW,GAAG,CAAC,QAAwB,EAAa,EAAE,CAC1D,QAAQ;KACL,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,QAAQ,CAAC;KAC9C,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;IACjB,IAAI,EAAE,OAAO,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM;IACzD,OAAO,EAAE,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,cAAc,CAAC;CAC/C,CAAC,CAAC,CAAC;AAER,MAAM,qBAAqB,GAAG,CAAC,KAAc,EAAE,EAAE;IAC/C,MAAM,GAAG,GAAG,KAAqF,CAAC;IAClG,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO,IAAI,yBAAyB,CAAC;IACzD,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IACpC,MAAM,MAAM,GAAG,GAAG,CAAC,SAAS,EAAE,cAAc,CAAC;IAE7C,IAAI,MAAM,KAAK,GAAG,IAAI,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,QAAQ,CAAC,cAAc,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;QACvG,OAAO,IAAI,kBAAkB,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;IAC3D,CAAC;IAED,IAAI,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;QAC9E,OAAO,IAAI,eAAe,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;IACxD,CAAC;IAED,OAAO,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC;AAC7D,CAAC,CAAC;AAEF,MAAM,oBAAoB;IAKb;IACQ;IALV,QAAQ,GAAG,SAAS,CAAC;IACrB,YAAY,GAAG,YAAY,CAAC;IAErC,YACW,OAAe,EACP,MAA4B;QADpC,YAAO,GAAP,OAAO,CAAQ;QACP,WAAM,GAAN,MAAM,CAAsB;IAC5C,CAAC;IAEJ,KAAK,CAAC,QAAQ,CAAC,KAAyB;QACtC,MAAM,EAAE,OAAO,EAAE,GAAG,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAE7C,IAAI,CAAC;YACH,MAAM,YAAY,GAAyB;gBACzC,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,QAAQ,EAAE,WAAW,CAAC,KAAK,CAAC,QAAQ,CAAC;gBACrC,MAAM,EAAE,wBAAwB,CAAC,KAAK,CAAC,QAAQ,CAAC;gBAChD,eAAe,EAAE;oBACf,WAAW,EAAE,KAAK,CAAC,WAAW;oBAC9B,SAAS,EAAE,KAAK,CAAC,SAAS;iBAC3B;gBACD,GAAG,KAAK,CAAC,eAAe;aACzB,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,eAAe,CAAC,YAAY,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACjH,MAAM,qBAAqB,CAAC,KAAK,CAAC,CAAC;YACrC,CAAC,CAAC,CAAC;YAEH,MAAM,IAAI,GACR,QAAQ,CAAC,MAAM,EAAE,OAAO,EAAE,OAAO;gBAC/B,EAAE,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,MAAM,IAAI,KAAK,IAAI,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;iBAClE,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;YAEpB,OAAO;gBACL,QAAQ,EAAE,IAAI;oBACZ,CAAC,CAAC;wBACE;4BACE,IAAI,EAAE,WAAW;4BACjB,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;yBAChC;qBACF;oBACH,CAAC,CAAC,EAAE;gBACN,IAAI;gBACJ,YAAY,EAAE,qBAAqB,CAAC,QAAQ,CAAC,UAAU,CAAC;gBACxD,oBAAoB,EAAE,QAAQ,CAAC,UAAU;gBACzC,KAAK,EAAE;oBACL,WAAW,EAAE,QAAQ,CAAC,KAAK,EAAE,WAAW;oBACxC,YAAY,EAAE,QAAQ,CAAC,KAAK,EAAE,YAAY;oBAC1C,WAAW,EACT,QAAQ,CAAC,KAAK,EAAE,WAAW,IAAI,CAAC,CAAC,QAAQ,CAAC,KAAK,EAAE,WAAW,IAAI,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,EAAE,YAAY,IAAI,CAAC,CAAC,CAAC;iBAC5G;gBACD,WAAW,EAAE,QAAQ;aACtB,CAAC;QACJ,CAAC;gBAAS,CAAC;YACT,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;CACF;AAED,MAAM,CAAC,MAAM,aAAa,GAAG,CAAC,UAAkC,EAAE,EAA6C,EAAE;IAC/G,MAAM,MAAM,GACV,OAAO,CAAC,MAAM;QACd,CAAC,GAAG,EAAE;YACJ,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC;YACxD,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,IAAI,kBAAkB,CAAC,iCAAiC,CAAC,CAAC;YAClE,CAAC;YACD,OAAO,IAAI,oBAAoB,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;QAC9C,CAAC,CAAC,EAAE,CAAC;IAEP,OAAO,qBAAqB,CAAC;QAC3B,IAAI,EAAE,SAAS;QACf,aAAa,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,oBAAoB,CAAC,OAAO,EAAE,MAAM,CAAC;KACtE,CAAC,CAAC;AACL,CAAC,CAAC"}

package/package.json

@@ -0,0 +1,33 @@
+{
+  "name": "@zhivex-ai/bedrock",
+  "version": "0.1.0",
+  "license": "MIT",
+  "type": "module",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    }
+  },
+  "files": ["dist"],
+  "sideEffects": false,
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/Zhivex/zhivex-ai-sdk.git",
+    "directory": "packages/bedrock"
+  },
+  "homepage": "https://github.com/Zhivex/zhivex-ai-sdk#readme",
+  "bugs": {
+    "url": "https://github.com/Zhivex/zhivex-ai-sdk/issues"
+  },
+  "keywords": ["ai", "sdk", "bedrock", "aws", "typescript"],
+  "publishConfig": {
+    "access": "public"
+  },
+  "dependencies": {
+    "@aws-sdk/client-bedrock-runtime": "^3.999.0",
+    "@zhivex-ai/core": "^0.1.0"
+  }
+}