@aws-amplify/graphql-api-construct 1.13.0-ai.1 → 1.13.0-ai.3

package/node_modules/@aws-amplify/graphql-conversation-transformer/lib/resolvers/invoke-lambda-resolver.js

@@ -1,94 +1,68 @@
 "use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.invokeLambdaMappingTemplate = void 0;
 const graphql_transformer_core_1 = require("@aws-amplify/graphql-transformer-core");
-const ts_dedent_1 = require("ts-dedent");
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const ts_dedent_1 = __importDefault(require("ts-dedent"));
+const graphql_transformer_common_1 = require("graphql-transformer-common");
+const pluralize_1 = __importDefault(require("pluralize"));
 const invokeLambdaMappingTemplate = (config, ctx) => {
-    const req = createInvokeLambdaRequestFunction(config, ctx);
-    const res = createInvokeLambdaResponseFunction(config);
-    return { req, res };
+    const { TOOL_DEFINITIONS_LINE, TOOLS_CONFIGURATION_LINE } = generateToolLines(config);
+    const SELECTION_SET = selectionSet;
+    const GRAPHQL_API_ENDPOINT = ctx.api.graphqlUrl;
+    const MODEL_CONFIGURATION_LINE = generateModelConfigurationLine(config);
+    const RESPONSE_MUTATION_NAME = config.responseMutationName;
+    const RESPONSE_MUTATION_INPUT_TYPE_NAME = config.responseMutationInputTypeName;
+    const MESSAGE_MODEL_NAME = config.messageModel.messageModel.name.value;
+    const GET_QUERY_NAME = `getConversationMessage${(0, graphql_transformer_common_1.toUpper)(config.field.name.value)}`;
+    const GET_QUERY_INPUT_TYPE_NAME = 'ID';
+    const LIST_QUERY_NAME = `listConversationMessage${(0, graphql_transformer_common_1.toUpper)((0, pluralize_1.default)(config.field.name.value))}`;
+    const LIST_QUERY_INPUT_TYPE_NAME = `ModelConversationMessage${(0, graphql_transformer_common_1.toUpper)(config.field.name.value)}FilterInput`;
+    const LIST_QUERY_LIMIT = 'undefined';
+    const substitutions = {
+        TOOL_DEFINITIONS_LINE,
+        TOOLS_CONFIGURATION_LINE,
+        SELECTION_SET,
+        GRAPHQL_API_ENDPOINT,
+        MODEL_CONFIGURATION_LINE,
+        RESPONSE_MUTATION_NAME,
+        RESPONSE_MUTATION_INPUT_TYPE_NAME,
+        MESSAGE_MODEL_NAME,
+        GET_QUERY_NAME,
+        GET_QUERY_INPUT_TYPE_NAME,
+        LIST_QUERY_NAME,
+        LIST_QUERY_INPUT_TYPE_NAME,
+        LIST_QUERY_LIMIT,
+    };
+    let resolver = fs_1.default.readFileSync(path_1.default.join(__dirname, 'invoke-lambda-resolver-fn.template.js'), 'utf8');
+    Object.entries(substitutions).forEach(([key, value]) => {
+        const replaced = resolver.replace(new RegExp(`\\[\\[${key}\\]\\]`, 'g'), value);
+        resolver = replaced;
+    });
+    return graphql_transformer_core_1.MappingTemplate.inlineTemplateFromString(resolver);
 };
 exports.invokeLambdaMappingTemplate = invokeLambdaMappingTemplate;
-const createInvokeLambdaRequestFunction = (config, ctx) => {
-    const { responseMutationInputTypeName, responseMutationName } = config;
+const generateToolLines = (config) => {
     const toolDefinitions = JSON.stringify(config.toolSpec);
-    const toolDefinitionsLine = toolDefinitions ? `const toolDefinitions = ${toolDefinitions};` : '';
-    const modelConfigurationLine = generateModelConfigurationLine(config);
-    const graphqlEndpoint = ctx.api.graphqlUrl;
-    const toolsConfigurationLine = toolDefinitions
-        ? (0, ts_dedent_1.dedent) `const dataTools = toolDefinitions.tools;
+    const TOOL_DEFINITIONS_LINE = toolDefinitions ? `const toolDefinitions = ${toolDefinitions};` : '';
+    const TOOLS_CONFIGURATION_LINE = toolDefinitions
+        ? (0, ts_dedent_1.default) `const dataTools = toolDefinitions.tools;
      const toolsConfiguration = {
       dataTools,
       clientTools,
     };`
-        : (0, ts_dedent_1.dedent) `const toolsConfiguration = {
+        : (0, ts_dedent_1.default) `const toolsConfiguration = {
       clientTools
     };`;
-    const requestFunctionString = `
-  import { util } from '@aws-appsync/utils';
-
-  export function request(ctx) {
-    const { args, identity, request, prev } = ctx;
-    ${toolDefinitionsLine}
-    const selectionSet = '${selectionSet}';
-    const graphqlApiEndpoint = '${graphqlEndpoint}';
-
-    const messages = prev.result.items;
-    const responseMutation = {
-      name: '${responseMutationName}',
-      inputTypeName: '${responseMutationInputTypeName}',
-      selectionSet,
-    };
-    const currentMessageId = ctx.stash.defaultValues.id;
-    ${modelConfigurationLine}
-
-    const clientTools = args.toolConfiguration?.tools?.map((tool) => { return { ...tool.toolSpec }});
-    ${toolsConfigurationLine}
-
-    const authHeader = request.headers['authorization'];
-    const payload = {
-      conversationId: args.conversationId,
-      currentMessageId,
-      responseMutation,
-      graphqlApiEndpoint,
-      modelConfiguration,
-      request: { headers: { authorization: authHeader }},
-      messages,
-      toolsConfiguration,
-    };
-
-    return {
-      operation: 'Invoke',
-      payload,
-      invocationType: 'Event'
-    };
-  }`;
-    return graphql_transformer_core_1.MappingTemplate.inlineTemplateFromString((0, ts_dedent_1.dedent)(requestFunctionString));
-};
-const createInvokeLambdaResponseFunction = (config) => {
-    const responseFunctionString = `
-  export function response(ctx) {
-    let success = true;
-    if (ctx.error) {
-      util.appendError(ctx.error.message, ctx.error.type);
-      success = false;
-    }
-    const response = {
-        __typename: '${config.messageModel.messageModel.name.value}',
-        id: ctx.stash.defaultValues.id,
-        conversationId: ctx.args.conversationId,
-        role: 'user',
-        content: ctx.args.content,
-        createdAt: ctx.stash.defaultValues.createdAt,
-        updatedAt: ctx.stash.defaultValues.updatedAt,
-    };
-    return response;
-  }`;
-    return graphql_transformer_core_1.MappingTemplate.inlineTemplateFromString((0, ts_dedent_1.dedent)(responseFunctionString));
+    return { TOOL_DEFINITIONS_LINE, TOOLS_CONFIGURATION_LINE };
 };
 const generateModelConfigurationLine = (config) => {
     const { aiModel, systemPrompt } = config;
-    return (0, ts_dedent_1.dedent) `const modelConfiguration = {
+    return (0, ts_dedent_1.default) `const modelConfiguration = {
     modelId: '${aiModel}',
     systemPrompt: ${JSON.stringify(systemPrompt)},
     ${generateModelInferenceConfigurationLine(config)}
@@ -97,7 +71,7 @@ const generateModelConfigurationLine = (config) => {
 const generateModelInferenceConfigurationLine = (config) => {
     const { inferenceConfiguration } = config;
     return inferenceConfiguration && Object.keys(inferenceConfiguration).length > 0
-        ? (0, ts_dedent_1.dedent) `inferenceConfiguration: ${JSON.stringify(config.inferenceConfiguration)},`
+        ? (0, ts_dedent_1.default) `inferenceConfiguration: ${JSON.stringify(config.inferenceConfiguration)},`
         : '';
 };
 const selectionSet = `id conversationId content { image { format source { bytes }} text toolUse { toolUseId name input } toolResult { status toolUseId content { json text image { format source { bytes }} document { format name source { bytes }} }}} role owner createdAt updatedAt`;

package/node_modules/@aws-amplify/graphql-conversation-transformer/lib/resolvers/invoke-lambda-resolver.js.map

@@ -1 +1 @@
-{"version":3,"file":"invoke-lambda-resolver.js","sourceRoot":"","sources":["../../src/resolvers/invoke-lambda-resolver.ts"],"names":[],"mappings":";;;AACA,oFAAwE;AAExE,yCAAmC;AAU5B,MAAM,2BAA2B,GAAG,CACzC,MAA0C,EAC1C,GAA+B,EACH,EAAE;IAC9B,MAAM,GAAG,GAAG,iCAAiC,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC3D,MAAM,GAAG,GAAG,kCAAkC,CAAC,MAAM,CAAC,CAAC;IACvD,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC;AACtB,CAAC,CAAC;AAPW,QAAA,2BAA2B,+BAOtC;AAUF,MAAM,iCAAiC,GAAG,CACxC,MAA0C,EAC1C,GAA+B,EACN,EAAE;IAC3B,MAAM,EAAE,6BAA6B,EAAE,oBAAoB,EAAE,GAAG,MAAM,CAAC;IACvE,MAAM,eAAe,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IACxD,MAAM,mBAAmB,GAAG,eAAe,CAAC,CAAC,CAAC,2BAA2B,eAAe,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;IACjG,MAAM,sBAAsB,GAAG,8BAA8B,CAAC,MAAM,CAAC,CAAC;IACtE,MAAM,eAAe,GAAG,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC;IAE3C,MAAM,sBAAsB,GAAG,eAAe;QAC5C,CAAC,CAAC,IAAA,kBAAM,EAAA;;;;OAIL;QACH,CAAC,CAAC,IAAA,kBAAM,EAAA;;OAEL,CAAC;IAEN,MAAM,qBAAqB,GAAG;;;;;MAK1B,mBAAmB;4BACG,YAAY;kCACN,eAAe;;;;eAIlC,oBAAoB;wBACX,6BAA6B;;;;MAI/C,sBAAsB;;;MAGtB,sBAAsB;;;;;;;;;;;;;;;;;;;IAmBxB,CAAC;IAEH,OAAO,0CAAe,CAAC,wBAAwB,CAAC,IAAA,kBAAM,EAAC,qBAAqB,CAAC,CAAC,CAAC;AACjF,CAAC,CAAC;AAEF,MAAM,kCAAkC,GAAG,CAAC,MAA0C,EAA2B,EAAE;IACjH,MAAM,sBAAsB,GAAG;;;;;;;;uBAQV,MAAM,CAAC,YAAY,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK;;;;;;;;;IAS9D,CAAC;IAEH,OAAO,0CAAe,CAAC,wBAAwB,CAAC,IAAA,kBAAM,EAAC,sBAAsB,CAAC,CAAC,CAAC;AAClF,CAAC,CAAC;AAQF,MAAM,8BAA8B,GAAG,CAAC,MAA0C,EAAE,EAAE;IACpF,MAAM,EAAE,OAAO,EAAE,YAAY,EAAE,GAAG,MAAM,CAAC;IAEzC,OAAO,IAAA,kBAAM,EAAA;gBACC,OAAO;oBACH,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC;MAC1C,uCAAuC,CAAC,MAAM,CAAC;KAChD,CAAC;AACN,CAAC,CAAC;AAQF,MAAM,uCAAuC,GAAG,CAAC,MAA0C,EAAE,EAAE;IAC7F,MAAM,EAAE,sBAAsB,EAAE,GAAG,MAAM,CAAC;IAC1C,OAAO,sBAAsB,IAAI,MAAM,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC,MAAM,GAAG,CAAC;QAC7E,CAAC,CAAC,IAAA,kBAAM,EAAA,2BAA2B,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,sBAAsB,CAAC,GAAG;QACnF,CAAC,CAAC,EAAE,CAAC;AACT,CAAC,CAAC;AAKF,MAAM,YAAY,GAAG,mQAAmQ,CAAC"}
+{"version":3,"file":"invoke-lambda-resolver.js","sourceRoot":"","sources":["../../src/resolvers/invoke-lambda-resolver.ts"],"names":[],"mappings":";;;;;;AACA,oFAAwE;AAExE,4CAAoB;AACpB,gDAAwB;AACxB,0DAA+B;AAC/B,2EAAqD;AACrD,0DAAkC;AAS3B,MAAM,2BAA2B,GAAG,CACzC,MAA0C,EAC1C,GAA+B,EACN,EAAE;IAC3B,MAAM,EAAE,qBAAqB,EAAE,wBAAwB,EAAE,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAAC;IACtF,MAAM,aAAa,GAAG,YAAY,CAAC;IACnC,MAAM,oBAAoB,GAAG,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC;IAChD,MAAM,wBAAwB,GAAG,8BAA8B,CAAC,MAAM,CAAC,CAAC;IACxE,MAAM,sBAAsB,GAAG,MAAM,CAAC,oBAAoB,CAAC;IAC3D,MAAM,iCAAiC,GAAG,MAAM,CAAC,6BAA6B,CAAC;IAC/E,MAAM,kBAAkB,GAAG,MAAM,CAAC,YAAY,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC;IAIvE,MAAM,cAAc,GAAG,yBAAyB,IAAA,oCAAO,EAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;IACnF,MAAM,yBAAyB,GAAG,IAAI,CAAC;IACvC,MAAM,eAAe,GAAG,0BAA0B,IAAA,oCAAO,EAAC,IAAA,mBAAS,EAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;IAChG,MAAM,0BAA0B,GAAG,2BAA2B,IAAA,oCAAO,EAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC;IAC5G,MAAM,gBAAgB,GAAG,WAAW,CAAC;IAErC,MAAM,aAAa,GAAG;QACpB,qBAAqB;QACrB,wBAAwB;QACxB,aAAa;QACb,oBAAoB;QACpB,wBAAwB;QACxB,sBAAsB;QACtB,iCAAiC;QACjC,kBAAkB;QAClB,cAAc;QACd,yBAAyB;QACzB,eAAe;QACf,0BAA0B;QAC1B,gBAAgB;KACjB,CAAC;IAEF,IAAI,QAAQ,GAAG,YAAE,CAAC,YAAY,CAAC,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,uCAAuC,CAAC,EAAE,MAAM,CAAC,CAAC;IACtG,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;QACrD,MAAM,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,MAAM,CAAC,SAAS,GAAG,QAAQ,EAAE,GAAG,CAAC,EAAE,KAAK,CAAC,CAAC;QAChF,QAAQ,GAAG,QAAQ,CAAC;IACtB,CAAC,CAAC,CAAC;IAKH,OAAO,0CAAe,CAAC,wBAAwB,CAAC,QAAQ,CAAC,CAAC;AAC5D,CAAC,CAAC;AA9CW,QAAA,2BAA2B,+BA8CtC;AAEF,MAAM,iBAAiB,GAAG,CAAC,MAA0C,EAAE,EAAE;IACvE,MAAM,eAAe,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IACxD,MAAM,qBAAqB,GAAG,eAAe,CAAC,CAAC,CAAC,2BAA2B,eAAe,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;IAEnG,MAAM,wBAAwB,GAAG,eAAe;QAC9C,CAAC,CAAC,IAAA,mBAAM,EAAA;;;;OAIL;QACH,CAAC,CAAC,IAAA,mBAAM,EAAA;;OAEL,CAAC;IAEN,OAAO,EAAE,qBAAqB,EAAE,wBAAwB,EAAE,CAAC;AAC7D,CAAC,CAAC;AAQF,MAAM,8BAA8B,GAAG,CAAC,MAA0C,EAAE,EAAE;IACpF,MAAM,EAAE,OAAO,EAAE,YAAY,EAAE,GAAG,MAAM,CAAC;IAEzC,OAAO,IAAA,mBAAM,EAAA;gBACC,OAAO;oBACH,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC;MAC1C,uCAAuC,CAAC,MAAM,CAAC;KAChD,CAAC;AACN,CAAC,CAAC;AAQF,MAAM,uCAAuC,GAAG,CAAC,MAA0C,EAAE,EAAE;IAC7F,MAAM,EAAE,sBAAsB,EAAE,GAAG,MAAM,CAAC;IAC1C,OAAO,sBAAsB,IAAI,MAAM,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC,MAAM,GAAG,CAAC;QAC7E,CAAC,CAAC,IAAA,mBAAM,EAAA,2BAA2B,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,sBAAsB,CAAC,GAAG;QACnF,CAAC,CAAC,EAAE,CAAC;AACT,CAAC,CAAC;AAKF,MAAM,YAAY,GAAG,mQAAmQ,CAAC"}

package/node_modules/@aws-amplify/graphql-conversation-transformer/lib/resolvers/list-messages-init-resolver.d.ts

@@ -0,0 +1,4 @@
+import { MappingTemplateProvider } from '@aws-amplify/graphql-transformer-interfaces';
+import { ConversationDirectiveConfiguration } from '../grapqhl-conversation-transformer';
+export declare const listMessageInitMappingTemplate: (config: ConversationDirectiveConfiguration) => MappingTemplateProvider;
+//# sourceMappingURL=list-messages-init-resolver.d.ts.map

package/node_modules/@aws-amplify/graphql-conversation-transformer/lib/resolvers/list-messages-init-resolver.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"list-messages-init-resolver.d.ts","sourceRoot":"","sources":["../../src/resolvers/list-messages-init-resolver.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,uBAAuB,EAAE,MAAM,6CAA6C,CAAC;AAGtF,OAAO,EAAE,kCAAkC,EAAE,MAAM,qCAAqC,CAAC;AAOzF,eAAO,MAAM,8BAA8B,WAAY,kCAAkC,KAAG,uBAI3F,CAAC"}

package/node_modules/@aws-amplify/graphql-conversation-transformer/lib/resolvers/list-messages-init-resolver.js

@@ -0,0 +1,16 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.listMessageInitMappingTemplate = void 0;
+const graphql_transformer_core_1 = require("@aws-amplify/graphql-transformer-core");
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const listMessageInitMappingTemplate = (config) => {
+    const resolver = fs_1.default.readFileSync(path_1.default.join(__dirname, 'list-messages-init-resolver-fn.js'), 'utf8');
+    const templateName = `Query.${config.field.name.value}.list-message-init.js`;
+    return graphql_transformer_core_1.MappingTemplate.s3MappingFunctionCodeFromString(resolver, templateName);
+};
+exports.listMessageInitMappingTemplate = listMessageInitMappingTemplate;
+//# sourceMappingURL=list-messages-init-resolver.js.map

package/node_modules/@aws-amplify/graphql-conversation-transformer/lib/resolvers/list-messages-init-resolver.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"list-messages-init-resolver.js","sourceRoot":"","sources":["../../src/resolvers/list-messages-init-resolver.ts"],"names":[],"mappings":";;;;;;AAAA,oFAAwE;AAExE,4CAAoB;AACpB,gDAAwB;AAQjB,MAAM,8BAA8B,GAAG,CAAC,MAA0C,EAA2B,EAAE;IACpH,MAAM,QAAQ,GAAG,YAAE,CAAC,YAAY,CAAC,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,mCAAmC,CAAC,EAAE,MAAM,CAAC,CAAC;IACpG,MAAM,YAAY,GAAG,SAAS,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,uBAAuB,CAAC;IAC7E,OAAO,0CAAe,CAAC,+BAA+B,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;AACjF,CAAC,CAAC;AAJW,QAAA,8BAA8B,kCAIzC"}

package/node_modules/@aws-amplify/graphql-conversation-transformer/lib/resolvers/verify-session-owner-resolver-fn.template.js

@@ -0,0 +1,31 @@
+export function request(ctx) {
+  const { authFilter } = ctx.stash;
+  const { conversationId } = [[CONVERSATION_ID_PARENT]];
+
+  const query = {
+    expression: 'id = :id',
+    expressionValues: util.dynamodb.toMapValues({
+      ':id': conversationId,
+    }),
+  };
+
+  const filter = JSON.parse(util.transform.toDynamoDBFilterExpression(authFilter));
+
+  return {
+    operation: 'Query',
+    query,
+    filter,
+  };
+}
+
+export function response(ctx) {
+  if (ctx.error) {
+    util.error(ctx.error.message, ctx.error.type);
+  }
+
+  if (ctx.result.items.length !== 0) {
+    return ctx.result.items[0];
+  }
+
+  util.error('Conversation not found', 'ResourceNotFound');
+}

package/node_modules/@aws-amplify/graphql-conversation-transformer/lib/resolvers/verify-session-owner-resolver.d.ts

@@ -1,3 +1,5 @@
-import { JSResolverFunctionProvider } from './js-resolver-function-provider';
-export declare const verifySessionOwnerMappingTemplate: () => JSResolverFunctionProvider;
+import { MappingTemplateProvider } from '@aws-amplify/graphql-transformer-interfaces';
+import { ConversationDirectiveConfiguration } from '../grapqhl-conversation-transformer';
+export declare const verifySessionOwnerSendMessageMappingTemplate: (config: ConversationDirectiveConfiguration) => MappingTemplateProvider;
+export declare const verifySessionOwnerAssistantResponseMappingTemplate: (config: ConversationDirectiveConfiguration) => MappingTemplateProvider;
 //# sourceMappingURL=verify-session-owner-resolver.d.ts.map

package/node_modules/@aws-amplify/graphql-conversation-transformer/lib/resolvers/verify-session-owner-resolver.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"verify-session-owner-resolver.d.ts","sourceRoot":"","sources":["../../src/resolvers/verify-session-owner-resolver.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,0BAA0B,EAAE,MAAM,iCAAiC,CAAC;AAO7E,eAAO,MAAM,iCAAiC,QAAO,0BAIpD,CAAC"}
+{"version":3,"file":"verify-session-owner-resolver.d.ts","sourceRoot":"","sources":["../../src/resolvers/verify-session-owner-resolver.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,uBAAuB,EAAE,MAAM,6CAA6C,CAAC;AAGtF,OAAO,EAAE,kCAAkC,EAAE,MAAM,qCAAqC,CAAC;AAOzF,eAAO,MAAM,4CAA4C,WAAY,kCAAkC,KAAG,uBAMzG,CAAC;AAOF,eAAO,MAAM,kDAAkD,WAAY,kCAAkC,KAAG,uBAM/G,CAAC"}

package/node_modules/@aws-amplify/graphql-conversation-transformer/lib/resolvers/verify-session-owner-resolver.js

@@ -1,50 +1,35 @@
 "use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.verifySessionOwnerMappingTemplate = void 0;
+exports.verifySessionOwnerAssistantResponseMappingTemplate = exports.verifySessionOwnerSendMessageMappingTemplate = void 0;
 const graphql_transformer_core_1 = require("@aws-amplify/graphql-transformer-core");
-const ts_dedent_1 = require("ts-dedent");
-const verifySessionOwnerMappingTemplate = () => {
-    const req = createVerifySessionOwnerRequestFunction();
-    const res = createVerifySessionOwnerResponseFunction();
-    return { req, res };
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const verifySessionOwnerSendMessageMappingTemplate = (config) => {
+    const substitutions = {
+        CONVERSATION_ID_PARENT: 'ctx.args',
+    };
+    const templateName = `Mutation.${config.field.name.value}.verify-session-owner.js`;
+    return verifySessionOwnerMappingTemplate(templateName, substitutions);
 };
-exports.verifySessionOwnerMappingTemplate = verifySessionOwnerMappingTemplate;
-const createVerifySessionOwnerRequestFunction = () => {
-    const requestFunctionString = `
-      export function request(ctx) {
-        const { authFilter } = ctx.stash;
-
-        const query = {
-          expression: 'id = :id',
-          expressionValues: util.dynamodb.toMapValues({
-            ':id': ctx.args.conversationId
-          })
-        };
-
-        const filter = JSON.parse(util.transform.toDynamoDBFilterExpression(authFilter));
-
-        return {
-          operation: 'Query',
-          query,
-          filter
-        };
-      }
-      `;
-    return graphql_transformer_core_1.MappingTemplate.inlineTemplateFromString((0, ts_dedent_1.dedent)(requestFunctionString));
+exports.verifySessionOwnerSendMessageMappingTemplate = verifySessionOwnerSendMessageMappingTemplate;
+const verifySessionOwnerAssistantResponseMappingTemplate = (config) => {
+    const substitutions = {
+        CONVERSATION_ID_PARENT: 'ctx.args.input',
+    };
+    const templateName = `Mutation.${config.field.name.value}AssistantResponse.verify-session-owner.js`;
+    return verifySessionOwnerMappingTemplate(templateName, substitutions);
 };
-const createVerifySessionOwnerResponseFunction = () => {
-    const responseFunctionString = `
-      export function response(ctx) {
-        if (ctx.error) {
-          util.error(ctx.error.message, ctx.error.type);
-        }
-
-        if (ctx.result.items.length !== 0) {
-          return ctx.result.items[0];
-        }
-
-        util.error('Conversation not found', 'ResourceNotFound');
-      }`;
-    return graphql_transformer_core_1.MappingTemplate.inlineTemplateFromString((0, ts_dedent_1.dedent)(responseFunctionString));
+exports.verifySessionOwnerAssistantResponseMappingTemplate = verifySessionOwnerAssistantResponseMappingTemplate;
+const verifySessionOwnerMappingTemplate = (name, substitute) => {
+    let resolver = fs_1.default.readFileSync(path_1.default.join(__dirname, 'verify-session-owner-resolver-fn.template.js'), 'utf8');
+    Object.entries(substitute).forEach(([key, value]) => {
+        const replaced = resolver.replace(new RegExp(`\\[\\[${key}\\]\\]`, 'g'), value);
+        resolver = replaced;
+    });
+    const templateName = `Mutation.${name}.verify-session-owner.js`;
+    return graphql_transformer_core_1.MappingTemplate.s3MappingFunctionCodeFromString(resolver, templateName);
 };
 //# sourceMappingURL=verify-session-owner-resolver.js.map

package/node_modules/@aws-amplify/graphql-conversation-transformer/lib/resolvers/verify-session-owner-resolver.js.map

@@ -1 +1 @@
-{"version":3,"file":"verify-session-owner-resolver.js","sourceRoot":"","sources":["../../src/resolvers/verify-session-owner-resolver.ts"],"names":[],"mappings":";;;AAAA,oFAAwE;AAExE,yCAAmC;AAQ5B,MAAM,iCAAiC,GAAG,GAA+B,EAAE;IAChF,MAAM,GAAG,GAAG,uCAAuC,EAAE,CAAC;IACtD,MAAM,GAAG,GAAG,wCAAwC,EAAE,CAAC;IACvD,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC;AACtB,CAAC,CAAC;AAJW,QAAA,iCAAiC,qCAI5C;AAOF,MAAM,uCAAuC,GAAG,GAA4B,EAAE;IAC5E,MAAM,qBAAqB,GAAG;;;;;;;;;;;;;;;;;;;OAmBzB,CAAC;IAEN,OAAO,0CAAe,CAAC,wBAAwB,CAAC,IAAA,kBAAM,EAAC,qBAAqB,CAAC,CAAC,CAAC;AACjF,CAAC,CAAC;AAOF,MAAM,wCAAwC,GAAG,GAA4B,EAAE;IAC7E,MAAM,sBAAsB,GAAG;;;;;;;;;;;QAWzB,CAAC;IAEP,OAAO,0CAAe,CAAC,wBAAwB,CAAC,IAAA,kBAAM,EAAC,sBAAsB,CAAC,CAAC,CAAC;AAClF,CAAC,CAAC"}
+{"version":3,"file":"verify-session-owner-resolver.js","sourceRoot":"","sources":["../../src/resolvers/verify-session-owner-resolver.ts"],"names":[],"mappings":";;;;;;AAAA,oFAAwE;AAExE,4CAAoB;AACpB,gDAAwB;AAQjB,MAAM,4CAA4C,GAAG,CAAC,MAA0C,EAA2B,EAAE;IAClI,MAAM,aAAa,GAAG;QACpB,sBAAsB,EAAE,UAAU;KACnC,CAAC;IACF,MAAM,YAAY,GAAG,YAAY,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,0BAA0B,CAAC;IACnF,OAAO,iCAAiC,CAAC,YAAY,EAAE,aAAa,CAAC,CAAC;AACxE,CAAC,CAAC;AANW,QAAA,4CAA4C,gDAMvD;AAOK,MAAM,kDAAkD,GAAG,CAAC,MAA0C,EAA2B,EAAE;IACxI,MAAM,aAAa,GAAG;QACpB,sBAAsB,EAAE,gBAAgB;KACzC,CAAC;IACF,MAAM,YAAY,GAAG,YAAY,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,2CAA2C,CAAC;IACpG,OAAO,iCAAiC,CAAC,YAAY,EAAE,aAAa,CAAC,CAAC;AACxE,CAAC,CAAC;AANW,QAAA,kDAAkD,sDAM7D;AAEF,MAAM,iCAAiC,GAAG,CAAC,IAAY,EAAE,UAAkC,EAAE,EAAE;IAC7F,IAAI,QAAQ,GAAG,YAAE,CAAC,YAAY,CAAC,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,8CAA8C,CAAC,EAAE,MAAM,CAAC,CAAC;IAC7G,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;QAClD,MAAM,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,MAAM,CAAC,SAAS,GAAG,QAAQ,EAAE,GAAG,CAAC,EAAE,KAAK,CAAC,CAAC;QAChF,QAAQ,GAAG,QAAQ,CAAC;IACtB,CAAC,CAAC,CAAC;IACH,MAAM,YAAY,GAAG,YAAY,IAAI,0BAA0B,CAAC;IAChE,OAAO,0CAAe,CAAC,+BAA+B,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;AACjF,CAAC,CAAC"}

package/node_modules/@aws-amplify/graphql-conversation-transformer/lib/resolvers/write-message-to-table-resolver-fn.template.js

@@ -0,0 +1,24 @@
+import { util } from '@aws-appsync/utils';
+import * as ddb from '@aws-appsync/utils/dynamodb';
+
+export function request(ctx) {
+  const args = ctx.stash.transformedArgs ?? ctx.args;
+  const defaultValues = ctx.stash.defaultValues ?? {};
+  const message = {
+    __typename: '[[CONVERSATION_MESSAGE_TYPE_NAME]]',
+    role: 'user',
+    ...args,
+    ...defaultValues,
+  };
+  const id = ctx.stash.defaultValues.id;
+
+  return ddb.put({ key: { id }, item: message });
+}
+
+export function response(ctx) {
+  if (ctx.error) {
+    util.error(ctx.error.message, ctx.error.type);
+  } else {
+    return ctx.result;
+  }
+}

package/node_modules/@aws-amplify/graphql-conversation-transformer/lib/resolvers/write-message-to-table-resolver.d.ts

@@ -1,3 +1,3 @@
-import { JSResolverFunctionProvider } from './js-resolver-function-provider';
-export declare const writeMessageToTableMappingTemplate: (fieldName: string) => JSResolverFunctionProvider;
+import { MappingTemplateProvider } from '@aws-amplify/graphql-transformer-interfaces';
+export declare const writeMessageToTableMappingTemplate: (fieldName: string) => MappingTemplateProvider;
 //# sourceMappingURL=write-message-to-table-resolver.d.ts.map

package/node_modules/@aws-amplify/graphql-conversation-transformer/lib/resolvers/write-message-to-table-resolver.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"write-message-to-table-resolver.d.ts","sourceRoot":"","sources":["../../src/resolvers/write-message-to-table-resolver.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,0BAA0B,EAAE,MAAM,iCAAiC,CAAC;AAQ7E,eAAO,MAAM,kCAAkC,cAAe,MAAM,KAAG,0BAItE,CAAC"}
+{"version":3,"file":"write-message-to-table-resolver.d.ts","sourceRoot":"","sources":["../../src/resolvers/write-message-to-table-resolver.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,uBAAuB,EAAE,MAAM,6CAA6C,CAAC;AAStF,eAAO,MAAM,kCAAkC,cAAe,MAAM,KAAG,uBAYtE,CAAC"}

package/node_modules/@aws-amplify/graphql-conversation-transformer/lib/resolvers/write-message-to-table-resolver.js

@@ -1,45 +1,23 @@
 "use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.writeMessageToTableMappingTemplate = void 0;
 const graphql_transformer_core_1 = require("@aws-amplify/graphql-transformer-core");
-const ts_dedent_1 = require("ts-dedent");
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
 const writeMessageToTableMappingTemplate = (fieldName) => {
-    const req = createWriteMessageToTableRequestFunction(fieldName);
-    const res = createWriteMessageToTableResponseFunction();
-    return { req, res };
+    const substitutions = {
+        CONVERSATION_MESSAGE_TYPE_NAME: `ConversationMessage${fieldName}`,
+    };
+    let resolver = fs_1.default.readFileSync(path_1.default.join(__dirname, 'write-message-to-table-resolver-fn.template.js'), 'utf8');
+    Object.entries(substitutions).forEach(([key, value]) => {
+        const replaced = resolver.replace(new RegExp(`\\[\\[${key}\\]\\]`, 'g'), value);
+        resolver = replaced;
+    });
+    const templateName = `Mutation.${fieldName}.write-message-to-table.js`;
+    return graphql_transformer_core_1.MappingTemplate.s3MappingFunctionCodeFromString(resolver, templateName);
 };
 exports.writeMessageToTableMappingTemplate = writeMessageToTableMappingTemplate;
-const createWriteMessageToTableRequestFunction = (fieldName) => {
-    const requestFunctionString = `
-    import { util } from '@aws-appsync/utils'
-    import * as ddb from '@aws-appsync/utils/dynamodb'
-
-    export function request(ctx) {
-      const args = ctx.stash.transformedArgs ?? ctx.args;
-      const defaultValues = ctx.stash.defaultValues ?? {};
-      const message = {
-          __typename: 'ConversationMessage${fieldName}',
-          role: 'user',
-          ...args,
-          ...defaultValues,
-      };
-      const id = ctx.stash.defaultValues.id;
-
-      return ddb.put({ key: { id }, item: message });
-    }
-    `;
-    return graphql_transformer_core_1.MappingTemplate.inlineTemplateFromString((0, ts_dedent_1.dedent)(requestFunctionString));
-};
-const createWriteMessageToTableResponseFunction = () => {
-    const responseFunctionString = `
-    export function response(ctx) {
-      if (ctx.error) {
-        util.error(ctx.error.message, ctx.error.type);
-      } else {
-        return ctx.result;
-      }
-    }
-    `;
-    return graphql_transformer_core_1.MappingTemplate.inlineTemplateFromString((0, ts_dedent_1.dedent)(responseFunctionString));
-};
 //# sourceMappingURL=write-message-to-table-resolver.js.map

package/node_modules/@aws-amplify/graphql-conversation-transformer/lib/resolvers/write-message-to-table-resolver.js.map

@@ -1 +1 @@
-{"version":3,"file":"write-message-to-table-resolver.js","sourceRoot":"","sources":["../../src/resolvers/write-message-to-table-resolver.ts"],"names":[],"mappings":";;;AAAA,oFAAwE;AAExE,yCAAmC;AAS5B,MAAM,kCAAkC,GAAG,CAAC,SAAiB,EAA8B,EAAE;IAClG,MAAM,GAAG,GAAG,wCAAwC,CAAC,SAAS,CAAC,CAAC;IAChE,MAAM,GAAG,GAAG,yCAAyC,EAAE,CAAC;IACxD,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC;AACtB,CAAC,CAAC;AAJW,QAAA,kCAAkC,sCAI7C;AAQF,MAAM,wCAAwC,GAAG,CAAC,SAAiB,EAA2B,EAAE;IAC9F,MAAM,qBAAqB,GAAG;;;;;;;;4CAQY,SAAS;;;;;;;;;KAShD,CAAC;IAEJ,OAAO,0CAAe,CAAC,wBAAwB,CAAC,IAAA,kBAAM,EAAC,qBAAqB,CAAC,CAAC,CAAC;AACjF,CAAC,CAAC;AAOF,MAAM,yCAAyC,GAAG,GAA4B,EAAE;IAC9E,MAAM,sBAAsB,GAAG;;;;;;;;KAQ5B,CAAC;IAEJ,OAAO,0CAAe,CAAC,wBAAwB,CAAC,IAAA,kBAAM,EAAC,sBAAsB,CAAC,CAAC,CAAC;AAClF,CAAC,CAAC"}
+{"version":3,"file":"write-message-to-table-resolver.js","sourceRoot":"","sources":["../../src/resolvers/write-message-to-table-resolver.ts"],"names":[],"mappings":";;;;;;AAAA,oFAAwE;AAExE,4CAAoB;AACpB,gDAAwB;AAOjB,MAAM,kCAAkC,GAAG,CAAC,SAAiB,EAA2B,EAAE;IAC/F,MAAM,aAAa,GAAG;QACpB,8BAA8B,EAAE,sBAAsB,SAAS,EAAE;KAClE,CAAC;IACF,IAAI,QAAQ,GAAG,YAAE,CAAC,YAAY,CAAC,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,gDAAgD,CAAC,EAAE,MAAM,CAAC,CAAC;IAC/G,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;QACrD,MAAM,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,MAAM,CAAC,SAAS,GAAG,QAAQ,EAAE,GAAG,CAAC,EAAE,KAAK,CAAC,CAAC;QAChF,QAAQ,GAAG,QAAQ,CAAC;IACtB,CAAC,CAAC,CAAC;IAEH,MAAM,YAAY,GAAG,YAAY,SAAS,4BAA4B,CAAC;IACvE,OAAO,0CAAe,CAAC,+BAA+B,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;AACjF,CAAC,CAAC;AAZW,QAAA,kCAAkC,sCAY7C"}

package/node_modules/@aws-amplify/graphql-conversation-transformer/lib/transformer-steps/conversation-resolver-generator.d.ts

@@ -14,8 +14,8 @@ export declare class ConversationResolverGenerator {
     private createAssistantResponseResolver;
     private createAssistantResponseSubscriptionResolver;
     private addLambdaDataSource;
-    private setupMessageTableIndex;
     private addPostProcessingSlotToListMessagesPipeline;
+    private setupMessageTableIndex;
     private addGlobalSecondaryIndex;
 }
 //# sourceMappingURL=conversation-resolver-generator.d.ts.map

package/node_modules/@aws-amplify/graphql-conversation-transformer/lib/transformer-steps/conversation-resolver-generator.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"conversation-resolver-generator.d.ts","sourceRoot":"","sources":["../../src/transformer-steps/conversation-resolver-generator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,0BAA0B,EAAE,MAAM,6CAA6C,CAAC;AACzF,OAAO,EAAE,kCAAkC,EAAE,MAAM,qCAAqC,CAAC;AA2BzF,qBAAa,6BAA6B;IACxC,iBAAiB,CAAC,UAAU,EAAE,kCAAkC,EAAE,EAAE,GAAG,EAAE,0BAA0B,GAAG,IAAI;IAQ1G,OAAO,CAAC,wBAAwB;IAOhC,OAAO,CAAC,6BAA6B;IA0BrC,OAAO,CAAC,mBAAmB;IAW3B,OAAO,CAAC,uBAAuB;IAkB/B,OAAO,CAAC,+BAA+B;IAiBvC,OAAO,CAAC,iBAAiB;IAYzB,OAAO,CAAC,+BAA+B;IAgCvC,OAAO,CAAC,kCAAkC;IAmC1C,OAAO,CAAC,4BAA4B;IAyDpC,OAAO,CAAC,+BAA+B;IA+BvC,OAAO,CAAC,2CAA2C;IAgCnD,OAAO,CAAC,mBAAmB;IAkB3B,OAAO,CAAC,sBAAsB;IAwB9B,OAAO,CAAC,2CAA2C;IAanD,OAAO,CAAC,uBAAuB;CAgChC"}
+{"version":3,"file":"conversation-resolver-generator.d.ts","sourceRoot":"","sources":["../../src/transformer-steps/conversation-resolver-generator.ts"],"names":[],"mappings":"AAAA,OAAO,EAA2B,0BAA0B,EAAE,MAAM,6CAA6C,CAAC;AAClH,OAAO,EAAE,kCAAkC,EAAE,MAAM,qCAAqC,CAAC;AA4BzF,qBAAa,6BAA6B;IACxC,iBAAiB,CAAC,UAAU,EAAE,kCAAkC,EAAE,EAAE,GAAG,EAAE,0BAA0B,GAAG,IAAI;IAQ1G,OAAO,CAAC,wBAAwB;IAOhC,OAAO,CAAC,6BAA6B;IA8CrC,OAAO,CAAC,mBAAmB;IAW3B,OAAO,CAAC,uBAAuB;IAkB/B,OAAO,CAAC,+BAA+B;IAiBvC,OAAO,CAAC,iBAAiB;IAYzB,OAAO,CAAC,+BAA+B;IAgCvC,OAAO,CAAC,kCAAkC;IA2C1C,OAAO,CAAC,4BAA4B;IAkCpC,OAAO,CAAC,+BAA+B;IA2CvC,OAAO,CAAC,2CAA2C;IAqCnD,OAAO,CAAC,mBAAmB;IAa3B,OAAO,CAAC,2CAA2C;IAqBnD,OAAO,CAAC,sBAAsB;IAsB9B,OAAO,CAAC,uBAAuB;CAgChC"}

package/node_modules/@aws-amplify/graphql-conversation-transformer/lib/transformer-steps/conversation-resolver-generator.js

@@ -38,13 +38,12 @@ const init_resolver_1 = require("../resolvers/init-resolver");
 const auth_resolver_1 = require("../resolvers/auth-resolver");
 const verify_session_owner_resolver_1 = require("../resolvers/verify-session-owner-resolver");
 const write_message_to_table_resolver_1 = require("../resolvers/write-message-to-table-resolver");
-const message_history_resolver_1 = require("../resolvers/message-history-resolver");
 const invoke_lambda_resolver_1 = require("../resolvers/invoke-lambda-resolver");
 const assistant_mutation_resolver_1 = require("../resolvers/assistant-mutation-resolver");
 const assistant_messages_subscription_resolver_1 = require("../resolvers/assistant-messages-subscription-resolver");
 const graphql_index_transformer_1 = require("@aws-amplify/graphql-index-transformer");
 const pluralize_1 = __importDefault(require("pluralize"));
-const list_message_post_data_load_resolver_1 = require("../resolvers/list-message-post-data-load-resolver");
+const list_messages_init_resolver_1 = require("../resolvers/list-messages-init-resolver");
 class ConversationResolverGenerator {
     generateResolvers(directives, ctx) {
         for (const directive of directives) {
@@ -66,12 +65,16 @@ class ConversationResolverGenerator {
         const fieldName = field.name.value;
         const functionStack = this.createFunctionStack(ctx, capitalizedFieldName);
         const { functionDataSourceId, referencedFunction } = this.setupFunctionDataSource(directive, functionStack, capitalizedFieldName);
-        this.createAssistantResponseResolver(ctx, directive, capitalizedFieldName);
-        this.createAssistantResponseSubscriptionResolver(ctx, capitalizedFieldName);
         const functionDataSource = this.addLambdaDataSource(ctx, functionDataSourceId, referencedFunction, capitalizedFieldName);
         const invokeLambdaFunction = (0, invoke_lambda_resolver_1.invokeLambdaMappingTemplate)(directive, ctx);
         this.setupMessageTableIndex(ctx, directive);
-        this.createConversationPipelineResolver(ctx, parentName, fieldName, capitalizedFieldName, functionDataSource, invokeLambdaFunction);
+        const initResolverFunction = (0, init_resolver_1.initMappingTemplate)(directive);
+        const authResolverFunction = (0, auth_resolver_1.authMappingTemplate)(directive);
+        const verifySessionOwnerSendMessageResolverFunction = (0, verify_session_owner_resolver_1.verifySessionOwnerSendMessageMappingTemplate)(directive);
+        const verifySessionOwnerAssistantResponseResolverFunction = (0, verify_session_owner_resolver_1.verifySessionOwnerAssistantResponseMappingTemplate)(directive);
+        this.createConversationPipelineResolver(ctx, parentName, fieldName, capitalizedFieldName, functionDataSource, invokeLambdaFunction, initResolverFunction, authResolverFunction, verifySessionOwnerSendMessageResolverFunction);
+        this.createAssistantResponseResolver(ctx, directive, capitalizedFieldName, initResolverFunction, authResolverFunction, verifySessionOwnerAssistantResponseResolverFunction);
+        this.createAssistantResponseSubscriptionResolver(ctx, directive, capitalizedFieldName);
     }
     createFunctionStack(ctx, capitalizedFieldName) {
         return ctx.stackManager.createStack(`${capitalizedFieldName}ConversationDirectiveLambdaStack`);
@@ -106,52 +109,65 @@ class ConversationResolverGenerator {
         const referencedFunction = defaultConversationHandler.resources.lambda;
         return { functionDataSourceId, referencedFunction };
     }
-    createConversationPipelineResolver(ctx, parentName, fieldName, capitalizedFieldName, functionDataSource, invokeLambdaFunction) {
+    createConversationPipelineResolver(ctx, parentName, fieldName, capitalizedFieldName, functionDataSource, invokeLambdaFunction, initResolverFunction, authResolverFunction, verifySessionOwnerResolverFunction) {
         const resolverResourceId = graphql_transformer_common_1.ResolverResourceIDs.ResolverResourceID(parentName, fieldName);
-        const runtime = { name: 'APPSYNC_JS', runtimeVersion: '1.0.0' };
-        const conversationPipelineResolver = new graphql_transformer_core_1.TransformerResolver(parentName, fieldName, resolverResourceId, invokeLambdaFunction.req, invokeLambdaFunction.res, ['init', 'auth', 'verifySessionOwner', 'writeMessageToTable', 'retrieveMessageHistory'], ['handleLambdaResponse', 'finish'], functionDataSource, runtime);
-        this.addPipelineResolverFunctions(ctx, conversationPipelineResolver, capitalizedFieldName, runtime);
+        const runtime = graphql_transformer_core_1.APPSYNC_JS_RUNTIME;
+        const conversationPipelineResolver = new graphql_transformer_core_1.TransformerResolver(parentName, fieldName, resolverResourceId, { codeMappingTemplate: invokeLambdaFunction }, ['init', 'auth', 'verifySessionOwner', 'writeMessageToTable'], ['handleLambdaResponse', 'finish'], functionDataSource, runtime);
+        this.addPipelineResolverFunctions(ctx, conversationPipelineResolver, capitalizedFieldName, initResolverFunction, authResolverFunction, verifySessionOwnerResolverFunction);
         ctx.resolvers.addResolver(parentName, fieldName, conversationPipelineResolver);
     }
-    addPipelineResolverFunctions(ctx, resolver, capitalizedFieldName, runtime) {
-        const initFunction = (0, init_resolver_1.initMappingTemplate)();
-        resolver.addToSlot('init', initFunction.req, initFunction.res, undefined, runtime);
-        const authFunction = (0, auth_resolver_1.authMappingTemplate)();
-        resolver.addToSlot('auth', authFunction.req, authFunction.res, undefined, runtime);
-        const verifySessionOwnerFunction = (0, verify_session_owner_resolver_1.verifySessionOwnerMappingTemplate)();
+    addPipelineResolverFunctions(ctx, resolver, capitalizedFieldName, initResolverFunction, authResolverFunction, verifySessionOwnerResolverFunction) {
+        resolver.addJsFunctionToSlot('init', initResolverFunction);
+        resolver.addJsFunctionToSlot('auth', authResolverFunction);
         const sessionModelName = `Conversation${capitalizedFieldName}`;
         const sessionModelDDBDataSourceName = (0, graphql_transformer_core_2.getModelDataSourceNameForTypeName)(ctx, sessionModelName);
         const conversationSessionDDBDataSource = ctx.api.host.getDataSource(sessionModelDDBDataSourceName);
-        resolver.addToSlot('verifySessionOwner', verifySessionOwnerFunction.req, verifySessionOwnerFunction.res, conversationSessionDDBDataSource, runtime);
+        resolver.addJsFunctionToSlot('verifySessionOwner', verifySessionOwnerResolverFunction, conversationSessionDDBDataSource);
         const writeMessageToTableFunction = (0, write_message_to_table_resolver_1.writeMessageToTableMappingTemplate)(capitalizedFieldName);
         const messageModelName = `ConversationMessage${capitalizedFieldName}`;
         const messageModelDDBDataSourceName = (0, graphql_transformer_core_2.getModelDataSourceNameForTypeName)(ctx, messageModelName);
         const messageDDBDataSource = ctx.api.host.getDataSource(messageModelDDBDataSourceName);
-        resolver.addToSlot('writeMessageToTable', writeMessageToTableFunction.req, writeMessageToTableFunction.res, messageDDBDataSource, runtime);
-        const retrieveMessageHistoryFunction = (0, message_history_resolver_1.readHistoryMappingTemplate)();
-        resolver.addToSlot('retrieveMessageHistory', retrieveMessageHistoryFunction.req, retrieveMessageHistoryFunction.res, messageDDBDataSource, runtime);
+        resolver.addJsFunctionToSlot('writeMessageToTable', writeMessageToTableFunction, messageDDBDataSource);
     }
-    createAssistantResponseResolver(ctx, directive, capitalizedFieldName) {
+    createAssistantResponseResolver(ctx, directive, capitalizedFieldName, initResolverFunction, authResolverFunction, verifySessionOwnerResolverFunction) {
         const assistantResponseResolverResourceId = graphql_transformer_common_1.ResolverResourceIDs.ResolverResourceID('Mutation', directive.responseMutationName);
-        const assistantResponseResolverFunction = (0, assistant_mutation_resolver_1.assistantMutationResolver)();
+        const assistantResponseResolverFunction = (0, assistant_mutation_resolver_1.assistantMutationResolver)(directive);
         const conversationMessageDataSourceName = (0, graphql_transformer_core_2.getModelDataSourceNameForTypeName)(ctx, `ConversationMessage${capitalizedFieldName}`);
         const conversationMessageDataSource = ctx.api.host.getDataSource(conversationMessageDataSourceName);
-        const runtime = { name: 'APPSYNC_JS', runtimeVersion: '1.0.0' };
-        const assistantResponseResolver = new graphql_transformer_core_1.TransformerResolver('Mutation', directive.responseMutationName, assistantResponseResolverResourceId, assistantResponseResolverFunction.req, assistantResponseResolverFunction.res, [], [], conversationMessageDataSource, runtime);
-        ctx.resolvers.addResolver('Mutation', directive.responseMutationName, assistantResponseResolver);
+        const resolver = new graphql_transformer_core_1.TransformerResolver('Mutation', directive.responseMutationName, assistantResponseResolverResourceId, { codeMappingTemplate: assistantResponseResolverFunction }, ['init', 'auth', 'verifySessionOwner'], [], conversationMessageDataSource, graphql_transformer_core_1.APPSYNC_JS_RUNTIME);
+        resolver.addJsFunctionToSlot('init', initResolverFunction);
+        resolver.addJsFunctionToSlot('auth', authResolverFunction);
+        const sessionModelName = `Conversation${capitalizedFieldName}`;
+        const sessionModelDDBDataSourceName = (0, graphql_transformer_core_2.getModelDataSourceNameForTypeName)(ctx, sessionModelName);
+        const conversationSessionDDBDataSource = ctx.api.host.getDataSource(sessionModelDDBDataSourceName);
+        resolver.addJsFunctionToSlot('verifySessionOwner', verifySessionOwnerResolverFunction, conversationSessionDDBDataSource);
+        ctx.resolvers.addResolver('Mutation', directive.responseMutationName, resolver);
     }
-    createAssistantResponseSubscriptionResolver(ctx, capitalizedFieldName) {
+    createAssistantResponseSubscriptionResolver(ctx, directive, capitalizedFieldName) {
         const onAssistantResponseSubscriptionFieldName = `onCreateAssistantResponse${capitalizedFieldName}`;
         const onAssistantResponseSubscriptionResolverResourceId = graphql_transformer_common_1.ResolverResourceIDs.ResolverResourceID('Subscription', onAssistantResponseSubscriptionFieldName);
-        const onAssistantResponseSubscriptionResolverFunction = (0, assistant_messages_subscription_resolver_1.conversationMessageSubscriptionMappingTamplate)();
-        const runtime = { name: 'APPSYNC_JS', runtimeVersion: '1.0.0' };
-        const onAssistantResponseSubscriptionResolver = new graphql_transformer_core_1.TransformerResolver('Subscription', onAssistantResponseSubscriptionFieldName, onAssistantResponseSubscriptionResolverResourceId, onAssistantResponseSubscriptionResolverFunction.req, onAssistantResponseSubscriptionResolverFunction.res, [], [], undefined, runtime);
+        const onAssistantResponseSubscriptionResolverFunction = (0, assistant_messages_subscription_resolver_1.conversationMessageSubscriptionMappingTamplate)(directive);
+        const mappingTemplate = {
+            codeMappingTemplate: onAssistantResponseSubscriptionResolverFunction,
+        };
+        const onAssistantResponseSubscriptionResolver = new graphql_transformer_core_1.TransformerResolver('Subscription', onAssistantResponseSubscriptionFieldName, onAssistantResponseSubscriptionResolverResourceId, mappingTemplate, [], [], undefined, graphql_transformer_core_1.APPSYNC_JS_RUNTIME);
         ctx.resolvers.addResolver('Subscription', onAssistantResponseSubscriptionFieldName, onAssistantResponseSubscriptionResolver);
     }
     addLambdaDataSource(ctx, functionDataSourceId, referencedFunction, capitalizedFieldName) {
         const functionDataSourceScope = ctx.stackManager.getScopeFor(functionDataSourceId, `${capitalizedFieldName}ConversationDirectiveLambdaStack`);
         return ctx.api.host.addLambdaDataSource(functionDataSourceId, referencedFunction, {}, functionDataSourceScope);
     }
+    addPostProcessingSlotToListMessagesPipeline(ctx, directive) {
+        const messageModelName = directive.messageModel.messageModel.name.value;
+        const pluralized = (0, pluralize_1.default)(messageModelName);
+        const listMessagesResolver = ctx.resolvers.getResolver('Query', `list${pluralized}`);
+        const initResolverFn = (0, list_messages_init_resolver_1.listMessageInitMappingTemplate)(directive);
+        if (listMessagesResolver) {
+            listMessagesResolver.addJsFunctionToSlot('init', initResolverFn);
+        }
+        else {
+        }
+    }
     setupMessageTableIndex(ctx, directive) {
         const messageModelName = directive.messageModel.messageModel.name.value;
         const referenceFieldName = 'conversationId';
@@ -164,15 +180,6 @@ class ConversationResolverGenerator {
         const indexName = 'gsi-ConversationMessage.conversationId.createdAt';
         this.addGlobalSecondaryIndex(conversationMessagesTable, indexName, { name: gsiPartitionKeyName, type: gsiPartitionKeyType }, { name: gsiSortKeyName, type: gsiSortKeyType }, ctx, messageModelName);
     }
-    addPostProcessingSlotToListMessagesPipeline(ctx, directive) {
-        const messageModelName = directive.messageModel.messageModel.name.value;
-        const pluralized = (0, pluralize_1.default)(messageModelName);
-        const { req, res } = (0, list_message_post_data_load_resolver_1.listMessagePostDataLoadMappingTemplate)();
-        const listMessagesResolver = ctx.resolvers.getResolver('Query', `list${pluralized}`);
-        if (listMessagesResolver) {
-            listMessagesResolver.addToSlot('postDataLoad', req, res, undefined, { name: 'APPSYNC_JS', runtimeVersion: '1.0.0' });
-        }
-    }
     addGlobalSecondaryIndex(table, indexName, partitionKey, sortKey, ctx, typeName) {
         table.addGlobalSecondaryIndex({
             indexName,