@cdklabs/cdk-appmod-catalog-blueprints 1.16.1 → 1.17.1

package/lib/document-processing/default-document-processing-config.js

@@ -1,14 +1,12 @@
 "use strict";
-var _a;
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.DefaultDocumentProcessingConfig = void 0;
 const JSII_RTTI_SYMBOL_1 = Symbol.for("jsii.rtti");
 // Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 // SPDX-License-Identifier: Apache-2.0
 class DefaultDocumentProcessingConfig {
+    static [JSII_RTTI_SYMBOL_1] = { fqn: "@cdklabs/cdk-appmod-catalog-blueprints.DefaultDocumentProcessingConfig", version: "1.17.1" };
+    static DEFAULT_OBSERVABILITY_METRIC_SVC_NAME = 'document-processing';
 }
 exports.DefaultDocumentProcessingConfig = DefaultDocumentProcessingConfig;
-_a = JSII_RTTI_SYMBOL_1;
-DefaultDocumentProcessingConfig[_a] = { fqn: "@cdklabs/cdk-appmod-catalog-blueprints.DefaultDocumentProcessingConfig", version: "1.16.1" };
-DefaultDocumentProcessingConfig.DEFAULT_OBSERVABILITY_METRIC_SVC_NAME = 'document-processing';
-//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiZGVmYXVsdC1kb2N1bWVudC1wcm9jZXNzaW5nLWNvbmZpZy5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uL3VzZS1jYXNlcy9kb2N1bWVudC1wcm9jZXNzaW5nL2RlZmF1bHQtZG9jdW1lbnQtcHJvY2Vzc2luZy1jb25maWcudHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6Ijs7Ozs7QUFBQSxxRUFBcUU7QUFDckUsc0NBQXNDO0FBRXRDLE1BQWEsK0JBQStCOztBQUE1QywwRUFFQzs7O0FBRHdCLHFFQUFxQyxHQUFHLHFCQUFxQixDQUFDIiwic291cmNlc0NvbnRlbnQiOlsiLy8gQ29weXJpZ2h0IEFtYXpvbi5jb20sIEluYy4gb3IgaXRzIGFmZmlsaWF0ZXMuIEFsbCBSaWdodHMgUmVzZXJ2ZWQuXG4vLyBTUERYLUxpY2Vuc2UtSWRlbnRpZmllcjogQXBhY2hlLTIuMFxuXG5leHBvcnQgY2xhc3MgRGVmYXVsdERvY3VtZW50UHJvY2Vzc2luZ0NvbmZpZyB7XG4gIHB1YmxpYyBzdGF0aWMgcmVhZG9ubHkgREVGQVVMVF9PQlNFUlZBQklMSVRZX01FVFJJQ19TVkNfTkFNRSA9ICdkb2N1bWVudC1wcm9jZXNzaW5nJztcbn0iXX0=
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiZGVmYXVsdC1kb2N1bWVudC1wcm9jZXNzaW5nLWNvbmZpZy5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uL3VzZS1jYXNlcy9kb2N1bWVudC1wcm9jZXNzaW5nL2RlZmF1bHQtZG9jdW1lbnQtcHJvY2Vzc2luZy1jb25maWcudHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6Ijs7OztBQUFBLHFFQUFxRTtBQUNyRSxzQ0FBc0M7QUFFdEMsTUFBYSwrQkFBK0I7O0lBQ25DLE1BQU0sQ0FBVSxxQ0FBcUMsR0FBRyxxQkFBcUIsQ0FBQzs7QUFEdkYsMEVBRUMiLCJzb3VyY2VzQ29udGVudCI6WyIvLyBDb3B5cmlnaHQgQW1hem9uLmNvbSwgSW5jLiBvciBpdHMgYWZmaWxpYXRlcy4gQWxsIFJpZ2h0cyBSZXNlcnZlZC5cbi8vIFNQRFgtTGljZW5zZS1JZGVudGlmaWVyOiBBcGFjaGUtMi4wXG5cbmV4cG9ydCBjbGFzcyBEZWZhdWx0RG9jdW1lbnRQcm9jZXNzaW5nQ29uZmlnIHtcbiAgcHVibGljIHN0YXRpYyByZWFkb25seSBERUZBVUxUX09CU0VSVkFCSUxJVFlfTUVUUklDX1NWQ19OQU1FID0gJ2RvY3VtZW50LXByb2Nlc3NpbmcnO1xufSJdfQ==

package/lib/document-processing/index.d.ts

@@ -1,6 +1,8 @@
 export * from './base-document-processing';
 export * from './bedrock-document-processing';
 export * from './agentic-document-processing';
+export * from './localstack-bedrock-document-processing';
+export * from './localstack-agentic-document-processing';
 export * from './adapter';
 export * from './default-document-processing-config';
 export * from './chunking-config';

package/lib/document-processing/index.js

@@ -17,7 +17,9 @@ Object.defineProperty(exports, "__esModule", { value: true });
 __exportStar(require("./base-document-processing"), exports);
 __exportStar(require("./bedrock-document-processing"), exports);
 __exportStar(require("./agentic-document-processing"), exports);
+__exportStar(require("./localstack-bedrock-document-processing"), exports);
+__exportStar(require("./localstack-agentic-document-processing"), exports);
 __exportStar(require("./adapter"), exports);
 __exportStar(require("./default-document-processing-config"), exports);
 __exportStar(require("./chunking-config"), exports);
-//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi91c2UtY2FzZXMvZG9jdW1lbnQtcHJvY2Vzc2luZy9pbmRleC50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7Ozs7Ozs7Ozs7Ozs7O0FBQUEsNkRBQTJDO0FBQzNDLGdFQUE4QztBQUM5QyxnRUFBOEM7QUFDOUMsNENBQTBCO0FBQzFCLHVFQUFxRDtBQUNyRCxvREFBa0MiLCJzb3VyY2VzQ29udGVudCI6WyJleHBvcnQgKiBmcm9tICcuL2Jhc2UtZG9jdW1lbnQtcHJvY2Vzc2luZyc7XG5leHBvcnQgKiBmcm9tICcuL2JlZHJvY2stZG9jdW1lbnQtcHJvY2Vzc2luZyc7XG5leHBvcnQgKiBmcm9tICcuL2FnZW50aWMtZG9jdW1lbnQtcHJvY2Vzc2luZyc7XG5leHBvcnQgKiBmcm9tICcuL2FkYXB0ZXInO1xuZXhwb3J0ICogZnJvbSAnLi9kZWZhdWx0LWRvY3VtZW50LXByb2Nlc3NpbmctY29uZmlnJztcbmV4cG9ydCAqIGZyb20gJy4vY2h1bmtpbmctY29uZmlnJzsiXX0=
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi91c2UtY2FzZXMvZG9jdW1lbnQtcHJvY2Vzc2luZy9pbmRleC50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7Ozs7Ozs7Ozs7Ozs7O0FBQUEsNkRBQTJDO0FBQzNDLGdFQUE4QztBQUM5QyxnRUFBOEM7QUFDOUMsMkVBQXlEO0FBQ3pELDJFQUF5RDtBQUN6RCw0Q0FBMEI7QUFDMUIsdUVBQXFEO0FBQ3JELG9EQUFrQyIsInNvdXJjZXNDb250ZW50IjpbImV4cG9ydCAqIGZyb20gJy4vYmFzZS1kb2N1bWVudC1wcm9jZXNzaW5nJztcbmV4cG9ydCAqIGZyb20gJy4vYmVkcm9jay1kb2N1bWVudC1wcm9jZXNzaW5nJztcbmV4cG9ydCAqIGZyb20gJy4vYWdlbnRpYy1kb2N1bWVudC1wcm9jZXNzaW5nJztcbmV4cG9ydCAqIGZyb20gJy4vbG9jYWxzdGFjay1iZWRyb2NrLWRvY3VtZW50LXByb2Nlc3NpbmcnO1xuZXhwb3J0ICogZnJvbSAnLi9sb2NhbHN0YWNrLWFnZW50aWMtZG9jdW1lbnQtcHJvY2Vzc2luZyc7XG5leHBvcnQgKiBmcm9tICcuL2FkYXB0ZXInO1xuZXhwb3J0ICogZnJvbSAnLi9kZWZhdWx0LWRvY3VtZW50LXByb2Nlc3NpbmctY29uZmlnJztcbmV4cG9ydCAqIGZyb20gJy4vY2h1bmtpbmctY29uZmlnJztcbiJdfQ==

package/lib/document-processing/localstack-agentic-document-processing.d.ts

@@ -0,0 +1,20 @@
+import { Construct } from 'constructs';
+import { AgenticDocumentProcessing, AgenticDocumentProcessingProps } from './agentic-document-processing';
+import { DocumentProcessingStepType } from './base-document-processing';
+import { LocalStackEndpointOverrides } from '../framework/localstack';
+export interface LocalStackAgenticDocumentProcessingProps extends AgenticDocumentProcessingProps {
+    /**
+     * LocalStack endpoint routing configuration for Lambda runtime SDK calls.
+     *
+     * @default { enabled: true }
+     */
+    readonly localStack?: LocalStackEndpointOverrides;
+}
+export declare class LocalStackAgenticDocumentProcessing extends AgenticDocumentProcessing {
+    private _localStackBatchAgent?;
+    private _localStackProcessingStepCounter?;
+    constructor(scope: Construct, id: string, props: LocalStackAgenticDocumentProcessingProps);
+    protected resolveBedrockInvokeEntry(): string;
+    protected processingStep(): DocumentProcessingStepType;
+    private applyLocalStackEnvironment;
+}

package/lib/document-processing/localstack-agentic-document-processing.js

@@ -0,0 +1,99 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LocalStackAgenticDocumentProcessing = void 0;
+const JSII_RTTI_SYMBOL_1 = Symbol.for("jsii.rtti");
+const path = __importStar(require("path"));
+const aws_lambda_1 = require("aws-cdk-lib/aws-lambda");
+const aws_stepfunctions_tasks_1 = require("aws-cdk-lib/aws-stepfunctions-tasks");
+const agentic_document_processing_1 = require("./agentic-document-processing");
+const localstack_batch_agent_1 = require("../framework/agents/localstack-batch-agent");
+const localstack_1 = require("../framework/localstack");
+class LocalStackAgenticDocumentProcessing extends agentic_document_processing_1.AgenticDocumentProcessing {
+    static [JSII_RTTI_SYMBOL_1] = { fqn: "@cdklabs/cdk-appmod-catalog-blueprints.LocalStackAgenticDocumentProcessing", version: "1.17.1" };
+    _localStackBatchAgent;
+    _localStackProcessingStepCounter;
+    constructor(scope, id, props) {
+        super(scope, id, {
+            ...props,
+            _skipBedrockVpcEndpoints: true,
+        });
+        this.applyLocalStackEnvironment(props.localStack || {});
+    }
+    resolveBedrockInvokeEntry() {
+        return path.join(__dirname, 'resources/default-localstack-invoke');
+    }
+    processingStep() {
+        if (!this._localStackBatchAgent) {
+            const agentProps = this.bedrockDocumentProcessingProps;
+            const processingAgentProps = agentProps.processingAgentParameters;
+            this._localStackBatchAgent = new localstack_batch_agent_1.LocalStackBatchAgent(this, 'IDPLocalStackBatchAgent', {
+                ...processingAgentProps,
+                localStack: agentProps.localStack,
+            });
+            const adapterPolicyStatements = this.ingressAdapter.generateAdapterIAMPolicies();
+            for (const statement of adapterPolicyStatements) {
+                this._localStackBatchAgent.agentRole.addToPrincipalPolicy(statement);
+            }
+        }
+        if (this._localStackProcessingStepCounter === undefined) {
+            this._localStackProcessingStepCounter = 0;
+        }
+        const stepId = `ProcessingStep-${this._localStackProcessingStepCounter}`;
+        this._localStackProcessingStepCounter++;
+        return new aws_stepfunctions_tasks_1.LambdaInvoke(this, stepId, {
+            lambdaFunction: this._localStackBatchAgent.agentFunction,
+            resultPath: '$.processingResult',
+            resultSelector: {
+                'result.$': '$.Payload.result',
+            },
+        });
+    }
+    applyLocalStackEnvironment(localStack) {
+        const localStackEnv = localstack_1.LocalStackIntegrationUtils.toLambdaEnvironment({
+            enabled: true,
+            ...localStack,
+        });
+        for (const child of this.node.findAll()) {
+            if (child instanceof aws_lambda_1.Function) {
+                for (const [key, value] of Object.entries(localStackEnv)) {
+                    child.addEnvironment(key, value);
+                }
+            }
+        }
+    }
+}
+exports.LocalStackAgenticDocumentProcessing = LocalStackAgenticDocumentProcessing;
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoibG9jYWxzdGFjay1hZ2VudGljLWRvY3VtZW50LXByb2Nlc3NpbmcuanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi91c2UtY2FzZXMvZG9jdW1lbnQtcHJvY2Vzc2luZy9sb2NhbHN0YWNrLWFnZW50aWMtZG9jdW1lbnQtcHJvY2Vzc2luZy50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7O0FBQUEsMkNBQTZCO0FBQzdCLHVEQUFvRTtBQUNwRSxpRkFBbUU7QUFFbkUsK0VBQTBHO0FBRTFHLHVGQUFrRjtBQUNsRix3REFBa0c7QUFXbEcsTUFBYSxtQ0FBb0MsU0FBUSx1REFBeUI7O0lBQ3hFLHFCQUFxQixDQUF3QjtJQUM3QyxnQ0FBZ0MsQ0FBVTtJQUVsRCxZQUFZLEtBQWdCLEVBQUUsRUFBVSxFQUFFLEtBQStDO1FBQ3ZGLEtBQUssQ0FBQyxLQUFLLEVBQUUsRUFBRSxFQUFFO1lBQ2YsR0FBRyxLQUFLO1lBQ1Isd0JBQXdCLEVBQUUsSUFBSTtTQUNHLENBQUMsQ0FBQztRQUNyQyxJQUFJLENBQUMsMEJBQTBCLENBQUMsS0FBSyxDQUFDLFVBQVUsSUFBSSxFQUFFLENBQUMsQ0FBQztJQUMxRCxDQUFDO0lBRVMseUJBQXlCO1FBQ2pDLE9BQU8sSUFBSSxDQUFDLElBQUksQ0FBQyxTQUFTLEVBQUUscUNBQXFDLENBQUMsQ0FBQztJQUNyRSxDQUFDO0lBRVMsY0FBYztRQUN0QixJQUFJLENBQUMsSUFBSSxDQUFDLHFCQUFxQixFQUFFLENBQUM7WUFDaEMsTUFBTSxVQUFVLEdBQUcsSUFBSSxDQUFDLDhCQUEwRSxDQUFDO1lBQ25HLE1BQU0sb0JBQW9CLEdBQUcsVUFBVSxDQUFDLHlCQUF5QixDQUFDO1lBRWxFLElBQUksQ0FBQyxxQkFBcUIsR0FBRyxJQUFJLDZDQUFvQixDQUFDLElBQUksRUFBRSx5QkFBeUIsRUFBRTtnQkFDckYsR0FBRyxvQkFBb0I7Z0JBQ3ZCLFVBQVUsRUFBRSxVQUFVLENBQUMsVUFBVTthQUNsQyxDQUFDLENBQUM7WUFFSCxNQUFNLHVCQUF1QixHQUFHLElBQUksQ0FBQyxjQUFjLENBQUMsMEJBQTBCLEVBQUUsQ0FBQztZQUNqRixLQUFLLE1BQU0sU0FBUyxJQUFJLHVCQUF1QixFQUFFLENBQUM7Z0JBQ2hELElBQUksQ0FBQyxxQkFBcUIsQ0FBQyxTQUFTLENBQUMsb0JBQW9CLENBQUMsU0FBUyxDQUFDLENBQUM7WUFDdkUsQ0FBQztRQUNILENBQUM7UUFFRCxJQUFJLElBQUksQ0FBQyxnQ0FBZ0MsS0FBSyxTQUFTLEVBQUUsQ0FBQztZQUN4RCxJQUFJLENBQUMsZ0NBQWdDLEdBQUcsQ0FBQyxDQUFDO1FBQzVDLENBQUM7UUFFRCxNQUFNLE1BQU0sR0FBRyxrQkFBa0IsSUFBSSxDQUFDLGdDQUFnQyxFQUFFLENBQUM7UUFDekUsSUFBSSxDQUFDLGdDQUFnQyxFQUFFLENBQUM7UUFFeEMsT0FBTyxJQUFJLHNDQUFZLENBQUMsSUFBSSxFQUFFLE1BQU0sRUFBRTtZQUNwQyxjQUFjLEVBQUUsSUFBSSxDQUFDLHFCQUFxQixDQUFDLGFBQWM7WUFDekQsVUFBVSxFQUFFLG9CQUFvQjtZQUNoQyxjQUFjLEVBQUU7Z0JBQ2QsVUFBVSxFQUFFLGtCQUFrQjthQUMvQjtTQUNGLENBQUMsQ0FBQztJQUNMLENBQUM7SUFFTywwQkFBMEIsQ0FBQyxVQUF1QztRQUN4RSxNQUFNLGFBQWEsR0FBRyx1Q0FBMEIsQ0FBQyxtQkFBbUIsQ0FBQztZQUNuRSxPQUFPLEVBQUUsSUFBSTtZQUNiLEdBQUcsVUFBVTtTQUNkLENBQUMsQ0FBQztRQUVILEtBQUssTUFBTSxLQUFLLElBQUksSUFBSSxDQUFDLElBQUksQ0FBQyxPQUFPLEVBQUUsRUFBRSxDQUFDO1lBQ3hDLElBQUksS0FBSyxZQUFZLHFCQUFjLEVBQUUsQ0FBQztnQkFDcEMsS0FBSyxNQUFNLENBQUMsR0FBRyxFQUFFLEtBQUssQ0FBQyxJQUFJLE1BQU0sQ0FBQyxPQUFPLENBQUMsYUFBYSxDQUFDLEVBQUUsQ0FBQztvQkFDekQsS0FBSyxDQUFDLGNBQWMsQ0FBQyxHQUFHLEVBQUUsS0FBSyxDQUFDLENBQUM7Z0JBQ25DLENBQUM7WUFDSCxDQUFDO1FBQ0gsQ0FBQztJQUNILENBQUM7O0FBN0RILGtGQThEQyIsInNvdXJjZXNDb250ZW50IjpbImltcG9ydCAqIGFzIHBhdGggZnJvbSAncGF0aCc7XG5pbXBvcnQgeyBGdW5jdGlvbiBhcyBMYW1iZGFGdW5jdGlvbiB9IGZyb20gJ2F3cy1jZGstbGliL2F3cy1sYW1iZGEnO1xuaW1wb3J0IHsgTGFtYmRhSW52b2tlIH0gZnJvbSAnYXdzLWNkay1saWIvYXdzLXN0ZXBmdW5jdGlvbnMtdGFza3MnO1xuaW1wb3J0IHsgQ29uc3RydWN0IH0gZnJvbSAnY29uc3RydWN0cyc7XG5pbXBvcnQgeyBBZ2VudGljRG9jdW1lbnRQcm9jZXNzaW5nLCBBZ2VudGljRG9jdW1lbnRQcm9jZXNzaW5nUHJvcHMgfSBmcm9tICcuL2FnZW50aWMtZG9jdW1lbnQtcHJvY2Vzc2luZyc7XG5pbXBvcnQgeyBEb2N1bWVudFByb2Nlc3NpbmdTdGVwVHlwZSB9IGZyb20gJy4vYmFzZS1kb2N1bWVudC1wcm9jZXNzaW5nJztcbmltcG9ydCB7IExvY2FsU3RhY2tCYXRjaEFnZW50IH0gZnJvbSAnLi4vZnJhbWV3b3JrL2FnZW50cy9sb2NhbHN0YWNrLWJhdGNoLWFnZW50JztcbmltcG9ydCB7IExvY2FsU3RhY2tFbmRwb2ludE92ZXJyaWRlcywgTG9jYWxTdGFja0ludGVncmF0aW9uVXRpbHMgfSBmcm9tICcuLi9mcmFtZXdvcmsvbG9jYWxzdGFjayc7XG5cbmV4cG9ydCBpbnRlcmZhY2UgTG9jYWxTdGFja0FnZW50aWNEb2N1bWVudFByb2Nlc3NpbmdQcm9wcyBleHRlbmRzIEFnZW50aWNEb2N1bWVudFByb2Nlc3NpbmdQcm9wcyB7XG4gIC8qKlxuICAgKiBMb2NhbFN0YWNrIGVuZHBvaW50IHJvdXRpbmcgY29uZmlndXJhdGlvbiBmb3IgTGFtYmRhIHJ1bnRpbWUgU0RLIGNhbGxzLlxuICAgKlxuICAgKiBAZGVmYXVsdCB7IGVuYWJsZWQ6IHRydWUgfVxuICAgKi9cbiAgcmVhZG9ubHkgbG9jYWxTdGFjaz86IExvY2FsU3RhY2tFbmRwb2ludE92ZXJyaWRlcztcbn1cblxuZXhwb3J0IGNsYXNzIExvY2FsU3RhY2tBZ2VudGljRG9jdW1lbnRQcm9jZXNzaW5nIGV4dGVuZHMgQWdlbnRpY0RvY3VtZW50UHJvY2Vzc2luZyB7XG4gIHByaXZhdGUgX2xvY2FsU3RhY2tCYXRjaEFnZW50PzogTG9jYWxTdGFja0JhdGNoQWdlbnQ7XG4gIHByaXZhdGUgX2xvY2FsU3RhY2tQcm9jZXNzaW5nU3RlcENvdW50ZXI/OiBudW1iZXI7XG5cbiAgY29uc3RydWN0b3Ioc2NvcGU6IENvbnN0cnVjdCwgaWQ6IHN0cmluZywgcHJvcHM6IExvY2FsU3RhY2tBZ2VudGljRG9jdW1lbnRQcm9jZXNzaW5nUHJvcHMpIHtcbiAgICBzdXBlcihzY29wZSwgaWQsIHtcbiAgICAgIC4uLnByb3BzLFxuICAgICAgX3NraXBCZWRyb2NrVnBjRW5kcG9pbnRzOiB0cnVlLFxuICAgIH0gYXMgQWdlbnRpY0RvY3VtZW50UHJvY2Vzc2luZ1Byb3BzKTtcbiAgICB0aGlzLmFwcGx5TG9jYWxTdGFja0Vudmlyb25tZW50KHByb3BzLmxvY2FsU3RhY2sgfHwge30pO1xuICB9XG5cbiAgcHJvdGVjdGVkIHJlc29sdmVCZWRyb2NrSW52b2tlRW50cnkoKTogc3RyaW5nIHtcbiAgICByZXR1cm4gcGF0aC5qb2luKF9fZGlybmFtZSwgJ3Jlc291cmNlcy9kZWZhdWx0LWxvY2Fsc3RhY2staW52b2tlJyk7XG4gIH1cblxuICBwcm90ZWN0ZWQgcHJvY2Vzc2luZ1N0ZXAoKTogRG9jdW1lbnRQcm9jZXNzaW5nU3RlcFR5cGUge1xuICAgIGlmICghdGhpcy5fbG9jYWxTdGFja0JhdGNoQWdlbnQpIHtcbiAgICAgIGNvbnN0IGFnZW50UHJvcHMgPSB0aGlzLmJlZHJvY2tEb2N1bWVudFByb2Nlc3NpbmdQcm9wcyBhcyBMb2NhbFN0YWNrQWdlbnRpY0RvY3VtZW50UHJvY2Vzc2luZ1Byb3BzO1xuICAgICAgY29uc3QgcHJvY2Vzc2luZ0FnZW50UHJvcHMgPSBhZ2VudFByb3BzLnByb2Nlc3NpbmdBZ2VudFBhcmFtZXRlcnM7XG5cbiAgICAgIHRoaXMuX2xvY2FsU3RhY2tCYXRjaEFnZW50ID0gbmV3IExvY2FsU3RhY2tCYXRjaEFnZW50KHRoaXMsICdJRFBMb2NhbFN0YWNrQmF0Y2hBZ2VudCcsIHtcbiAgICAgICAgLi4ucHJvY2Vzc2luZ0FnZW50UHJvcHMsXG4gICAgICAgIGxvY2FsU3RhY2s6IGFnZW50UHJvcHMubG9jYWxTdGFjayxcbiAgICAgIH0pO1xuXG4gICAgICBjb25zdCBhZGFwdGVyUG9saWN5U3RhdGVtZW50cyA9IHRoaXMuaW5ncmVzc0FkYXB0ZXIuZ2VuZXJhdGVBZGFwdGVySUFNUG9saWNpZXMoKTtcbiAgICAgIGZvciAoY29uc3Qgc3RhdGVtZW50IG9mIGFkYXB0ZXJQb2xpY3lTdGF0ZW1lbnRzKSB7XG4gICAgICAgIHRoaXMuX2xvY2FsU3RhY2tCYXRjaEFnZW50LmFnZW50Um9sZS5hZGRUb1ByaW5jaXBhbFBvbGljeShzdGF0ZW1lbnQpO1xuICAgICAgfVxuICAgIH1cblxuICAgIGlmICh0aGlzLl9sb2NhbFN0YWNrUHJvY2Vzc2luZ1N0ZXBDb3VudGVyID09PSB1bmRlZmluZWQpIHtcbiAgICAgIHRoaXMuX2xvY2FsU3RhY2tQcm9jZXNzaW5nU3RlcENvdW50ZXIgPSAwO1xuICAgIH1cblxuICAgIGNvbnN0IHN0ZXBJZCA9IGBQcm9jZXNzaW5nU3RlcC0ke3RoaXMuX2xvY2FsU3RhY2tQcm9jZXNzaW5nU3RlcENvdW50ZXJ9YDtcbiAgICB0aGlzLl9sb2NhbFN0YWNrUHJvY2Vzc2luZ1N0ZXBDb3VudGVyKys7XG5cbiAgICByZXR1cm4gbmV3IExhbWJkYUludm9rZSh0aGlzLCBzdGVwSWQsIHtcbiAgICAgIGxhbWJkYUZ1bmN0aW9uOiB0aGlzLl9sb2NhbFN0YWNrQmF0Y2hBZ2VudC5hZ2VudEZ1bmN0aW9uISxcbiAgICAgIHJlc3VsdFBhdGg6ICckLnByb2Nlc3NpbmdSZXN1bHQnLFxuICAgICAgcmVzdWx0U2VsZWN0b3I6IHtcbiAgICAgICAgJ3Jlc3VsdC4kJzogJyQuUGF5bG9hZC5yZXN1bHQnLFxuICAgICAgfSxcbiAgICB9KTtcbiAgfVxuXG4gIHByaXZhdGUgYXBwbHlMb2NhbFN0YWNrRW52aXJvbm1lbnQobG9jYWxTdGFjazogTG9jYWxTdGFja0VuZHBvaW50T3ZlcnJpZGVzKTogdm9pZCB7XG4gICAgY29uc3QgbG9jYWxTdGFja0VudiA9IExvY2FsU3RhY2tJbnRlZ3JhdGlvblV0aWxzLnRvTGFtYmRhRW52aXJvbm1lbnQoe1xuICAgICAgZW5hYmxlZDogdHJ1ZSxcbiAgICAgIC4uLmxvY2FsU3RhY2ssXG4gICAgfSk7XG5cbiAgICBmb3IgKGNvbnN0IGNoaWxkIG9mIHRoaXMubm9kZS5maW5kQWxsKCkpIHtcbiAgICAgIGlmIChjaGlsZCBpbnN0YW5jZW9mIExhbWJkYUZ1bmN0aW9uKSB7XG4gICAgICAgIGZvciAoY29uc3QgW2tleSwgdmFsdWVdIG9mIE9iamVjdC5lbnRyaWVzKGxvY2FsU3RhY2tFbnYpKSB7XG4gICAgICAgICAgY2hpbGQuYWRkRW52aXJvbm1lbnQoa2V5LCB2YWx1ZSk7XG4gICAgICAgIH1cbiAgICAgIH1cbiAgICB9XG4gIH1cbn1cbiJdfQ==

package/lib/document-processing/localstack-bedrock-document-processing.d.ts

@@ -0,0 +1,16 @@
+import { Construct } from 'constructs';
+import { BedrockDocumentProcessing, BedrockDocumentProcessingProps } from './bedrock-document-processing';
+import { LocalStackEndpointOverrides } from '../framework/localstack';
+export interface LocalStackBedrockDocumentProcessingProps extends BedrockDocumentProcessingProps {
+    /**
+     * LocalStack endpoint routing configuration for Lambda runtime SDK calls.
+     *
+     * @default { enabled: true }
+     */
+    readonly localStack?: LocalStackEndpointOverrides;
+}
+export declare class LocalStackBedrockDocumentProcessing extends BedrockDocumentProcessing {
+    constructor(scope: Construct, id: string, props: LocalStackBedrockDocumentProcessingProps);
+    protected resolveBedrockInvokeEntry(): string;
+    private applyLocalStackEnvironment;
+}

package/lib/document-processing/localstack-bedrock-document-processing.js

@@ -0,0 +1,69 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LocalStackBedrockDocumentProcessing = void 0;
+const JSII_RTTI_SYMBOL_1 = Symbol.for("jsii.rtti");
+const path = __importStar(require("path"));
+const aws_lambda_1 = require("aws-cdk-lib/aws-lambda");
+const bedrock_document_processing_1 = require("./bedrock-document-processing");
+const localstack_1 = require("../framework/localstack");
+class LocalStackBedrockDocumentProcessing extends bedrock_document_processing_1.BedrockDocumentProcessing {
+    static [JSII_RTTI_SYMBOL_1] = { fqn: "@cdklabs/cdk-appmod-catalog-blueprints.LocalStackBedrockDocumentProcessing", version: "1.17.1" };
+    constructor(scope, id, props) {
+        super(scope, id, {
+            ...props,
+            _skipBedrockVpcEndpoints: true,
+        });
+        this.applyLocalStackEnvironment(props.localStack);
+    }
+    resolveBedrockInvokeEntry() {
+        return path.join(__dirname, 'resources/default-localstack-invoke');
+    }
+    applyLocalStackEnvironment(localStack) {
+        const localStackEnv = localstack_1.LocalStackIntegrationUtils.toLambdaEnvironment({
+            enabled: true,
+            ...localStack,
+        });
+        for (const child of this.node.findAll()) {
+            if (child instanceof aws_lambda_1.Function) {
+                for (const [key, value] of Object.entries(localStackEnv)) {
+                    child.addEnvironment(key, value);
+                }
+            }
+        }
+    }
+}
+exports.LocalStackBedrockDocumentProcessing = LocalStackBedrockDocumentProcessing;
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoibG9jYWxzdGFjay1iZWRyb2NrLWRvY3VtZW50LXByb2Nlc3NpbmcuanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi91c2UtY2FzZXMvZG9jdW1lbnQtcHJvY2Vzc2luZy9sb2NhbHN0YWNrLWJlZHJvY2stZG9jdW1lbnQtcHJvY2Vzc2luZy50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7O0FBQUEsMkNBQTZCO0FBQzdCLHVEQUFvRTtBQUVwRSwrRUFBMEc7QUFDMUcsd0RBQWtHO0FBV2xHLE1BQWEsbUNBQW9DLFNBQVEsdURBQXlCOztJQUNoRixZQUFZLEtBQWdCLEVBQUUsRUFBVSxFQUFFLEtBQStDO1FBQ3ZGLEtBQUssQ0FBQyxLQUFLLEVBQUUsRUFBRSxFQUFFO1lBQ2YsR0FBRyxLQUFLO1lBQ1Isd0JBQXdCLEVBQUUsSUFBSTtTQUNHLENBQUMsQ0FBQztRQUNyQyxJQUFJLENBQUMsMEJBQTBCLENBQUMsS0FBSyxDQUFDLFVBQVUsQ0FBQyxDQUFDO0lBQ3BELENBQUM7SUFFUyx5QkFBeUI7UUFDakMsT0FBTyxJQUFJLENBQUMsSUFBSSxDQUFDLFNBQVMsRUFBRSxxQ0FBcUMsQ0FBQyxDQUFDO0lBQ3JFLENBQUM7SUFFTywwQkFBMEIsQ0FBQyxVQUF3QztRQUN6RSxNQUFNLGFBQWEsR0FBRyx1Q0FBMEIsQ0FBQyxtQkFBbUIsQ0FBQztZQUNuRSxPQUFPLEVBQUUsSUFBSTtZQUNiLEdBQUcsVUFBVTtTQUNkLENBQUMsQ0FBQztRQUVILEtBQUssTUFBTSxLQUFLLElBQUksSUFBSSxDQUFDLElBQUksQ0FBQyxPQUFPLEVBQUUsRUFBRSxDQUFDO1lBQ3hDLElBQUksS0FBSyxZQUFZLHFCQUFjLEVBQUUsQ0FBQztnQkFDcEMsS0FBSyxNQUFNLENBQUMsR0FBRyxFQUFFLEtBQUssQ0FBQyxJQUFJLE1BQU0sQ0FBQyxPQUFPLENBQUMsYUFBYSxDQUFDLEVBQUUsQ0FBQztvQkFDekQsS0FBSyxDQUFDLGNBQWMsQ0FBQyxHQUFHLEVBQUUsS0FBSyxDQUFDLENBQUM7Z0JBQ25DLENBQUM7WUFDSCxDQUFDO1FBQ0gsQ0FBQztJQUNILENBQUM7O0FBMUJILGtGQTJCQyIsInNvdXJjZXNDb250ZW50IjpbImltcG9ydCAqIGFzIHBhdGggZnJvbSAncGF0aCc7XG5pbXBvcnQgeyBGdW5jdGlvbiBhcyBMYW1iZGFGdW5jdGlvbiB9IGZyb20gJ2F3cy1jZGstbGliL2F3cy1sYW1iZGEnO1xuaW1wb3J0IHsgQ29uc3RydWN0IH0gZnJvbSAnY29uc3RydWN0cyc7XG5pbXBvcnQgeyBCZWRyb2NrRG9jdW1lbnRQcm9jZXNzaW5nLCBCZWRyb2NrRG9jdW1lbnRQcm9jZXNzaW5nUHJvcHMgfSBmcm9tICcuL2JlZHJvY2stZG9jdW1lbnQtcHJvY2Vzc2luZyc7XG5pbXBvcnQgeyBMb2NhbFN0YWNrRW5kcG9pbnRPdmVycmlkZXMsIExvY2FsU3RhY2tJbnRlZ3JhdGlvblV0aWxzIH0gZnJvbSAnLi4vZnJhbWV3b3JrL2xvY2Fsc3RhY2snO1xuXG5leHBvcnQgaW50ZXJmYWNlIExvY2FsU3RhY2tCZWRyb2NrRG9jdW1lbnRQcm9jZXNzaW5nUHJvcHMgZXh0ZW5kcyBCZWRyb2NrRG9jdW1lbnRQcm9jZXNzaW5nUHJvcHMge1xuICAvKipcbiAgICogTG9jYWxTdGFjayBlbmRwb2ludCByb3V0aW5nIGNvbmZpZ3VyYXRpb24gZm9yIExhbWJkYSBydW50aW1lIFNESyBjYWxscy5cbiAgICpcbiAgICogQGRlZmF1bHQgeyBlbmFibGVkOiB0cnVlIH1cbiAgICovXG4gIHJlYWRvbmx5IGxvY2FsU3RhY2s/OiBMb2NhbFN0YWNrRW5kcG9pbnRPdmVycmlkZXM7XG59XG5cbmV4cG9ydCBjbGFzcyBMb2NhbFN0YWNrQmVkcm9ja0RvY3VtZW50UHJvY2Vzc2luZyBleHRlbmRzIEJlZHJvY2tEb2N1bWVudFByb2Nlc3Npbmcge1xuICBjb25zdHJ1Y3RvcihzY29wZTogQ29uc3RydWN0LCBpZDogc3RyaW5nLCBwcm9wczogTG9jYWxTdGFja0JlZHJvY2tEb2N1bWVudFByb2Nlc3NpbmdQcm9wcykge1xuICAgIHN1cGVyKHNjb3BlLCBpZCwge1xuICAgICAgLi4ucHJvcHMsXG4gICAgICBfc2tpcEJlZHJvY2tWcGNFbmRwb2ludHM6IHRydWUsXG4gICAgfSBhcyBCZWRyb2NrRG9jdW1lbnRQcm9jZXNzaW5nUHJvcHMpO1xuICAgIHRoaXMuYXBwbHlMb2NhbFN0YWNrRW52aXJvbm1lbnQocHJvcHMubG9jYWxTdGFjayk7XG4gIH1cblxuICBwcm90ZWN0ZWQgcmVzb2x2ZUJlZHJvY2tJbnZva2VFbnRyeSgpOiBzdHJpbmcge1xuICAgIHJldHVybiBwYXRoLmpvaW4oX19kaXJuYW1lLCAncmVzb3VyY2VzL2RlZmF1bHQtbG9jYWxzdGFjay1pbnZva2UnKTtcbiAgfVxuXG4gIHByaXZhdGUgYXBwbHlMb2NhbFN0YWNrRW52aXJvbm1lbnQobG9jYWxTdGFjaz86IExvY2FsU3RhY2tFbmRwb2ludE92ZXJyaWRlcyk6IHZvaWQge1xuICAgIGNvbnN0IGxvY2FsU3RhY2tFbnYgPSBMb2NhbFN0YWNrSW50ZWdyYXRpb25VdGlscy50b0xhbWJkYUVudmlyb25tZW50KHtcbiAgICAgIGVuYWJsZWQ6IHRydWUsXG4gICAgICAuLi5sb2NhbFN0YWNrLFxuICAgIH0pO1xuXG4gICAgZm9yIChjb25zdCBjaGlsZCBvZiB0aGlzLm5vZGUuZmluZEFsbCgpKSB7XG4gICAgICBpZiAoY2hpbGQgaW5zdGFuY2VvZiBMYW1iZGFGdW5jdGlvbikge1xuICAgICAgICBmb3IgKGNvbnN0IFtrZXksIHZhbHVlXSBvZiBPYmplY3QuZW50cmllcyhsb2NhbFN0YWNrRW52KSkge1xuICAgICAgICAgIGNoaWxkLmFkZEVudmlyb25tZW50KGtleSwgdmFsdWUpO1xuICAgICAgICB9XG4gICAgICB9XG4gICAgfVxuICB9XG59XG4iXX0=

package/lib/document-processing/resources/cleanup/handler.py

@@ -36,8 +36,23 @@ except ImportError:
 logger = logging.getLogger()
 logger.setLevel(os.environ.get('LOG_LEVEL', 'INFO'))
 
+def _resolve_endpoint_url(*service_env_keys: str) -> str | None:
+    for key in service_env_keys:
+        endpoint = os.getenv(key)
+        if endpoint:
+            return endpoint
+    return None
+
+
+def _create_boto3_client(service_name: str, *service_env_keys: str):
+    endpoint_url = _resolve_endpoint_url(*service_env_keys, 'AWS_ENDPOINT_URL')
+    if endpoint_url:
+        return boto3.client(service_name, endpoint_url=endpoint_url)
+    return boto3.client(service_name)
+
+
 # Initialize S3 client
-s3_client = boto3.client('s3')
+s3_client = _create_boto3_client('s3', 'AWS_ENDPOINT_URL_S3')
 
 
 def handler(event: Dict[str, Any], context: Any) -> Dict[str, Any]:

package/lib/document-processing/resources/default-localstack-invoke/index.py

@@ -0,0 +1,184 @@
+import base64
+import logging
+import os
+from aws_lambda_powertools import Metrics, Tracer
+from aws_lambda_powertools.metrics import MetricUnit
+
+from provider_runtime import (
+    MODEL_PROVIDER_OLLAMA,
+    create_boto3_client,
+    extract_pdf_text_for_ollama,
+    invoke_model,
+    normalize_model_response,
+    resolve_model_provider,
+)
+
+s3 = create_boto3_client('s3', 'AWS_ENDPOINT_URL_S3')
+bedrock = create_boto3_client('bedrock-runtime', 'AWS_ENDPOINT_URL_BEDROCK_RUNTIME')
+metrics = Metrics()
+tracer = Tracer()
+logger = logging.getLogger()
+logger.setLevel(os.getenv('LOG_LEVEL', 'INFO').upper())
+
+
+def parse_chunk_metadata(event):
+    """
+    Parse optional chunk metadata from the event payload.
+    
+    Returns a dictionary with chunk information if present, None otherwise.
+    Supports both direct chunk metadata and nested chunk object format.
+    """
+    # Check for direct chunkMetadata field
+    if 'chunkMetadata' in event:
+        return event['chunkMetadata']
+    
+    # Check for chunk object (from Map State iteration)
+    if 'chunk' in event:
+        chunk = event['chunk']
+        return {
+            'chunkIndex': chunk.get('chunkIndex', event.get('chunkIndex', 0)),
+            'totalChunks': event.get('totalChunks', 1),
+            'startPage': chunk.get('startPage', 0),
+            'endPage': chunk.get('endPage', 0),
+            'pageCount': chunk.get('pageCount', 0),
+            'estimatedTokens': chunk.get('estimatedTokens', 0),
+            'overlapPages': chunk.get('overlapPages', 0),
+        }
+    
+    return None
+
+
+def build_chunk_context_prompt(chunk_metadata):
+    """
+    Build a context prompt for chunk-aware processing.
+    
+    Args:
+        chunk_metadata: Dictionary containing chunk information
+        
+    Returns:
+        String with chunk context to prepend to the main prompt
+    """
+    if not chunk_metadata:
+        return ""
+    
+    chunk_index = chunk_metadata.get('chunkIndex', 0)
+    total_chunks = chunk_metadata.get('totalChunks', 1)
+    start_page = chunk_metadata.get('startPage', 0)
+    end_page = chunk_metadata.get('endPage', 0)
+    overlap_pages = chunk_metadata.get('overlapPages', 0)
+    
+    # Build context string
+    context_parts = [
+        f"You are analyzing chunk {chunk_index + 1} of {total_chunks} from pages {start_page + 1} to {end_page + 1}."
+    ]
+    
+    # Add overlap information if applicable
+    if overlap_pages > 0 and chunk_index > 0:
+        context_parts.append(
+            f"Note: This chunk includes {overlap_pages} overlapping pages from the previous chunk for context."
+        )
+    
+    return "\n".join(context_parts) + "\n\n"
+
+
+@metrics.log_metrics
+@tracer.capture_lambda_handler
+def handler(event, context):
+    invoke_type = os.environ["INVOKE_TYPE"]
+    tracer.put_annotation(key="invoke_type", value=invoke_type)
+    tracer.put_annotation(key="documentId", value=event["documentId"])
+    metrics.add_dimension(name="invoke_type", value=invoke_type)
+    content_type = event["contentType"]
+    
+    # Parse optional chunk metadata
+    chunk_metadata = parse_chunk_metadata(event)
+    if chunk_metadata:
+        tracer.put_annotation(key="chunkIndex", value=str(chunk_metadata.get('chunkIndex', 0)))
+        tracer.put_annotation(key="totalChunks", value=str(chunk_metadata.get('totalChunks', 1)))
+        metrics.add_dimension(name="is_chunked", value="true")
+    else:
+        metrics.add_dimension(name="is_chunked", value="false")
+    
+    # Format prompt if classification result exists
+    prompt = os.environ['PROMPT']
+    if 'classificationResult' in event:
+        classification = event['classificationResult']['documentClassification']
+        prompt = prompt.replace("[ACTUAL_CLASSIFICATION]", classification)
+    
+    # Add chunk context to prompt if processing a chunk
+    chunk_context = build_chunk_context_prompt(chunk_metadata)
+    if chunk_context:
+        prompt = chunk_context + prompt
+
+    model_provider = resolve_model_provider()
+    
+    # Build content based on file type
+    content = [{'type': 'text', 'text': prompt}]
+    if content_type == 'file':
+        content_location = event['content']['location']
+        
+        if content_location == 's3':
+            # Use chunk-specific S3 location if available, otherwise use original content
+            if chunk_metadata and 'bucket' in chunk_metadata and 'key' in chunk_metadata:
+                bucket = chunk_metadata['bucket']
+                key = chunk_metadata['key']
+            else:
+                bucket = event['content']['bucket']
+                key = event['content']['key']
+            
+            # Check file type
+            ext = key.lower().split('.')[-1]
+            if ext not in ['jpg', 'jpeg', 'png', 'pdf']:
+                raise ValueError(f"Unsupported file type: {ext}")
+            
+            media_type = {'jpg': 'image/jpeg', 'jpeg': 'image/jpeg', 'png': 'image/png', 'pdf': 'application/pdf'}[ext]
+            
+            # Download file to /tmp
+            local_path = f"/tmp/{key.split('/')[-1]}"
+            s3.download_file(bucket, key, local_path)
+
+            if model_provider == MODEL_PROVIDER_OLLAMA:
+                if ext == 'pdf':
+                    extracted_text = extract_pdf_text_for_ollama(local_path)
+                    if extracted_text:
+                        content.append({
+                            'type': 'text',
+                            'text': f'Document text extract:\n\n{extracted_text}'
+                        })
+                    else:
+                        content.append({
+                            'type': 'text',
+                            'text': (
+                                'Document is PDF, but text extraction returned no content. '
+                                'Classify based on any other provided context.'
+                            )
+                        })
+                else:
+                    content.append({
+                        'type': 'text',
+                        'text': (
+                            f'Input includes an image ({media_type}). '
+                            'Direct Ollama mode does not attach image bytes in this workflow.'
+                        )
+                    })
+            else:
+                # Read and encode file
+                with open(local_path, 'rb') as f:
+                    file_data = base64.b64encode(f.read()).decode('utf-8')
+
+                if ext == 'pdf':
+                    content.append({'type': 'document', 'source': {'type': 'base64', 'media_type': media_type, 'data': file_data}})
+                else:
+                    content.append({'type': 'image', 'source': {'type': 'base64', 'media_type': media_type, 'data': file_data}})
+    
+    elif content_type == 'data':
+        content.append({
+            'type': 'text',
+            'text': event['content']['data']
+        })
+
+    model_id = os.environ['MODEL_ID']
+    max_tokens = int(os.getenv('INVOKE_MAX_TOKENS', '1000'))
+    response_text = invoke_model(content, max_tokens, model_id, bedrock, logger)
+    metrics.add_metric(name="SuccessfulInvocation", unit=MetricUnit.Count, value=1)
+    return normalize_model_response(response_text, invoke_type)