@holokai/holo-provider-bedrock 1.0.0

package/dist/bedrock.response.factory.js

@@ -0,0 +1,9 @@
+export class BedrockResponseFactory {
+    static instance() {
+        return new BedrockResponseFactory();
+    }
+    createError(message, code) {
+        return { error: { message, ...(code ? { code } : {}) } };
+    }
+}
+//# sourceMappingURL=bedrock.response.factory.js.map

package/dist/bedrock.response.factory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bedrock.response.factory.js","sourceRoot":"","sources":["../src/bedrock.response.factory.ts"],"names":[],"mappings":"AAEA,MAAM,OAAO,sBAAsB;IAC/B,MAAM,CAAC,QAAQ;QACX,OAAO,IAAI,sBAAsB,EAAE,CAAC;IACxC,CAAC;IAED,WAAW,CAAC,OAAe,EAAE,IAAa;QACtC,OAAO,EAAC,KAAK,EAAE,EAAC,OAAO,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,EAAC,IAAI,EAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAC,EAAC,CAAC;IACvD,CAAC;CACJ"}

package/dist/bedrock.translator.d.ts

@@ -0,0 +1,30 @@
+import type { IProviderTranslator } from '@holokai/holo-types/provider';
+import type { HoloEmbedParams, HoloMessage, HoloRequest, HoloResponse, HoloStreamChunk } from '@holokai/holo-types/holo';
+import type { BedrockConverseRequest, BedrockConverseResponse } from './types';
+import { BedrockMessageTranslator, BedrockRequestTranslator, BedrockResponseTranslator, BedrockStreamTranslator } from './translators';
+import type { Message } from '@aws-sdk/client-bedrock-runtime';
+export declare class BedrockTranslator implements IProviderTranslator {
+    private readonly requestTranslator;
+    private readonly responseTranslator;
+    private readonly messageTranslator;
+    private readonly streamTranslator;
+    constructor(requestTranslator: BedrockRequestTranslator, responseTranslator: BedrockResponseTranslator, messageTranslator: BedrockMessageTranslator, streamTranslator: BedrockStreamTranslator);
+    static instance(): IProviderTranslator;
+    fromHoloRequest(request: HoloRequest): Promise<Partial<BedrockConverseRequest>>;
+    toHoloRequest(request: BedrockConverseRequest): Promise<Partial<HoloRequest>>;
+    fromHoloResponse(response: HoloResponse): Promise<Partial<BedrockConverseResponse>>;
+    toHoloResponse(response: BedrockConverseResponse): Promise<Partial<HoloResponse>>;
+    fromHoloMessages(messages: HoloMessage[]): Promise<Partial<Message>[]>;
+    toHoloMessages(messages: Message[]): Promise<Partial<HoloMessage>[]>;
+    fromHoloStreamChunks(chunks: HoloStreamChunk[]): Promise<unknown>;
+    fromHoloEmbedRequest(request: HoloEmbedParams): Promise<{
+        modelId: string;
+        body: Record<string, unknown>;
+    }>;
+    toHoloEmbedResponse(response: any): Promise<{
+        model: string;
+        embeddings: number[][];
+        usage?: any;
+    }>;
+}
+//# sourceMappingURL=bedrock.translator.d.ts.map

package/dist/bedrock.translator.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"bedrock.translator.d.ts","sourceRoot":"","sources":["../src/bedrock.translator.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAC,mBAAmB,EAAC,MAAM,8BAA8B,CAAC;AACtE,OAAO,KAAK,EAAC,eAAe,EAAE,WAAW,EAAE,WAAW,EAAE,YAAY,EAAE,eAAe,EAAC,MAAM,0BAA0B,CAAC;AACvH,OAAO,KAAK,EAAC,sBAAsB,EAAE,uBAAuB,EAAC,MAAM,SAAS,CAAC;AAC7E,OAAO,EAGH,wBAAwB,EACxB,wBAAwB,EACxB,yBAAyB,EACzB,uBAAuB,EAE1B,MAAM,eAAe,CAAC;AACvB,OAAO,KAAK,EAAC,OAAO,EAAC,MAAM,iCAAiC,CAAC;AAkD7D,qBAAa,iBAAkB,YAAW,mBAAmB;IAErD,OAAO,CAAC,QAAQ,CAAC,iBAAiB;IAClC,OAAO,CAAC,QAAQ,CAAC,kBAAkB;IACnC,OAAO,CAAC,QAAQ,CAAC,iBAAiB;IAClC,OAAO,CAAC,QAAQ,CAAC,gBAAgB;gBAHhB,iBAAiB,EAAE,wBAAwB,EAC3C,kBAAkB,EAAE,yBAAyB,EAC7C,iBAAiB,EAAE,wBAAwB,EAC3C,gBAAgB,EAAE,uBAAuB;IAI9D,MAAM,CAAC,QAAQ,IAAI,mBAAmB;IAchC,eAAe,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC;IAI/E,aAAa,CAAC,OAAO,EAAE,sBAAsB,GAAG,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IAI7E,gBAAgB,CAAC,QAAQ,EAAE,YAAY,GAAG,OAAO,CAAC,OAAO,CAAC,uBAAuB,CAAC,CAAC;IAInF,cAAc,CAAC,QAAQ,EAAE,uBAAuB,GAAG,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;IAIjF,gBAAgB,CAAC,QAAQ,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;IAItE,cAAc,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC;IAIpE,oBAAoB,CAAC,MAAM,EAAE,eAAe,EAAE,GAAG,OAAO,CAAC,OAAO,CAAC;IAIjE,oBAAoB,CAAC,OAAO,EAAE,eAAe,GAAG,OAAO,CAAC;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;KAAE,CAAC;IAQ3G,mBAAmB,CAAC,QAAQ,EAAE,GAAG,GAAG,OAAO,CAAC;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,MAAM,EAAE,EAAE,CAAC;QAAC,KAAK,CAAC,EAAE,GAAG,CAAA;KAAE,CAAC;CAQ5G"}

package/dist/bedrock.translator.js

@@ -0,0 +1,96 @@
+import { BedrockContentDeltaTranslator, BedrockMessageStopTranslator, BedrockMessageTranslator, BedrockRequestTranslator, BedrockResponseTranslator, BedrockStreamTranslator, BedrockToolTranslator, } from './translators';
+const TITAN_FORMATTER = {
+    formatRequest(_model, input) {
+        const text = Array.isArray(input) ? input.join(' ') : input;
+        return { inputText: text };
+    },
+    parseResponse(body) {
+        const embedding = body.embedding;
+        const inputTokenCount = body.inputTextTokenCount;
+        const result = { embeddings: embedding ? [embedding] : [] };
+        if (inputTokenCount != null) {
+            result.usage = { input_tokens: inputTokenCount, total_tokens: inputTokenCount };
+        }
+        return result;
+    },
+};
+const COHERE_FORMATTER = {
+    formatRequest(_model, input) {
+        const texts = Array.isArray(input) ? input : [input];
+        return { texts, input_type: 'search_document' };
+    },
+    parseResponse(body) {
+        const embeddings = body.embeddings;
+        return { embeddings: embeddings ?? [] };
+    },
+};
+const EMBED_FORMATTERS = new Map([
+    ['amazon.titan-embed', TITAN_FORMATTER],
+    ['cohere.embed', COHERE_FORMATTER],
+]);
+function getEmbedFormatter(modelId) {
+    for (const [prefix, formatter] of EMBED_FORMATTERS) {
+        if (modelId.startsWith(prefix))
+            return formatter;
+    }
+    return TITAN_FORMATTER;
+}
+export class BedrockTranslator {
+    requestTranslator;
+    responseTranslator;
+    messageTranslator;
+    streamTranslator;
+    constructor(requestTranslator, responseTranslator, messageTranslator, streamTranslator) {
+        this.requestTranslator = requestTranslator;
+        this.responseTranslator = responseTranslator;
+        this.messageTranslator = messageTranslator;
+        this.streamTranslator = streamTranslator;
+    }
+    static instance() {
+        const messageTranslator = new BedrockMessageTranslator();
+        const toolTranslator = new BedrockToolTranslator();
+        const requestTranslator = new BedrockRequestTranslator(messageTranslator, toolTranslator);
+        const responseTranslator = new BedrockResponseTranslator(messageTranslator);
+        const contentDeltaTranslator = new BedrockContentDeltaTranslator();
+        const messageStopTranslator = new BedrockMessageStopTranslator();
+        const streamTranslator = new BedrockStreamTranslator(contentDeltaTranslator, messageStopTranslator);
+        return new BedrockTranslator(requestTranslator, responseTranslator, messageTranslator, streamTranslator);
+    }
+    async fromHoloRequest(request) {
+        return this.requestTranslator.fromHolo(request);
+    }
+    async toHoloRequest(request) {
+        return this.requestTranslator.toHolo(request);
+    }
+    async fromHoloResponse(response) {
+        return this.responseTranslator.fromHolo(response);
+    }
+    async toHoloResponse(response) {
+        return this.responseTranslator.toHolo(response);
+    }
+    async fromHoloMessages(messages) {
+        return this.messageTranslator.fromHoloArray(messages);
+    }
+    async toHoloMessages(messages) {
+        return this.messageTranslator.toHoloArray(messages);
+    }
+    async fromHoloStreamChunks(chunks) {
+        return this.streamTranslator.fromHoloManyArray(chunks);
+    }
+    async fromHoloEmbedRequest(request) {
+        const formatter = getEmbedFormatter(request.model);
+        return {
+            modelId: request.model,
+            body: formatter.formatRequest(request.model, request.input),
+        };
+    }
+    async toHoloEmbedResponse(response) {
+        const model = response?.modelId ?? response?.model ?? '';
+        const body = response?.body ?? response;
+        const modelForLookup = response?.modelId ?? model;
+        const formatter = getEmbedFormatter(modelForLookup);
+        const { embeddings, usage } = formatter.parseResponse(body);
+        return { model, embeddings, usage };
+    }
+}
+//# sourceMappingURL=bedrock.translator.js.map

package/dist/bedrock.translator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bedrock.translator.js","sourceRoot":"","sources":["../src/bedrock.translator.ts"],"names":[],"mappings":"AAGA,OAAO,EACH,6BAA6B,EAC7B,4BAA4B,EAC5B,wBAAwB,EACxB,wBAAwB,EACxB,yBAAyB,EACzB,uBAAuB,EACvB,qBAAqB,GACxB,MAAM,eAAe,CAAC;AAYvB,MAAM,eAAe,GAAmB;IACpC,aAAa,CAAC,MAAM,EAAE,KAAK;QACvB,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;QAC5D,OAAO,EAAC,SAAS,EAAE,IAAI,EAAC,CAAC;IAC7B,CAAC;IACD,aAAa,CAAC,IAAI;QACd,MAAM,SAAS,GAAI,IAAY,CAAC,SAAiC,CAAC;QAClE,MAAM,eAAe,GAAI,IAAY,CAAC,mBAAyC,CAAC;QAChF,MAAM,MAAM,GAAgD,EAAC,UAAU,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,EAAC,CAAC;QACvG,IAAI,eAAe,IAAI,IAAI,EAAE,CAAC;YAC1B,MAAM,CAAC,KAAK,GAAG,EAAC,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,eAAe,EAAC,CAAC;QAClF,CAAC;QACD,OAAO,MAAM,CAAC;IAClB,CAAC;CACJ,CAAC;AAEF,MAAM,gBAAgB,GAAmB;IACrC,aAAa,CAAC,MAAM,EAAE,KAAK;QACvB,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;QACrD,OAAO,EAAC,KAAK,EAAE,UAAU,EAAE,iBAAiB,EAAC,CAAC;IAClD,CAAC;IACD,aAAa,CAAC,IAAI;QACd,MAAM,UAAU,GAAI,IAAY,CAAC,UAAoC,CAAC;QACtE,OAAO,EAAC,UAAU,EAAE,UAAU,IAAI,EAAE,EAAC,CAAC;IAC1C,CAAC;CACJ,CAAC;AAEF,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAyB;IACrD,CAAC,oBAAoB,EAAE,eAAe,CAAC;IACvC,CAAC,cAAc,EAAE,gBAAgB,CAAC;CACrC,CAAC,CAAC;AAEH,SAAS,iBAAiB,CAAC,OAAe;IACtC,KAAK,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,IAAI,gBAAgB,EAAE,CAAC;QACjD,IAAI,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC;YAAE,OAAO,SAAS,CAAC;IACrD,CAAC;IACD,OAAO,eAAe,CAAC;AAC3B,CAAC;AAED,MAAM,OAAO,iBAAiB;IAEL;IACA;IACA;IACA;IAJrB,YACqB,iBAA2C,EAC3C,kBAA6C,EAC7C,iBAA2C,EAC3C,gBAAyC;QAHzC,sBAAiB,GAAjB,iBAAiB,CAA0B;QAC3C,uBAAkB,GAAlB,kBAAkB,CAA2B;QAC7C,sBAAiB,GAAjB,iBAAiB,CAA0B;QAC3C,qBAAgB,GAAhB,gBAAgB,CAAyB;IAE9D,CAAC;IAED,MAAM,CAAC,QAAQ;QACX,MAAM,iBAAiB,GAAG,IAAI,wBAAwB,EAAE,CAAC;QACzD,MAAM,cAAc,GAAG,IAAI,qBAAqB,EAAE,CAAC;QAEnD,MAAM,iBAAiB,GAAG,IAAI,wBAAwB,CAAC,iBAAiB,EAAE,cAAc,CAAC,CAAC;QAC1F,MAAM,kBAAkB,GAAG,IAAI,yBAAyB,CAAC,iBAAiB,CAAC,CAAC;QAE5E,MAAM,sBAAsB,GAAG,IAAI,6BAA6B,EAAE,CAAC;QACnE,MAAM,qBAAqB,GAAG,IAAI,4BAA4B,EAAE,CAAC;QACjE,MAAM,gBAAgB,GAAG,IAAI,uBAAuB,CAAC,sBAAsB,EAAE,qBAAqB,CAAC,CAAC;QAEpG,OAAO,IAAI,iBAAiB,CAAC,iBAAiB,EAAE,kBAAkB,EAAE,iBAAiB,EAAE,gBAAgB,CAAC,CAAC;IAC7G,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,OAAoB;QACtC,OAAO,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IACpD,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,OAA+B;QAC/C,OAAO,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAClD,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,QAAsB;QACzC,OAAO,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,QAAiC;QAClD,OAAO,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IACpD,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,QAAuB;QAC1C,OAAO,IAAI,CAAC,iBAAiB,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;IAC1D,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,QAAmB;QACpC,OAAO,IAAI,CAAC,iBAAiB,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;IACxD,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAC,MAAyB;QAChD,OAAO,IAAI,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAC3D,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAC,OAAwB;QAC/C,MAAM,SAAS,GAAG,iBAAiB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACnD,OAAO;YACH,OAAO,EAAE,OAAO,CAAC,KAAK;YACtB,IAAI,EAAE,SAAS,CAAC,aAAa,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,KAAK,CAAC;SAC9D,CAAC;IACN,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,QAAa;QACnC,MAAM,KAAK,GAAG,QAAQ,EAAE,OAAO,IAAI,QAAQ,EAAE,KAAK,IAAI,EAAE,CAAC;QACzD,MAAM,IAAI,GAAG,QAAQ,EAAE,IAAI,IAAI,QAAQ,CAAC;QACxC,MAAM,cAAc,GAAG,QAAQ,EAAE,OAAO,IAAI,KAAK,CAAC;QAClD,MAAM,SAAS,GAAG,iBAAiB,CAAC,cAAc,CAAC,CAAC;QACpD,MAAM,EAAC,UAAU,EAAE,KAAK,EAAC,GAAG,SAAS,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;QAC1D,OAAO,EAAC,KAAK,EAAE,UAAU,EAAE,KAAK,EAAC,CAAC;IACtC,CAAC;CACJ"}

package/dist/bedrock.wire.adapter.d.ts

@@ -0,0 +1,9 @@
+import { BaseWireAdapter } from '@holokai/holo-sdk/provider';
+import type { ProviderEvent, WireChunk } from '@holokai/holo-types/provider';
+export declare class BedrockWireAdapter extends BaseWireAdapter {
+    formatWire(data: any): string;
+    protected onDoneStreaming(ev: Extract<ProviderEvent, {
+        type: 'done';
+    }>): Promise<WireChunk[]>;
+}
+//# sourceMappingURL=bedrock.wire.adapter.d.ts.map

package/dist/bedrock.wire.adapter.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"bedrock.wire.adapter.d.ts","sourceRoot":"","sources":["../src/bedrock.wire.adapter.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,eAAe,EAAC,MAAM,4BAA4B,CAAC;AAC3D,OAAO,KAAK,EAAC,aAAa,EAAE,SAAS,EAAC,MAAM,8BAA8B,CAAC;AAE3E,qBAAa,kBAAmB,SAAQ,eAAe;IACnD,UAAU,CAAC,IAAI,EAAE,GAAG,GAAG,MAAM;cAIJ,eAAe,CAAC,EAAE,EAAE,OAAO,CAAC,aAAa,EAAE;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;CAQ/G"}

package/dist/bedrock.wire.adapter.js

@@ -0,0 +1,15 @@
+import { BaseWireAdapter } from '@holokai/holo-sdk/provider';
+export class BedrockWireAdapter extends BaseWireAdapter {
+    formatWire(data) {
+        return `data: ${JSON.stringify(data)}\n\n`;
+    }
+    async onDoneStreaming(ev) {
+        const chunks = [];
+        if (ev.message) {
+            chunks.push(await this.chunkify(ev, async () => this.formatWire(ev.message)));
+        }
+        chunks.push(await this.chunkify(ev, async () => 'data: [DONE]\n\n', true, { fullText: ev.text }));
+        return chunks;
+    }
+}
+//# sourceMappingURL=bedrock.wire.adapter.js.map

package/dist/bedrock.wire.adapter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bedrock.wire.adapter.js","sourceRoot":"","sources":["../src/bedrock.wire.adapter.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,eAAe,EAAC,MAAM,4BAA4B,CAAC;AAG3D,MAAM,OAAO,kBAAmB,SAAQ,eAAe;IACnD,UAAU,CAAC,IAAS;QAChB,OAAO,SAAS,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC;IAC/C,CAAC;IAEkB,KAAK,CAAC,eAAe,CAAC,EAA4C;QACjF,MAAM,MAAM,GAAgB,EAAE,CAAC;QAC/B,IAAI,EAAE,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QAClF,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,KAAK,IAAI,EAAE,CAAC,kBAAkB,EAAE,IAAI,EAAE,EAAC,QAAQ,EAAE,EAAE,CAAC,IAAI,EAAC,CAAC,CAAC,CAAC;QAChG,OAAO,MAAM,CAAC;IAClB,CAAC;CACJ"}

package/dist/index.d.ts

@@ -0,0 +1,14 @@
+import { BedrockProviderPlugin } from './plugin';
+export * from './translators';
+export * from './types';
+export * from './manifest';
+export * from './bedrock.auditor';
+export * from './bedrock.provider';
+export * from './bedrock.translator';
+export * from './bedrock.wire.adapter';
+export * from './bedrock.response.factory';
+export * from './bedrock.pricing';
+export * from './plugin';
+declare const _default: BedrockProviderPlugin;
+export default _default;
+//# 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,EAAC,qBAAqB,EAAC,MAAM,UAAU,CAAC;AAE/C,cAAc,eAAe,CAAC;AAC9B,cAAc,SAAS,CAAC;AACxB,cAAc,YAAY,CAAC;AAC3B,cAAc,mBAAmB,CAAC;AAClC,cAAc,oBAAoB,CAAC;AACnC,cAAc,sBAAsB,CAAC;AACrC,cAAc,wBAAwB,CAAC;AACvC,cAAc,4BAA4B,CAAC;AAC3C,cAAc,mBAAmB,CAAC;AAClC,cAAc,UAAU,CAAC;;AAEzB,wBAA2C"}

package/dist/index.js

@@ -0,0 +1,13 @@
+import { BedrockProviderPlugin } from './plugin';
+export * from './translators';
+export * from './types';
+export * from './manifest';
+export * from './bedrock.auditor';
+export * from './bedrock.provider';
+export * from './bedrock.translator';
+export * from './bedrock.wire.adapter';
+export * from './bedrock.response.factory';
+export * from './bedrock.pricing';
+export * from './plugin';
+export default new BedrockProviderPlugin();
+//# 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,EAAC,qBAAqB,EAAC,MAAM,UAAU,CAAC;AAE/C,cAAc,eAAe,CAAC;AAC9B,cAAc,SAAS,CAAC;AACxB,cAAc,YAAY,CAAC;AAC3B,cAAc,mBAAmB,CAAC;AAClC,cAAc,oBAAoB,CAAC;AACnC,cAAc,sBAAsB,CAAC;AACrC,cAAc,wBAAwB,CAAC;AACvC,cAAc,4BAA4B,CAAC;AAC3C,cAAc,mBAAmB,CAAC;AAClC,cAAc,UAAU,CAAC;AAEzB,eAAe,IAAI,qBAAqB,EAAE,CAAC"}

package/dist/manifest.d.ts

@@ -0,0 +1,3 @@
+import type { PluginManifest } from '@holokai/holo-types/plugin';
+export declare const manifest: PluginManifest;
+//# sourceMappingURL=manifest.d.ts.map

package/dist/manifest.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"manifest.d.ts","sourceRoot":"","sources":["../src/manifest.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAC,cAAc,EAAC,MAAM,4BAA4B,CAAC;AAI/D,eAAO,MAAM,QAAQ,EAAE,cAOtB,CAAC"}

package/dist/manifest.js

@@ -0,0 +1,11 @@
+import { PluginType } from '@holokai/holo-types/plugin';
+import { VERSION } from './version';
+export const manifest = {
+    name: '@holokai/provider-bedrock',
+    version: VERSION,
+    pluginType: PluginType.PROVIDER,
+    family: 'bedrock',
+    displayName: 'Amazon Bedrock Provider',
+    description: 'Amazon Bedrock provider plugin for Holo.',
+};
+//# sourceMappingURL=manifest.js.map

package/dist/manifest.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"manifest.js","sourceRoot":"","sources":["../src/manifest.ts"],"names":[],"mappings":"AACA,OAAO,EAAC,UAAU,EAAC,MAAM,4BAA4B,CAAC;AACtD,OAAO,EAAC,OAAO,EAAC,MAAM,WAAW,CAAC;AAElC,MAAM,CAAC,MAAM,QAAQ,GAAmB;IACpC,IAAI,EAAE,2BAA2B;IACjC,OAAO,EAAE,OAAO;IAChB,UAAU,EAAE,UAAU,CAAC,QAAQ;IAC/B,MAAM,EAAE,SAAS;IACjB,WAAW,EAAE,yBAAyB;IACtC,WAAW,EAAE,0CAA0C;CAC1D,CAAC"}

package/dist/plugin.d.ts

@@ -0,0 +1,52 @@
+import { BasePlugin } from '@holokai/holo-sdk/plugin';
+import type { IProviderPlugin, PluginContext, PluginPricingSheet, PluginSchema } from '@holokai/holo-types/plugin';
+import type { PricingSheetModel } from '@holokai/holo-types/entities';
+import { ProtocolCapability } from '@holokai/holo-types/entities';
+import type { IProvider, IWireAdapter, ProviderCapabilities, WireAdapterParams } from '@holokai/holo-types/provider';
+import type { RouteDefinition } from '@holokai/holo-types/routing';
+export declare const BedrockProtocols: {
+    readonly CONVERSE: "bedrock.converse";
+    readonly EMBED: "bedrock.embed";
+    readonly MODELS: "bedrock.models";
+};
+export type BedrockProtocols = typeof BedrockProtocols[keyof typeof BedrockProtocols];
+export declare class BedrockProviderPlugin extends BasePlugin implements IProviderPlugin {
+    manifest: import("@holokai/holo-types/plugin").PluginManifest;
+    translator: import("@holokai/holo-types/provider").IProviderTranslator;
+    defaultRouteHandler: "request";
+    protocols: {
+        readonly CONVERSE: "bedrock.converse";
+        readonly EMBED: "bedrock.embed";
+        readonly MODELS: "bedrock.models";
+    };
+    defaultProtocol: "bedrock.converse";
+    createProvider(id: string, name: string, config: any): Promise<IProvider>;
+    createWireAdapter(params: WireAdapterParams): Promise<IWireAdapter>;
+    getProtocolByCapability(capability: ProtocolCapability): string | undefined;
+    getSchema(): PluginSchema;
+    getTestModels(): {
+        chat: string;
+        embed: string;
+    };
+    getCapabilities(): ProviderCapabilities;
+    getRoutes(): RouteDefinition[];
+    getPricingSheets(): Map<string, PluginPricingSheet>;
+    getPricingModelIds(): import("@holokai/holo-types/plugin").PricingModelId[];
+    getDefaultPricing(): PluginPricingSheet;
+    protected calculateExtraCosts(tokens: Record<string, number>, pricing: PricingSheetModel): {
+        total: number;
+        detail: {
+            cache_read: {
+                tokens: number;
+                cost: number;
+            };
+            cache_write: {
+                tokens: number;
+                cost: number;
+            };
+        };
+    };
+    protected onInitialize(_context: PluginContext): Promise<void>;
+    protected onDestroy(): Promise<void>;
+}
+//# sourceMappingURL=plugin.d.ts.map

package/dist/plugin.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"plugin.d.ts","sourceRoot":"","sources":["../src/plugin.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,UAAU,EAA0B,MAAM,0BAA0B,CAAC;AAC7E,OAAO,KAAK,EAAC,eAAe,EAAE,aAAa,EAAE,kBAAkB,EAAE,YAAY,EAAC,MAAM,4BAA4B,CAAC;AACjH,OAAO,KAAK,EAAC,iBAAiB,EAAC,MAAM,8BAA8B,CAAC;AACpE,OAAO,EAAC,kBAAkB,EAAC,MAAM,8BAA8B,CAAC;AAEhE,OAAO,KAAK,EAAC,SAAS,EAAE,YAAY,EAAE,oBAAoB,EAAE,iBAAiB,EAAC,MAAM,8BAA8B,CAAC;AAGnH,OAAO,KAAK,EAAC,eAAe,EAAC,MAAM,6BAA6B,CAAC;AAKjE,eAAO,MAAM,gBAAgB;;;;CAInB,CAAC;AAEX,MAAM,MAAM,gBAAgB,GAAG,OAAO,gBAAgB,CAAC,MAAM,OAAO,gBAAgB,CAAC,CAAC;AAEtF,qBAAa,qBAAsB,SAAQ,UAAW,YAAW,eAAe;IAC5E,QAAQ,sDAAY;IACpB,UAAU,6DAAgC;IAC1C,mBAAmB,YAAwB;IAC3C,SAAS;;;;MAAoB;IAC7B,eAAe,qBAA6B;IAEtC,cAAc,CAAC,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,GAAG,OAAO,CAAC,SAAS,CAAC;IAIzE,iBAAiB,CAAC,MAAM,EAAE,iBAAiB,GAAG,OAAO,CAAC,YAAY,CAAC;IAIzE,uBAAuB,CAAC,UAAU,EAAE,kBAAkB,GAAG,MAAM,GAAG,SAAS;IAK3E,SAAS,IAAI,YAAY;IAsCzB,aAAa;;;;IAIb,eAAe,IAAI,oBAAoB;IAUvC,SAAS,IAAI,eAAe,EAAE;IAoC9B,gBAAgB,IAAI,GAAG,CAAC,MAAM,EAAE,kBAAkB,CAAC;IAInD,kBAAkB;IAIlB,iBAAiB,IAAI,kBAAkB;IAQvC,SAAS,CAAC,mBAAmB,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,OAAO,EAAE,iBAAiB;;;;;;;;;;;;;IAcxF,SAAS,CAAC,YAAY,CAAC,QAAQ,EAAE,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC;IAI9D,SAAS,CAAC,SAAS,IAAI,OAAO,CAAC,IAAI,CAAC;CAGvC"}

package/dist/plugin.js

@@ -0,0 +1,145 @@
+import { BasePlugin, normalizePricingDataset } from '@holokai/holo-sdk/plugin';
+import { ProtocolCapability } from '@holokai/holo-types/entities';
+import { manifest } from './manifest';
+import { BedrockProvider } from './bedrock.provider';
+import { BedrockWireAdapter } from './bedrock.wire.adapter';
+import { RouteHandler } from '@holokai/holo-types/routing';
+import { BedrockTranslator } from './bedrock.translator';
+import { BEDROCK_PRICING_DATASET } from './bedrock.pricing';
+export const BedrockProtocols = {
+    CONVERSE: 'bedrock.converse',
+    EMBED: 'bedrock.embed',
+    MODELS: 'bedrock.models',
+};
+export class BedrockProviderPlugin extends BasePlugin {
+    manifest = manifest;
+    translator = BedrockTranslator.instance();
+    defaultRouteHandler = RouteHandler.REQUEST;
+    protocols = BedrockProtocols;
+    defaultProtocol = BedrockProtocols.CONVERSE;
+    async createProvider(id, name, config) {
+        return new BedrockProvider(id, name, this, config);
+    }
+    async createWireAdapter(params) {
+        return new BedrockWireAdapter(params.requestId, params.isStreaming);
+    }
+    getProtocolByCapability(capability) {
+        const route = this.getRoutes().find(r => r.protocol.capability === capability);
+        return route?.protocol.name;
+    }
+    getSchema() {
+        return {
+            connection: {
+                type: 'object',
+                properties: {
+                    region: { type: 'string', title: 'AWS Region', default: 'us-east-1' },
+                    credentials: {
+                        type: 'object',
+                        title: 'AWS Credentials',
+                        properties: {
+                            accessKeyId: { type: 'string', title: 'Access Key ID' },
+                            secretAccessKey: { type: 'string', title: 'Secret Access Key' },
+                            sessionToken: { type: 'string', title: 'Session Token' },
+                        },
+                        required: ['accessKeyId', 'secretAccessKey'],
+                    },
+                    endpoint: { type: 'string', title: 'Endpoint Override', format: 'uri' },
+                    maxAttempts: { type: 'integer', title: 'Max Retry Attempts', minimum: 1, default: 3 },
+                    requestTimeout: { type: 'integer', title: 'Request Timeout (ms)', minimum: 0 },
+                    connectionTimeout: { type: 'integer', title: 'Connection Timeout (ms)', minimum: 0 },
+                    customUserAgent: { type: 'string', title: 'Custom User Agent' },
+                    tls: { type: 'boolean', title: 'TLS', default: true },
+                },
+                required: ['region'],
+                encrypted: ['credentials.secretAccessKey'],
+            },
+            parameters: {
+                type: 'object',
+                properties: {
+                    temperature: { type: 'number', title: 'Temperature', minimum: 0, maximum: 1, default: 1 },
+                    max_tokens: { type: 'integer', title: 'Max Tokens', minimum: 1, maximum: 200000 },
+                    top_p: { type: 'number', title: 'Top P', minimum: 0, maximum: 1 },
+                    stop_sequences: { type: 'array', title: 'Stop Sequences', items: { type: 'string' } },
+                },
+            },
+        };
+    }
+    getTestModels() {
+        return { chat: 'amazon.nova-micro-v1:0', embed: 'amazon.titan-embed-text-v1' };
+    }
+    getCapabilities() {
+        return {
+            streaming: true,
+            tools: true,
+            vision: true,
+            functionCalling: true,
+            maxTokens: 200000,
+        };
+    }
+    getRoutes() {
+        return [
+            {
+                paths: ['/converse'],
+                method: 'POST',
+                handler: RouteHandler.REQUEST,
+                protocol: {
+                    name: BedrockProtocols.CONVERSE,
+                    capability: ProtocolCapability.CHAT,
+                    streamEventSequence: {
+                        ordered: ['content_delta', 'message_delta', 'message_stop'],
+                        repeatable: ['content_delta'],
+                    },
+                },
+            },
+            {
+                paths: ['/embed'],
+                method: 'POST',
+                handler: RouteHandler.REQUEST,
+                protocol: {
+                    name: BedrockProtocols.EMBED,
+                    capability: ProtocolCapability.EMBED,
+                },
+            },
+            {
+                paths: ['/models'],
+                method: 'GET',
+                handler: RouteHandler.MODELS,
+                protocol: {
+                    name: BedrockProtocols.MODELS,
+                    capability: ProtocolCapability.MODELS,
+                },
+            },
+        ];
+    }
+    getPricingSheets() {
+        return normalizePricingDataset(BEDROCK_PRICING_DATASET);
+    }
+    getPricingModelIds() {
+        return BEDROCK_PRICING_DATASET.model_ids;
+    }
+    getDefaultPricing() {
+        const sheets = this.getPricingSheets();
+        const sorted = Array.from(sheets.values()).sort((a, b) => b.effective_from.localeCompare(a.effective_from));
+        return sorted[0];
+    }
+    calculateExtraCosts(tokens, pricing) {
+        const cacheReadRate = pricing.token_costs?.cache_read ?? Number(pricing.cache_read_cost ?? 0);
+        const cacheWriteRate = pricing.token_costs?.cache_write ?? Number(pricing.cache_write_cost ?? 0);
+        const cacheReadCost = (tokens.cache_read ?? 0) * cacheReadRate;
+        const cacheWriteCost = (tokens.cache_write ?? 0) * cacheWriteRate;
+        return {
+            total: cacheReadCost + cacheWriteCost,
+            detail: {
+                cache_read: { tokens: tokens.cache_read ?? 0, cost: cacheReadCost },
+                cache_write: { tokens: tokens.cache_write ?? 0, cost: cacheWriteCost },
+            }
+        };
+    }
+    onInitialize(_context) {
+        return Promise.resolve();
+    }
+    onDestroy() {
+        return Promise.resolve();
+    }
+}
+//# sourceMappingURL=plugin.js.map

package/dist/plugin.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"plugin.js","sourceRoot":"","sources":["../src/plugin.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,UAAU,EAAE,uBAAuB,EAAC,MAAM,0BAA0B,CAAC;AAG7E,OAAO,EAAC,kBAAkB,EAAC,MAAM,8BAA8B,CAAC;AAChE,OAAO,EAAC,QAAQ,EAAC,MAAM,YAAY,CAAC;AAEpC,OAAO,EAAC,eAAe,EAAC,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAC,kBAAkB,EAAC,MAAM,wBAAwB,CAAC;AAE1D,OAAO,EAAC,YAAY,EAAC,MAAM,6BAA6B,CAAC;AACzD,OAAO,EAAC,iBAAiB,EAAC,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAC,uBAAuB,EAAC,MAAM,mBAAmB,CAAC;AAE1D,MAAM,CAAC,MAAM,gBAAgB,GAAG;IAC5B,QAAQ,EAAE,kBAAkB;IAC5B,KAAK,EAAE,eAAe;IACtB,MAAM,EAAE,gBAAgB;CAClB,CAAC;AAIX,MAAM,OAAO,qBAAsB,SAAQ,UAAU;IACjD,QAAQ,GAAG,QAAQ,CAAC;IACpB,UAAU,GAAG,iBAAiB,CAAC,QAAQ,EAAE,CAAC;IAC1C,mBAAmB,GAAG,YAAY,CAAC,OAAO,CAAC;IAC3C,SAAS,GAAG,gBAAgB,CAAC;IAC7B,eAAe,GAAG,gBAAgB,CAAC,QAAQ,CAAC;IAE5C,KAAK,CAAC,cAAc,CAAC,EAAU,EAAE,IAAY,EAAE,MAAW;QACtD,OAAO,IAAI,eAAe,CAAC,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;IACvD,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,MAAyB;QAC7C,OAAO,IAAI,kBAAkB,CAAC,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC;IACxE,CAAC;IAED,uBAAuB,CAAC,UAA8B;QAClD,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,UAAU,KAAK,UAAU,CAAC,CAAC;QAC/E,OAAO,KAAK,EAAE,QAAQ,CAAC,IAAI,CAAC;IAChC,CAAC;IAED,SAAS;QACL,OAAO;YACH,UAAU,EAAE;gBACR,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE;oBACR,MAAM,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,YAAY,EAAE,OAAO,EAAE,WAAW,EAAC;oBACnE,WAAW,EAAE;wBACT,IAAI,EAAE,QAAQ;wBACd,KAAK,EAAE,iBAAiB;wBACxB,UAAU,EAAE;4BACR,WAAW,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,eAAe,EAAC;4BACrD,eAAe,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,mBAAmB,EAAC;4BAC7D,YAAY,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,eAAe,EAAC;yBACzD;wBACD,QAAQ,EAAE,CAAC,aAAa,EAAE,iBAAiB,CAAC;qBAC/C;oBACD,QAAQ,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,mBAAmB,EAAE,MAAM,EAAE,KAAK,EAAC;oBACrE,WAAW,EAAE,EAAC,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,oBAAoB,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAC;oBACnF,cAAc,EAAE,EAAC,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,sBAAsB,EAAE,OAAO,EAAE,CAAC,EAAC;oBAC5E,iBAAiB,EAAE,EAAC,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,yBAAyB,EAAE,OAAO,EAAE,CAAC,EAAC;oBAClF,eAAe,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,mBAAmB,EAAC;oBAC7D,GAAG,EAAE,EAAC,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAC;iBACtD;gBACD,QAAQ,EAAE,CAAC,QAAQ,CAAC;gBACpB,SAAS,EAAE,CAAC,6BAA6B,CAAC;aAC7C;YACD,UAAU,EAAE;gBACR,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE;oBACR,WAAW,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,aAAa,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAC;oBACvF,UAAU,EAAE,EAAC,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,YAAY,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,MAAM,EAAC;oBAC/E,KAAK,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAC;oBAC/D,cAAc,EAAE,EAAC,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,gBAAgB,EAAE,KAAK,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAC,EAAC;iBACpF;aACJ;SACJ,CAAC;IACN,CAAC;IAED,aAAa;QACT,OAAO,EAAC,IAAI,EAAE,wBAAwB,EAAE,KAAK,EAAE,4BAA4B,EAAC,CAAC;IACjF,CAAC;IAED,eAAe;QACX,OAAO;YACH,SAAS,EAAE,IAAI;YACf,KAAK,EAAE,IAAI;YACX,MAAM,EAAE,IAAI;YACZ,eAAe,EAAE,IAAI;YACrB,SAAS,EAAE,MAAM;SACpB,CAAC;IACN,CAAC;IAED,SAAS;QACL,OAAO;YACH;gBACI,KAAK,EAAE,CAAC,WAAW,CAAC;gBACpB,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,YAAY,CAAC,OAAO;gBAC7B,QAAQ,EAAE;oBACN,IAAI,EAAE,gBAAgB,CAAC,QAAQ;oBAC/B,UAAU,EAAE,kBAAkB,CAAC,IAAI;oBACnC,mBAAmB,EAAE;wBACjB,OAAO,EAAE,CAAC,eAAe,EAAE,eAAe,EAAE,cAAc,CAAC;wBAC3D,UAAU,EAAE,CAAC,eAAe,CAAC;qBAChC;iBACJ;aACJ;YACD;gBACI,KAAK,EAAE,CAAC,QAAQ,CAAC;gBACjB,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,YAAY,CAAC,OAAO;gBAC7B,QAAQ,EAAE;oBACN,IAAI,EAAE,gBAAgB,CAAC,KAAK;oBAC5B,UAAU,EAAE,kBAAkB,CAAC,KAAK;iBACvC;aACJ;YACD;gBACI,KAAK,EAAE,CAAC,SAAS,CAAC;gBAClB,MAAM,EAAE,KAAK;gBACb,OAAO,EAAE,YAAY,CAAC,MAAM;gBAC5B,QAAQ,EAAE;oBACN,IAAI,EAAE,gBAAgB,CAAC,MAAM;oBAC7B,UAAU,EAAE,kBAAkB,CAAC,MAAM;iBACxC;aACJ;SACJ,CAAC;IACN,CAAC;IAED,gBAAgB;QACZ,OAAO,uBAAuB,CAAC,uBAAuB,CAAC,CAAC;IAC5D,CAAC;IAED,kBAAkB;QACd,OAAO,uBAAuB,CAAC,SAAS,CAAC;IAC7C,CAAC;IAED,iBAAiB;QACb,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACvC,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAC3C,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC,CAAC,cAAc,CAAC,CAC7D,CAAC;QACF,OAAO,MAAM,CAAC,CAAC,CAAC,CAAC;IACrB,CAAC;IAES,mBAAmB,CAAC,MAA8B,EAAE,OAA0B;QACpF,MAAM,aAAa,GAAG,OAAO,CAAC,WAAW,EAAE,UAAU,IAAI,MAAM,CAAC,OAAO,CAAC,eAAe,IAAI,CAAC,CAAC,CAAC;QAC9F,MAAM,cAAc,GAAG,OAAO,CAAC,WAAW,EAAE,WAAW,IAAI,MAAM,CAAC,OAAO,CAAC,gBAAgB,IAAI,CAAC,CAAC,CAAC;QACjG,MAAM,aAAa,GAAG,CAAC,MAAM,CAAC,UAAU,IAAI,CAAC,CAAC,GAAG,aAAa,CAAC;QAC/D,MAAM,cAAc,GAAG,CAAC,MAAM,CAAC,WAAW,IAAI,CAAC,CAAC,GAAG,cAAc,CAAC;QAClE,OAAO;YACH,KAAK,EAAE,aAAa,GAAG,cAAc;YACrC,MAAM,EAAE;gBACJ,UAAU,EAAE,EAAC,MAAM,EAAE,MAAM,CAAC,UAAU,IAAI,CAAC,EAAE,IAAI,EAAE,aAAa,EAAC;gBACjE,WAAW,EAAE,EAAC,MAAM,EAAE,MAAM,CAAC,WAAW,IAAI,CAAC,EAAE,IAAI,EAAE,cAAc,EAAC;aACvE;SACJ,CAAC;IACN,CAAC;IAES,YAAY,CAAC,QAAuB;QAC1C,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;IAC7B,CAAC;IAES,SAAS;QACf,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;IAC7B,CAAC;CACJ"}

package/dist/translators/bedrock.message.translator.d.ts

@@ -0,0 +1,10 @@
+import type { HoloMessage } from '@holokai/holo-types/holo';
+import { BaseTranslator } from '@holokai/holo-sdk/provider';
+import type { Message } from '@aws-sdk/client-bedrock-runtime';
+export declare class BedrockMessageTranslator extends BaseTranslator<HoloMessage, Message> {
+    protected holoDefaults: Partial<HoloMessage>;
+    protected providerDefaults: Partial<Message>;
+    protected fromHoloImpl(source: HoloMessage): Promise<Partial<Message>>;
+    protected toHoloImpl(source: Message): Promise<Partial<HoloMessage>>;
+}
+//# sourceMappingURL=bedrock.message.translator.d.ts.map

package/dist/translators/bedrock.message.translator.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"bedrock.message.translator.d.ts","sourceRoot":"","sources":["../../src/translators/bedrock.message.translator.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAgC,WAAW,EAAC,MAAM,0BAA0B,CAAC;AACzF,OAAO,EAAC,cAAc,EAAC,MAAM,4BAA4B,CAAC;AAC1D,OAAO,KAAK,EAAe,OAAO,EAAC,MAAM,iCAAiC,CAAC;AAE3E,qBAAa,wBAAyB,SAAQ,cAAc,CAAC,WAAW,EAAE,OAAO,CAAC;IAC9E,SAAS,CAAC,YAAY,EAAE,OAAO,CAAC,WAAW,CAAC,CAAM;IAClD,SAAS,CAAC,gBAAgB,EAAE,OAAO,CAAC,OAAO,CAAC,CAAM;cAElC,YAAY,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;cAmE5D,UAAU,CAAC,MAAM,EAAE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;CA2D7E"}

package/dist/translators/bedrock.message.translator.js

@@ -0,0 +1,142 @@
+import { pickDefined } from '@holokai/holo-sdk';
+import { BaseTranslator } from '@holokai/holo-sdk/provider';
+export class BedrockMessageTranslator extends BaseTranslator {
+    holoDefaults = {};
+    providerDefaults = {};
+    async fromHoloImpl(source) {
+        const content = [];
+        if (typeof source.content === 'string') {
+            content.push({ text: source.content });
+        }
+        else if (Array.isArray(source.content)) {
+            for (const block of source.content) {
+                switch (block.type) {
+                    case 'text':
+                        content.push({ text: block.text });
+                        break;
+                    case 'image':
+                        if (block.data) {
+                            content.push({
+                                image: {
+                                    format: mimeToFormat(block.mime),
+                                    source: { bytes: Buffer.from(block.data, 'base64') },
+                                }
+                            });
+                        }
+                        break;
+                    case 'tool_call':
+                        content.push({
+                            toolUse: {
+                                toolUseId: block.id ?? `tool_${Date.now()}`,
+                                name: block.name,
+                                input: block.arguments,
+                            }
+                        });
+                        break;
+                    case 'tool_result':
+                        content.push({
+                            toolResult: {
+                                toolUseId: block.tool_call_id,
+                                content: typeof block.content === 'string'
+                                    ? [{ text: block.content }]
+                                    : block.content?.map((c) => c.type === 'text' ? { text: c.text } : { text: JSON.stringify(c) }) ?? [],
+                                status: block.is_error ? 'error' : 'success',
+                            }
+                        });
+                        break;
+                }
+            }
+        }
+        if (source.tool_calls?.length) {
+            for (const tc of source.tool_calls) {
+                content.push({
+                    toolUse: {
+                        toolUseId: tc.id ?? `tool_${Date.now()}`,
+                        name: tc.function.name,
+                        input: tc.function.arguments,
+                    }
+                });
+            }
+        }
+        const role = source.role === 'tool' ? 'user' : source.role;
+        return pickDefined({
+            role: role,
+            content: content.length > 0 ? content : undefined,
+        });
+    }
+    async toHoloImpl(source) {
+        if (!source.content?.length) {
+            return pickDefined({ role: source.role });
+        }
+        const holoContent = [];
+        const toolCalls = [];
+        for (const block of source.content) {
+            if (block.text) {
+                holoContent.push({ type: 'text', text: block.text });
+            }
+            else if (block.toolUse) {
+                const id = block.toolUse.toolUseId;
+                const name = block.toolUse.name;
+                const args = (block.toolUse.input ?? {});
+                if (id) {
+                    toolCalls.push({ id, type: 'function', function: { name, arguments: args } });
+                    holoContent.push({ type: 'tool_call', id, name, arguments: args });
+                }
+                else {
+                    toolCalls.push({ type: 'function', function: { name, arguments: args } });
+                    holoContent.push({ type: 'tool_call', name, arguments: args });
+                }
+            }
+            else if (block.toolResult) {
+                holoContent.push({
+                    type: 'tool_result',
+                    tool_call_id: block.toolResult.toolUseId,
+                    content: block.toolResult.content?.map(c => c.text ? { type: 'text', text: c.text } : {
+                        type: 'text',
+                        text: JSON.stringify(c)
+                    }) ?? [],
+                    is_error: block.toolResult.status === 'error',
+                });
+            }
+            else if (block.image) {
+                const imgBlock = { type: 'image', source_type: 'base64' };
+                if (block.image.source?.bytes) {
+                    imgBlock.data = Buffer.from(block.image.source.bytes).toString('base64');
+                }
+                const mime = formatToMime(block.image.format);
+                if (mime)
+                    imgBlock.mime = mime;
+                holoContent.push(imgBlock);
+            }
+        }
+        if (holoContent.length === 1 && holoContent[0].type === 'text') {
+            return pickDefined({
+                role: source.role,
+                content: holoContent[0].text,
+                ...(toolCalls.length ? { tool_calls: toolCalls } : {}),
+            });
+        }
+        return pickDefined({
+            role: source.role,
+            content: holoContent,
+            ...(toolCalls.length ? { tool_calls: toolCalls } : {}),
+        });
+    }
+}
+function mimeToFormat(mime) {
+    if (!mime)
+        return undefined;
+    if (mime.includes('png'))
+        return 'png';
+    if (mime.includes('gif'))
+        return 'gif';
+    if (mime.includes('webp'))
+        return 'webp';
+    return 'jpeg';
+}
+function formatToMime(format) {
+    if (!format)
+        return undefined;
+    return `image/${format}`;
+}
+//# sourceMappingURL=bedrock.message.translator.js.map

package/dist/translators/bedrock.message.translator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bedrock.message.translator.js","sourceRoot":"","sources":["../../src/translators/bedrock.message.translator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,WAAW,EAAC,MAAM,mBAAmB,CAAC;AAE9C,OAAO,EAAC,cAAc,EAAC,MAAM,4BAA4B,CAAC;AAG1D,MAAM,OAAO,wBAAyB,SAAQ,cAAoC;IACpE,YAAY,GAAyB,EAAE,CAAC;IACxC,gBAAgB,GAAqB,EAAE,CAAC;IAExC,KAAK,CAAC,YAAY,CAAC,MAAmB;QAC5C,MAAM,OAAO,GAAmB,EAAE,CAAC;QAEnC,IAAI,OAAO,MAAM,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;YACrC,OAAO,CAAC,IAAI,CAAC,EAAC,IAAI,EAAE,MAAM,CAAC,OAAO,EAAC,CAAC,CAAC;QACzC,CAAC;aAAM,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;YACvC,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACjC,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;oBACjB,KAAK,MAAM;wBACP,OAAO,CAAC,IAAI,CAAC,EAAC,IAAI,EAAE,KAAK,CAAC,IAAI,EAAC,CAAC,CAAC;wBACjC,MAAM;oBACV,KAAK,OAAO;wBACR,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;4BACb,OAAO,CAAC,IAAI,CAAC;gCACT,KAAK,EAAE;oCACH,MAAM,EAAE,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC;oCAChC,MAAM,EAAE,EAAC,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,EAAC;iCACrD;6BACJ,CAAC,CAAC;wBACP,CAAC;wBACD,MAAM;oBACV,KAAK,WAAW;wBACZ,OAAO,CAAC,IAAI,CAAC;4BACT,OAAO,EAAE;gCACL,SAAS,EAAE,KAAK,CAAC,EAAE,IAAI,QAAQ,IAAI,CAAC,GAAG,EAAE,EAAE;gCAC3C,IAAI,EAAE,KAAK,CAAC,IAAI;gCAChB,KAAK,EAAE,KAAK,CAAC,SAAgC;6BAChD;yBACJ,CAAC,CAAC;wBACH,MAAM;oBACV,KAAK,aAAa;wBACd,OAAO,CAAC,IAAI,CAAC;4BACT,UAAU,EAAE;gCACR,SAAS,EAAE,KAAK,CAAC,YAAY;gCAC7B,OAAO,EAAE,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ;oCACtC,CAAC,CAAC,CAAC,EAAC,IAAI,EAAE,KAAK,CAAC,OAAO,EAAC,CAAC;oCACzB,CAAC,CAAE,KAAK,CAAC,OAAyB,EAAE,GAAG,CACvC,CAAC,CAAc,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,EAAC,IAAI,EAAE,CAAC,CAAC,IAAI,EAAC,CAAC,CAAC,CAAC,EAAC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAC,CACrF,IAAI,EAAE;gCACP,MAAM,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;6BAC/C;yBACJ,CAAC,CAAC;wBACH,MAAM;gBACd,CAAC;YACL,CAAC;QACL,CAAC;QAED,IAAI,MAAM,CAAC,UAAU,EAAE,MAAM,EAAE,CAAC;YAC5B,KAAK,MAAM,EAAE,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;gBACjC,OAAO,CAAC,IAAI,CAAC;oBACT,OAAO,EAAE;wBACL,SAAS,EAAE,EAAE,CAAC,EAAE,IAAI,QAAQ,IAAI,CAAC,GAAG,EAAE,EAAE;wBACxC,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI;wBACtB,KAAK,EAAE,EAAE,CAAC,QAAQ,CAAC,SAAgC;qBACtD;iBACJ,CAAC,CAAC;YACP,CAAC;QACL,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC;QAE3D,OAAO,WAAW,CAAC;YACf,IAAI,EAAE,IAA4B;YAClC,OAAO,EAAE,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;SACpD,CAAqB,CAAC;IAC3B,CAAC;IAES,KAAK,CAAC,UAAU,CAAC,MAAe;QACtC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC;YAC1B,OAAO,WAAW,CAAC,EAAC,IAAI,EAAE,MAAM,CAAC,IAA2B,EAAC,CAAC,CAAC;QACnE,CAAC;QAED,MAAM,WAAW,GAAkB,EAAE,CAAC;QACtC,MAAM,SAAS,GAA2C,EAAE,CAAC;QAE7D,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACjC,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;gBACb,WAAW,CAAC,IAAI,CAAC,EAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAC,CAAC,CAAC;YACvD,CAAC;iBAAM,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;gBACvB,MAAM,EAAE,GAAG,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC;gBACnC,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,IAAK,CAAC;gBACjC,MAAM,IAAI,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE,CAA4B,CAAC;gBACpE,IAAI,EAAE,EAAE,CAAC;oBACL,SAAS,CAAC,IAAI,CAAC,EAAC,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAC,IAAI,EAAE,SAAS,EAAE,IAAI,EAAC,EAAC,CAAC,CAAC;oBAC1E,WAAW,CAAC,IAAI,CAAC,EAAC,IAAI,EAAE,WAAW,EAAE,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAC,CAAC,CAAC;gBACrE,CAAC;qBAAM,CAAC;oBACJ,SAAS,CAAC,IAAI,CAAC,EAAC,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAC,IAAI,EAAE,SAAS,EAAE,IAAI,EAAC,EAAC,CAAC,CAAC;oBACtE,WAAW,CAAC,IAAI,CAAC,EAAC,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAC,CAAC,CAAC;gBACjE,CAAC;YACL,CAAC;iBAAM,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;gBAC1B,WAAW,CAAC,IAAI,CAAC;oBACb,IAAI,EAAE,aAAa;oBACnB,YAAY,EAAE,KAAK,CAAC,UAAU,CAAC,SAAU;oBACzC,OAAO,EAAE,KAAK,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,CACvC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAC,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAC,CAAC,CAAC,CAAC;wBAC7C,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;qBAC1B,CACJ,IAAI,EAAE;oBACP,QAAQ,EAAE,KAAK,CAAC,UAAU,CAAC,MAAM,KAAK,OAAO;iBAChD,CAAC,CAAC;YACP,CAAC;iBAAM,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;gBACrB,MAAM,QAAQ,GAAqB,EAAC,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAC,CAAC;gBAC1E,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,CAAC;oBAC5B,QAAQ,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;gBAC7E,CAAC;gBACD,MAAM,IAAI,GAAG,YAAY,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAC9C,IAAI,IAAI;oBAAE,QAAQ,CAAC,IAAI,GAAG,IAAI,CAAC;gBAC/B,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC/B,CAAC;QACL,CAAC;QAED,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,IAAI,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC7D,OAAO,WAAW,CAAC;gBACf,IAAI,EAAE,MAAM,CAAC,IAA2B;gBACxC,OAAO,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI;gBAC5B,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAC,UAAU,EAAE,SAAS,EAAC,CAAC,CAAC,CAAC,EAAE,CAAC;aACvD,CAAC,CAAC;QACP,CAAC;QAED,OAAO,WAAW,CAAC;YACf,IAAI,EAAE,MAAM,CAAC,IAA2B;YACxC,OAAO,EAAE,WAAW;YACpB,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAC,UAAU,EAAE,SAAS,EAAC,CAAC,CAAC,CAAC,EAAE,CAAC;SACvD,CAAC,CAAC;IACP,CAAC;CACJ;AAED,SAAS,YAAY,CAAC,IAAa;IAC/B,IAAI,CAAC,IAAI;QAAE,OAAO,SAAS,CAAC;IAC5B,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IACvC,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IACvC,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;QAAE,OAAO,MAAM,CAAC;IACzC,OAAO,MAAM,CAAC;AAClB,CAAC;AAED,SAAS,YAAY,CAAC,MAAe;IACjC,IAAI,CAAC,MAAM;QAAE,OAAO,SAAS,CAAC;IAC9B,OAAO,SAAS,MAAM,EAAE,CAAC;AAC7B,CAAC"}