@aws-amplify/data-schema 1.9.0 → 1.9.2

package/dist/cjs/SchemaProcessor.js

@@ -300,13 +300,23 @@ function customOperationToGql(typeName, typeDef, authorization, isCustom = false
             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;
+        // This is done to escape newlines in potentially multi-line system prompts
+        // e.g.
+        // generateStuff: a.generation({
+        //   aiModel: a.ai.model('Claude 3 Haiku'),
+        //   systemPrompt: `Generate a haiku
+        //   make it multiline`,
+        // }),
+        //
+        // It doesn't affect non multi-line string inputs for system prompts
+        const escapedSystemPrompt = systemPrompt.replace(/\r?\n/g, '\\n');
         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}) `;
+        gqlHandlerContent += `@generation(aiModel: "${aiModel.resourcePath}", systemPrompt: "${escapedSystemPrompt}"${inferenceConfigurationGql}) `;
     }
     const gqlField = `${callSignature}: ${returnTypeName} ${gqlHandlerContent}${authString}`;
     return {
@@ -907,7 +917,9 @@ const schemaPreprocessor = (schema) => {
             }
             else if (isConversationRoute(typeDef)) {
                 // TODO: add inferenceConfiguration values to directive.
-                customMutations.push((0, ConversationSchemaTypes_1.createConversationField)(typeDef, typeName));
+                const { field, functionHandler } = (0, ConversationSchemaTypes_1.createConversationField)(typeDef, typeName);
+                customMutations.push(field);
+                Object.assign(lambdaFunctions, functionHandler);
                 shouldAddConversationTypes = true;
             }
         }