@aws-amplify/data-schema 1.4.1 → 1.5.1

package/dist/cjs/ClientSchema/ai/ClientConversation.js

@@ -0,0 +1,6 @@
+'use strict';
+
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=ClientConversation.js.map

package/dist/cjs/ClientSchema/ai/ClientConversation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ClientConversation.js","sources":["../../../../src/ClientSchema/ai/ClientConversation.ts"],"sourcesContent":["\"use strict\";\n// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.\n// SPDX-License-Identifier: Apache-2.0\nObject.defineProperty(exports, \"__esModule\", { value: true });\n"],"names":[],"mappings":";;AACA;AACA;AACA,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,YAAY,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;;"}

package/dist/cjs/CustomOperation.js

@@ -1,21 +1,23 @@
 'use strict';
 
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.subscription = exports.mutation = exports.query = exports.CustomOperationNames = void 0;
+exports.generation = exports.subscription = exports.mutation = exports.query = exports.CustomOperationNames = void 0;
 const util_1 = require("./util");
 const Authorization_1 = require("./Authorization");
 const queryBrand = 'queryCustomOperation';
 const mutationBrand = 'mutationCustomOperation';
 const subscriptionBrand = 'subscriptionCustomOperation';
+const generationBrand = 'generationCustomOperation';
 exports.CustomOperationNames = [
     'Query',
     'Mutation',
     'Subscription',
+    'Generation',
 ];
 function brandedBuilder(builder, brandValue) {
     return { ...builder, ...(0, util_1.brand)(brandValue) };
 }
-function _custom(typeName, brand) {
+function _custom(typeName, brand, input) {
     const data = {
         arguments: {},
         returnType: null,
@@ -23,6 +25,7 @@ function _custom(typeName, brand) {
         typeName: typeName,
         handlers: null,
         subscriptionSource: [],
+        input,
     };
     const builder = brandedBuilder({
         arguments(args) {
@@ -109,4 +112,21 @@ function subscription() {
     return _custom('Subscription', subscriptionBrand);
 }
 exports.subscription = subscription;
+/**
+ * @experimental
+ *
+ * Define an AI generation route for single request-response interaction with specified AI model.
+ * @example
+ * makeRecipe: a.generation({
+ *   aiModel: { resourcePath },
+ *   systemPrompt: 'Please make a recipe from the provided ingredients',
+ * })
+ *   .arguments({ ingredients: a.string().array() })
+ *   .returns(a.ref("Recipe"))
+ * @returns a generation route definition
+ */
+function generation(input) {
+    return _custom('Generation', generationBrand, input);
+}
+exports.generation = generation;
 //# sourceMappingURL=CustomOperation.js.map

package/dist/cjs/CustomOperation.js.map

@@ -1 +1 @@
-{"version":3,"file":"CustomOperation.js","sources":["../../src/CustomOperation.ts"],"sourcesContent":["\"use strict\";\nObject.defineProperty(exports, \"__esModule\", { value: true });\nexports.subscription = exports.mutation = exports.query = exports.CustomOperationNames = void 0;\nconst util_1 = require(\"./util\");\nconst Authorization_1 = require(\"./Authorization\");\nconst queryBrand = 'queryCustomOperation';\nconst mutationBrand = 'mutationCustomOperation';\nconst subscriptionBrand = 'subscriptionCustomOperation';\nexports.CustomOperationNames = [\n    'Query',\n    'Mutation',\n    'Subscription',\n];\nfunction brandedBuilder(builder, brandValue) {\n    return { ...builder, ...(0, util_1.brand)(brandValue) };\n}\nfunction _custom(typeName, brand) {\n    const data = {\n        arguments: {},\n        returnType: null,\n        authorization: [],\n        typeName: typeName,\n        handlers: null,\n        subscriptionSource: [],\n    };\n    const builder = brandedBuilder({\n        arguments(args) {\n            data.arguments = args;\n            return this;\n        },\n        returns(returnType) {\n            data.returnType = returnType;\n            return this;\n        },\n        authorization(callback) {\n            const rules = callback(Authorization_1.allowForCustomOperations);\n            data.authorization = Array.isArray(rules) ? rules : [rules];\n            return this;\n        },\n        handler(handlers) {\n            data.handlers = Array.isArray(handlers)\n                ? handlers\n                : [handlers];\n            return this;\n        },\n        for(source) {\n            data.subscriptionSource = Array.isArray(source) ? source : [source];\n            return this;\n        },\n    }, brand);\n    return { ...builder, data };\n}\n/**\n * Use a custom query to define an API request that will retrieve backend data.\n * @see {@link https://docs.amplify.aws/react/build-a-backend/data/custom-business-logic/}\n * @example\n * const schema = a.schema({\n *   echo: a\n *     .query()\n *     .arguments({ content: a.string() })\n *     .returns(a.ref('EchoResponse'))\n *     .authorization(allow => [allow.publicApiKey()])\n *     // 3. set the function has the handler\n *     .handler(a.handler.function(echoHandler)),\n *\n *   EchoResponse: a.customType({\n *     content: a.string(),\n *     executionDuration: a.float()\n *   }),\n * });\n * @returns a custom query\n */\nfunction query() {\n    return _custom('Query', queryBrand);\n}\nexports.query = query;\n/**\n * Use a custom mutation to define an API request that will modify backend data or trigger a subscription event.\n * @see {@link https://docs.amplify.aws/react/build-a-backend/data/custom-business-logic/}\n * @example\n * likePost: a\n *   .mutation()\n *   .arguments({ postId: a.string() })\n *   .returns(a.ref('Post'))\n *   .authorization(allow => [allow.publicApiKey()])\n *   .handler(a.handler.function(echoHandler))\n * @returns a custom mutation\n */\nfunction mutation() {\n    return _custom('Mutation', mutationBrand);\n}\nexports.mutation = mutation;\n/**\n * Define a custom subscription to receive an event when a mutation is triggered\n * @see {@link https://docs.amplify.aws/react/build-a-backend/data/custom-subscription/}\n * @example\n * // Subscribe to incoming messages\n * receive: a.subscription()\n *   // subscribes to the 'publish' mutation\n *   .for(a.ref('publish'))\n *   // subscription handler to set custom filters\n *   .handler(a.handler.custom({entry: './receive.js'}))\n *   // authorization rules as to who can subscribe to the data\n *   .authorization(allow => [allow.publicApiKey()]),\n * @returns a custom subscription\n */\nfunction subscription() {\n    return _custom('Subscription', subscriptionBrand);\n}\nexports.subscription = subscription;\n"],"names":[],"mappings":";;AACA,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,YAAY,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;AAC9D,OAAO,CAAC,YAAY,GAAG,OAAO,CAAC,QAAQ,GAAG,OAAO,CAAC,KAAK,GAAG,OAAO,CAAC,oBAAoB,GAAG,KAAK,CAAC,CAAC;AAChG,MAAM,MAAM,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;AACjC,MAAM,eAAe,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAAC;AACnD,MAAM,UAAU,GAAG,sBAAsB,CAAC;AAC1C,MAAM,aAAa,GAAG,yBAAyB,CAAC;AAChD,MAAM,iBAAiB,GAAG,6BAA6B,CAAC;AACxD,OAAO,CAAC,oBAAoB,GAAG;AAC/B,IAAI,OAAO;AACX,IAAI,UAAU;AACd,IAAI,cAAc;AAClB,CAAC,CAAC;AACF,SAAS,cAAc,CAAC,OAAO,EAAE,UAAU,EAAE;AAC7C,IAAI,OAAO,EAAE,GAAG,OAAO,EAAE,GAAG,IAAI,MAAM,CAAC,KAAK,EAAE,UAAU,CAAC,EAAE,CAAC;AAC5D,CAAC;AACD,SAAS,OAAO,CAAC,QAAQ,EAAE,KAAK,EAAE;AAClC,IAAI,MAAM,IAAI,GAAG;AACjB,QAAQ,SAAS,EAAE,EAAE;AACrB,QAAQ,UAAU,EAAE,IAAI;AACxB,QAAQ,aAAa,EAAE,EAAE;AACzB,QAAQ,QAAQ,EAAE,QAAQ;AAC1B,QAAQ,QAAQ,EAAE,IAAI;AACtB,QAAQ,kBAAkB,EAAE,EAAE;AAC9B,KAAK,CAAC;AACN,IAAI,MAAM,OAAO,GAAG,cAAc,CAAC;AACnC,QAAQ,SAAS,CAAC,IAAI,EAAE;AACxB,YAAY,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;AAClC,YAAY,OAAO,IAAI,CAAC;AACxB,SAAS;AACT,QAAQ,OAAO,CAAC,UAAU,EAAE;AAC5B,YAAY,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;AACzC,YAAY,OAAO,IAAI,CAAC;AACxB,SAAS;AACT,QAAQ,aAAa,CAAC,QAAQ,EAAE;AAChC,YAAY,MAAM,KAAK,GAAG,QAAQ,CAAC,eAAe,CAAC,wBAAwB,CAAC,CAAC;AAC7E,YAAY,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,KAAK,GAAG,CAAC,KAAK,CAAC,CAAC;AACxE,YAAY,OAAO,IAAI,CAAC;AACxB,SAAS;AACT,QAAQ,OAAO,CAAC,QAAQ,EAAE;AAC1B,YAAY,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC;AACnD,kBAAkB,QAAQ;AAC1B,kBAAkB,CAAC,QAAQ,CAAC,CAAC;AAC7B,YAAY,OAAO,IAAI,CAAC;AACxB,SAAS;AACT,QAAQ,GAAG,CAAC,MAAM,EAAE;AACpB,YAAY,IAAI,CAAC,kBAAkB,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,MAAM,GAAG,CAAC,MAAM,CAAC,CAAC;AAChF,YAAY,OAAO,IAAI,CAAC;AACxB,SAAS;AACT,KAAK,EAAE,KAAK,CAAC,CAAC;AACd,IAAI,OAAO,EAAE,GAAG,OAAO,EAAE,IAAI,EAAE,CAAC;AAChC,CAAC;AACD;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,SAAS,KAAK,GAAG;AACjB,IAAI,OAAO,OAAO,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;AACxC,CAAC;AACD,OAAO,CAAC,KAAK,GAAG,KAAK,CAAC;AACtB;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,SAAS,QAAQ,GAAG;AACpB,IAAI,OAAO,OAAO,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC;AAC9C,CAAC;AACD,OAAO,CAAC,QAAQ,GAAG,QAAQ,CAAC;AAC5B;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,SAAS,YAAY,GAAG;AACxB,IAAI,OAAO,OAAO,CAAC,cAAc,EAAE,iBAAiB,CAAC,CAAC;AACtD,CAAC;AACD,OAAO,CAAC,YAAY,GAAG,YAAY;;"}
+{"version":3,"file":"CustomOperation.js","sources":["../../src/CustomOperation.ts"],"sourcesContent":["\"use strict\";\nObject.defineProperty(exports, \"__esModule\", { value: true });\nexports.generation = exports.subscription = exports.mutation = exports.query = exports.CustomOperationNames = void 0;\nconst util_1 = require(\"./util\");\nconst Authorization_1 = require(\"./Authorization\");\nconst queryBrand = 'queryCustomOperation';\nconst mutationBrand = 'mutationCustomOperation';\nconst subscriptionBrand = 'subscriptionCustomOperation';\nconst generationBrand = 'generationCustomOperation';\nexports.CustomOperationNames = [\n    'Query',\n    'Mutation',\n    'Subscription',\n    'Generation',\n];\nfunction brandedBuilder(builder, brandValue) {\n    return { ...builder, ...(0, util_1.brand)(brandValue) };\n}\nfunction _custom(typeName, brand, input) {\n    const data = {\n        arguments: {},\n        returnType: null,\n        authorization: [],\n        typeName: typeName,\n        handlers: null,\n        subscriptionSource: [],\n        input,\n    };\n    const builder = brandedBuilder({\n        arguments(args) {\n            data.arguments = args;\n            return this;\n        },\n        returns(returnType) {\n            data.returnType = returnType;\n            return this;\n        },\n        authorization(callback) {\n            const rules = callback(Authorization_1.allowForCustomOperations);\n            data.authorization = Array.isArray(rules) ? rules : [rules];\n            return this;\n        },\n        handler(handlers) {\n            data.handlers = Array.isArray(handlers)\n                ? handlers\n                : [handlers];\n            return this;\n        },\n        for(source) {\n            data.subscriptionSource = Array.isArray(source) ? source : [source];\n            return this;\n        },\n    }, brand);\n    return { ...builder, data };\n}\n/**\n * Use a custom query to define an API request that will retrieve backend data.\n * @see {@link https://docs.amplify.aws/react/build-a-backend/data/custom-business-logic/}\n * @example\n * const schema = a.schema({\n *   echo: a\n *     .query()\n *     .arguments({ content: a.string() })\n *     .returns(a.ref('EchoResponse'))\n *     .authorization(allow => [allow.publicApiKey()])\n *     // 3. set the function has the handler\n *     .handler(a.handler.function(echoHandler)),\n *\n *   EchoResponse: a.customType({\n *     content: a.string(),\n *     executionDuration: a.float()\n *   }),\n * });\n * @returns a custom query\n */\nfunction query() {\n    return _custom('Query', queryBrand);\n}\nexports.query = query;\n/**\n * Use a custom mutation to define an API request that will modify backend data or trigger a subscription event.\n * @see {@link https://docs.amplify.aws/react/build-a-backend/data/custom-business-logic/}\n * @example\n * likePost: a\n *   .mutation()\n *   .arguments({ postId: a.string() })\n *   .returns(a.ref('Post'))\n *   .authorization(allow => [allow.publicApiKey()])\n *   .handler(a.handler.function(echoHandler))\n * @returns a custom mutation\n */\nfunction mutation() {\n    return _custom('Mutation', mutationBrand);\n}\nexports.mutation = mutation;\n/**\n * Define a custom subscription to receive an event when a mutation is triggered\n * @see {@link https://docs.amplify.aws/react/build-a-backend/data/custom-subscription/}\n * @example\n * // Subscribe to incoming messages\n * receive: a.subscription()\n *   // subscribes to the 'publish' mutation\n *   .for(a.ref('publish'))\n *   // subscription handler to set custom filters\n *   .handler(a.handler.custom({entry: './receive.js'}))\n *   // authorization rules as to who can subscribe to the data\n *   .authorization(allow => [allow.publicApiKey()]),\n * @returns a custom subscription\n */\nfunction subscription() {\n    return _custom('Subscription', subscriptionBrand);\n}\nexports.subscription = subscription;\n/**\n * @experimental\n *\n * Define an AI generation route for single request-response interaction with specified AI model.\n * @example\n * makeRecipe: a.generation({\n *   aiModel: { resourcePath },\n *   systemPrompt: 'Please make a recipe from the provided ingredients',\n * })\n *   .arguments({ ingredients: a.string().array() })\n *   .returns(a.ref(\"Recipe\"))\n * @returns a generation route definition\n */\nfunction generation(input) {\n    return _custom('Generation', generationBrand, input);\n}\nexports.generation = generation;\n"],"names":[],"mappings":";;AACA,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,YAAY,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;AAC9D,OAAO,CAAC,UAAU,GAAG,OAAO,CAAC,YAAY,GAAG,OAAO,CAAC,QAAQ,GAAG,OAAO,CAAC,KAAK,GAAG,OAAO,CAAC,oBAAoB,GAAG,KAAK,CAAC,CAAC;AACrH,MAAM,MAAM,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;AACjC,MAAM,eAAe,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAAC;AACnD,MAAM,UAAU,GAAG,sBAAsB,CAAC;AAC1C,MAAM,aAAa,GAAG,yBAAyB,CAAC;AAChD,MAAM,iBAAiB,GAAG,6BAA6B,CAAC;AACxD,MAAM,eAAe,GAAG,2BAA2B,CAAC;AACpD,OAAO,CAAC,oBAAoB,GAAG;AAC/B,IAAI,OAAO;AACX,IAAI,UAAU;AACd,IAAI,cAAc;AAClB,IAAI,YAAY;AAChB,CAAC,CAAC;AACF,SAAS,cAAc,CAAC,OAAO,EAAE,UAAU,EAAE;AAC7C,IAAI,OAAO,EAAE,GAAG,OAAO,EAAE,GAAG,IAAI,MAAM,CAAC,KAAK,EAAE,UAAU,CAAC,EAAE,CAAC;AAC5D,CAAC;AACD,SAAS,OAAO,CAAC,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE;AACzC,IAAI,MAAM,IAAI,GAAG;AACjB,QAAQ,SAAS,EAAE,EAAE;AACrB,QAAQ,UAAU,EAAE,IAAI;AACxB,QAAQ,aAAa,EAAE,EAAE;AACzB,QAAQ,QAAQ,EAAE,QAAQ;AAC1B,QAAQ,QAAQ,EAAE,IAAI;AACtB,QAAQ,kBAAkB,EAAE,EAAE;AAC9B,QAAQ,KAAK;AACb,KAAK,CAAC;AACN,IAAI,MAAM,OAAO,GAAG,cAAc,CAAC;AACnC,QAAQ,SAAS,CAAC,IAAI,EAAE;AACxB,YAAY,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;AAClC,YAAY,OAAO,IAAI,CAAC;AACxB,SAAS;AACT,QAAQ,OAAO,CAAC,UAAU,EAAE;AAC5B,YAAY,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;AACzC,YAAY,OAAO,IAAI,CAAC;AACxB,SAAS;AACT,QAAQ,aAAa,CAAC,QAAQ,EAAE;AAChC,YAAY,MAAM,KAAK,GAAG,QAAQ,CAAC,eAAe,CAAC,wBAAwB,CAAC,CAAC;AAC7E,YAAY,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,KAAK,GAAG,CAAC,KAAK,CAAC,CAAC;AACxE,YAAY,OAAO,IAAI,CAAC;AACxB,SAAS;AACT,QAAQ,OAAO,CAAC,QAAQ,EAAE;AAC1B,YAAY,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC;AACnD,kBAAkB,QAAQ;AAC1B,kBAAkB,CAAC,QAAQ,CAAC,CAAC;AAC7B,YAAY,OAAO,IAAI,CAAC;AACxB,SAAS;AACT,QAAQ,GAAG,CAAC,MAAM,EAAE;AACpB,YAAY,IAAI,CAAC,kBAAkB,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,MAAM,GAAG,CAAC,MAAM,CAAC,CAAC;AAChF,YAAY,OAAO,IAAI,CAAC;AACxB,SAAS;AACT,KAAK,EAAE,KAAK,CAAC,CAAC;AACd,IAAI,OAAO,EAAE,GAAG,OAAO,EAAE,IAAI,EAAE,CAAC;AAChC,CAAC;AACD;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,SAAS,KAAK,GAAG;AACjB,IAAI,OAAO,OAAO,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;AACxC,CAAC;AACD,OAAO,CAAC,KAAK,GAAG,KAAK,CAAC;AACtB;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,SAAS,QAAQ,GAAG;AACpB,IAAI,OAAO,OAAO,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC;AAC9C,CAAC;AACD,OAAO,CAAC,QAAQ,GAAG,QAAQ,CAAC;AAC5B;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,SAAS,YAAY,GAAG;AACxB,IAAI,OAAO,OAAO,CAAC,cAAc,EAAE,iBAAiB,CAAC,CAAC;AACtD,CAAC;AACD,OAAO,CAAC,YAAY,GAAG,YAAY,CAAC;AACpC;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,SAAS,UAAU,CAAC,KAAK,EAAE;AAC3B,IAAI,OAAO,OAAO,CAAC,YAAY,EAAE,eAAe,EAAE,KAAK,CAAC,CAAC;AACzD,CAAC;AACD,OAAO,CAAC,UAAU,GAAG,UAAU;;"}

package/dist/cjs/SchemaProcessor.js

@@ -11,6 +11,8 @@ const util_1 = require("./util");
 const Handler_1 = require("./Handler");
 const os = tslib_1.__importStar(require("os"));
 const path = tslib_1.__importStar(require("path"));
+const ConversationType_1 = require("./ai/ConversationType");
+const ConversationSchemaTypes_1 = require("./ai/ConversationSchemaTypes");
 function isInternalModel(model) {
     if (model.data &&
         !isCustomType(model) &&
@@ -31,6 +33,12 @@ function isCustomType(data) {
     }
     return false;
 }
+function isConversationRoute(type) {
+    return (0, util_1.getBrand)(type) === ConversationType_1.brandName;
+}
+function isGenerationInput(input) {
+    return Boolean(input?.aiModel && input?.systemPrompt);
+}
 function isCustomOperation(type) {
     if (CustomOperation_1.CustomOperationNames.includes(type?.data?.typeName)) {
         return true;
@@ -279,6 +287,19 @@ function customOperationToGql(typeName, typeDef, authorization, isCustom = false
             .join('", "');
         gqlHandlerContent += `@aws_subscribe(mutations: ["${subscriptionSources}"]) `;
     }
+    if (opType === 'Generation') {
+        if (!isGenerationInput(typeDef.data.input)) {
+            throw new Error(`Invalid Generation Route definition. A Generation Route must include a valid input. ${typeName} has an invalid or no input defined.`);
+        }
+        const { aiModel, systemPrompt, inferenceConfiguration } = typeDef.data.input;
+        const inferenceConfigurationEntries = Object.entries(inferenceConfiguration ?? {});
+        const inferenceConfigurationGql = inferenceConfigurationEntries.length > 0
+            ? `, inferenceConfiguration: { ${inferenceConfigurationEntries
+                .map(([key, value]) => `${key}: ${value}`)
+                .join(', ')} }`
+            : '';
+        gqlHandlerContent += `@generation(aiModel: "${aiModel.resourcePath}", systemPrompt: "${systemPrompt}"${inferenceConfigurationGql}) `;
+    }
     const gqlField = `${callSignature}: ${returnTypeName} ${gqlHandlerContent}${authString}`;
     return {
         gqlField,
@@ -786,6 +807,7 @@ const schemaPreprocessor = (schema) => {
     const customQueries = [];
     const customMutations = [];
     const customSubscriptions = [];
+    let shouldAddConversationTypes = false;
     // Dict of auth rules to be applied to custom types
     // Inherited from the auth configured on the custom operations that return these custom types
     const customTypeInheritedAuthRules = {};
@@ -829,6 +851,7 @@ const schemaPreprocessor = (schema) => {
                 gqlModels.push(model);
             }
             else if (isCustomOperation(typeDef)) {
+                // TODO: add generation route logic.
                 const { typeName: opType } = typeDef.data;
                 const { gqlField, implicitTypes, customTypeAuthRules, jsFunctionForField, lambdaFunctionDefinition, customSqlDataSourceStrategy, } = transformCustomOperations(typeDef, typeName, mostRelevantAuthRules, databaseType, getRefType);
                 topLevelTypes.push(...implicitTypes);
@@ -851,6 +874,7 @@ const schemaPreprocessor = (schema) => {
                 }
                 switch (opType) {
                     case 'Query':
+                    case 'Generation':
                         customQueries.push(gqlField);
                         break;
                     case 'Mutation':
@@ -861,6 +885,11 @@ const schemaPreprocessor = (schema) => {
                         break;
                 }
             }
+            else if (isConversationRoute(typeDef)) {
+                // TODO: add inferenceConfiguration values to directive.
+                customMutations.push((0, ConversationSchemaTypes_1.createConversationField)(typeDef, typeName));
+                shouldAddConversationTypes = true;
+            }
         }
         else if (staticSchema) {
             const fields = { ...typeDef.data.fields };
@@ -919,6 +948,9 @@ const schemaPreprocessor = (schema) => {
         subscriptions: customSubscriptions,
     };
     gqlModels.push(...generateCustomOperationTypes(customOperations));
+    if (shouldAddConversationTypes) {
+        gqlModels.push(...ConversationSchemaTypes_1.conversationTypes);
+    }
     const processedSchema = gqlModels.join('\n\n');
     return {
         schema: processedSchema,
@@ -932,8 +964,9 @@ function validateCustomOperations(typeDef, typeName, authRules, getRefType) {
     const { handlers, typeName: opType, subscriptionSource } = typeDef.data;
     const handlerConfigured = handlers?.length;
     const authConfigured = authRules.length > 0;
-    if ((authConfigured && !handlerConfigured) ||
-        (handlerConfigured && !authConfigured)) {
+    if (opType !== 'Generation' &&
+        ((authConfigured && !handlerConfigured) ||
+            (handlerConfigured && !authConfigured))) {
         // Deploying a custom operation with auth and no handler reference OR
         // with a handler reference but no auth
         // causes the CFN stack to reach an unrecoverable state. Ideally, this should be fixed
@@ -952,8 +985,9 @@ function validateCustomOperations(typeDef, typeName, authRules, getRefType) {
         }
     }
     if (typeDef.data.returnType === null &&
-        (opType === 'Query' || opType === 'Mutation')) {
-        throw new Error(`Invalid Custom ${opType} definition. A Custom ${opType} must include a return type. ${typeName} has no return type specified.`);
+        (opType === 'Query' || opType === 'Mutation' || opType === 'Generation')) {
+        const typeDescription = opType === 'Generation' ? 'Generation Route' : `Custom ${opType}`;
+        throw new Error(`Invalid ${typeDescription} definition. A ${typeDescription} must include a return type. ${typeName} has no return type specified.`);
     }
     if (opType !== 'Subscription' && subscriptionSource.length > 0) {
         throw new Error(`The .for() modifier function can only be used with a custom subscription. ${typeName} is not a custom subscription.`);
@@ -1067,7 +1101,9 @@ function transformCustomOperations(typeDef, typeName, authRules, databaseType, g
     let jsFunctionForField = undefined;
     validateCustomOperations(typeDef, typeName, authRules, getRefType);
     if (isCustomHandler(handlers)) {
-        jsFunctionForField = handleCustom(handlers, opType, typeName);
+        jsFunctionForField = handleCustom(handlers, 
+        // Generation routes should not have handlers
+        opType, typeName);
     }
     const isCustom = Boolean(jsFunctionForField);
     const { gqlField, implicitTypes, customTypeAuthRules, lambdaFunctionDefinition, customSqlDataSourceStrategy, } = customOperationToGql(typeName, typeDef, authRules, isCustom, databaseType, getRefType);