@aws-amplify/data-schema 1.13.2 → 1.13.3

package/dist/cjs/ai/ConversationSchemaGraphQLTypes.js

@@ -0,0 +1,190 @@
+'use strict';
+
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CONVERSATION_SCHEMA_GRAPHQL_TYPES = void 0;
+exports.CONVERSATION_SCHEMA_GRAPHQL_TYPES = 
+/* GraphQL */ `enum ConversationParticipantRole {
+  user
+  assistant
+}
+
+interface ConversationMessage {
+  id: ID!
+  conversationId: ID!
+  associatedUserMessageId: ID
+  role: ConversationParticipantRole
+  content: [ContentBlock]
+  aiContext: AWSJSON
+  toolConfiguration: ToolConfiguration
+  createdAt: AWSDateTime
+  updatedAt: AWSDateTime
+  owner: String
+}
+
+input DocumentBlockSourceInput {
+  bytes: String
+}
+
+input DocumentBlockInput {
+  format: String!
+  name: String!
+  source: DocumentBlockSourceInput!
+}
+
+input ImageBlockSourceInput {
+  bytes: String
+}
+
+input ImageBlockInput {
+  format: String!
+  source: ImageBlockSourceInput!
+}
+
+input ToolUseBlockInput {
+  toolUseId: String!
+  name: String!
+  input: AWSJSON!
+}
+
+input ToolResultContentBlockInput {
+  document: DocumentBlockInput
+  image: ImageBlockInput
+  json: AWSJSON
+  text: String
+}
+
+input ToolResultBlockInput {
+  content: [ToolResultContentBlockInput!]!
+  toolUseId: String!
+  status: String
+}
+
+type DocumentBlockSource {
+  bytes: String
+}
+
+type DocumentBlock {
+  format: String!
+  name: String!
+  source: DocumentBlockSource!
+}
+
+type ImageBlock {
+  format: String!
+  source: ImageBlockSource!
+}
+
+type ImageBlockSource {
+  bytes: String
+}
+
+type ToolUseBlock {
+  toolUseId: String!
+  name: String!
+  input: AWSJSON!
+}
+
+type ToolResultContentBlock {
+  document: DocumentBlock
+  image: ImageBlock
+  json: AWSJSON
+  text: String
+}
+
+type ToolResultBlock {
+  content: [ToolResultContentBlock!]!
+  toolUseId: String!
+  status: String
+}
+
+type ContentBlockText {
+  text: String
+}
+
+type ContentBlockImage {
+  image: ImageBlock
+}
+
+type ContentBlockDocument {
+  document: DocumentBlock
+}
+
+type ContentBlockToolUse {
+  toolUse: ToolUseBlock
+}
+
+type ContentBlockToolResult {
+  toolResult: ToolResultBlock
+}
+
+input ContentBlockInput {
+  text: String
+  document: DocumentBlockInput
+  image: ImageBlockInput
+  toolResult: ToolResultBlockInput
+  toolUse: ToolUseBlockInput
+}
+
+type ContentBlock {
+  text: String
+  document: DocumentBlock
+  image: ImageBlock
+  toolResult: ToolResultBlock
+  toolUse: ToolUseBlock
+}
+
+input ToolConfigurationInput {
+  tools: [ToolInput]
+}
+
+input ToolInput {
+  toolSpec: ToolSpecificationInput
+}
+
+input ToolSpecificationInput {
+  name: String!
+  description: String
+  inputSchema: ToolInputSchemaInput!
+}
+
+input ToolInputSchemaInput {
+  json: AWSJSON
+}
+
+type ToolConfiguration {
+  tools: [Tool]
+}
+
+type Tool {
+  toolSpec: ToolSpecification
+}
+
+type ToolSpecification {
+  name: String!
+  description: String
+  inputSchema: ToolInputSchema!
+}
+
+type ToolInputSchema {
+  json: AWSJSON
+}
+
+type ConversationMessageStreamPart @aws_cognito_user_pools {
+  id: ID!
+  owner: String
+  conversationId: ID!
+  associatedUserMessageId: ID!
+  contentBlockIndex: Int
+  contentBlockText: String
+  contentBlockDeltaIndex: Int
+  contentBlockToolUse: ToolUseBlock
+  contentBlockDoneAtIndex: Int
+  stopReason: String
+  errors: [ConversationTurnError]
+}
+
+type ConversationTurnError @aws_cognito_user_pools {
+  message: String!
+  errorType: String!
+}`;
+//# sourceMappingURL=ConversationSchemaGraphQLTypes.js.map

package/dist/cjs/ai/ConversationSchemaGraphQLTypes.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ConversationSchemaGraphQLTypes.js","sources":["../../../src/ai/ConversationSchemaGraphQLTypes.ts"],"sourcesContent":["\"use strict\";\nObject.defineProperty(exports, \"__esModule\", { value: true });\nexports.CONVERSATION_SCHEMA_GRAPHQL_TYPES = void 0;\nexports.CONVERSATION_SCHEMA_GRAPHQL_TYPES = \n/* GraphQL */ `enum ConversationParticipantRole {\n  user\n  assistant\n}\n\ninterface ConversationMessage {\n  id: ID!\n  conversationId: ID!\n  associatedUserMessageId: ID\n  role: ConversationParticipantRole\n  content: [ContentBlock]\n  aiContext: AWSJSON\n  toolConfiguration: ToolConfiguration\n  createdAt: AWSDateTime\n  updatedAt: AWSDateTime\n  owner: String\n}\n\ninput DocumentBlockSourceInput {\n  bytes: String\n}\n\ninput DocumentBlockInput {\n  format: String!\n  name: String!\n  source: DocumentBlockSourceInput!\n}\n\ninput ImageBlockSourceInput {\n  bytes: String\n}\n\ninput ImageBlockInput {\n  format: String!\n  source: ImageBlockSourceInput!\n}\n\ninput ToolUseBlockInput {\n  toolUseId: String!\n  name: String!\n  input: AWSJSON!\n}\n\ninput ToolResultContentBlockInput {\n  document: DocumentBlockInput\n  image: ImageBlockInput\n  json: AWSJSON\n  text: String\n}\n\ninput ToolResultBlockInput {\n  content: [ToolResultContentBlockInput!]!\n  toolUseId: String!\n  status: String\n}\n\ntype DocumentBlockSource {\n  bytes: String\n}\n\ntype DocumentBlock {\n  format: String!\n  name: String!\n  source: DocumentBlockSource!\n}\n\ntype ImageBlock {\n  format: String!\n  source: ImageBlockSource!\n}\n\ntype ImageBlockSource {\n  bytes: String\n}\n\ntype ToolUseBlock {\n  toolUseId: String!\n  name: String!\n  input: AWSJSON!\n}\n\ntype ToolResultContentBlock {\n  document: DocumentBlock\n  image: ImageBlock\n  json: AWSJSON\n  text: String\n}\n\ntype ToolResultBlock {\n  content: [ToolResultContentBlock!]!\n  toolUseId: String!\n  status: String\n}\n\ntype ContentBlockText {\n  text: String\n}\n\ntype ContentBlockImage {\n  image: ImageBlock\n}\n\ntype ContentBlockDocument {\n  document: DocumentBlock\n}\n\ntype ContentBlockToolUse {\n  toolUse: ToolUseBlock\n}\n\ntype ContentBlockToolResult {\n  toolResult: ToolResultBlock\n}\n\ninput ContentBlockInput {\n  text: String\n  document: DocumentBlockInput\n  image: ImageBlockInput\n  toolResult: ToolResultBlockInput\n  toolUse: ToolUseBlockInput\n}\n\ntype ContentBlock {\n  text: String\n  document: DocumentBlock\n  image: ImageBlock\n  toolResult: ToolResultBlock\n  toolUse: ToolUseBlock\n}\n\ninput ToolConfigurationInput {\n  tools: [ToolInput]\n}\n\ninput ToolInput {\n  toolSpec: ToolSpecificationInput\n}\n\ninput ToolSpecificationInput {\n  name: String!\n  description: String\n  inputSchema: ToolInputSchemaInput!\n}\n\ninput ToolInputSchemaInput {\n  json: AWSJSON\n}\n\ntype ToolConfiguration {\n  tools: [Tool]\n}\n\ntype Tool {\n  toolSpec: ToolSpecification\n}\n\ntype ToolSpecification {\n  name: String!\n  description: String\n  inputSchema: ToolInputSchema!\n}\n\ntype ToolInputSchema {\n  json: AWSJSON\n}\n\ntype ConversationMessageStreamPart @aws_cognito_user_pools {\n  id: ID!\n  owner: String\n  conversationId: ID!\n  associatedUserMessageId: ID!\n  contentBlockIndex: Int\n  contentBlockText: String\n  contentBlockDeltaIndex: Int\n  contentBlockToolUse: ToolUseBlock\n  contentBlockDoneAtIndex: Int\n  stopReason: String\n  errors: [ConversationTurnError]\n}\n\ntype ConversationTurnError @aws_cognito_user_pools {\n  message: String!\n  errorType: String!\n}`;\n"],"names":[],"mappings":";;AACA,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,YAAY,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;AAC9D,OAAO,CAAC,iCAAiC,GAAG,KAAK,CAAC,CAAC;AACnD,OAAO,CAAC,iCAAiC;AACzC,cAAc,CAAC;AACf;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,CAAC,CAAC;;"}

package/dist/cjs/ai/ConversationSchemaProcessor.js

@@ -0,0 +1,52 @@
+'use strict';
+
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createConversationField = void 0;
+const utils_1 = require("../runtime/utils");
+const createConversationField = (typeDef, typeName) => {
+    const { aiModel, systemPrompt, handler, tools } = typeDef;
+    const args = {
+        aiModel: aiModel.resourcePath,
+        // This is done to escape newlines in potentially multi-line system prompts
+        // e.g.
+        // realtorChat: a.conversation({
+        //   aiModel: a.ai.model('Claude 3 Haiku'),
+        //   systemPrompt: `You are a helpful real estate assistant
+        //   Respond in the poetic form of haiku.`,
+        // }),
+        //
+        // It doesn't affect non multi-line string inputs for system prompts
+        systemPrompt: systemPrompt.replace(/\r?\n/g, '\\n'),
+    };
+    const argsString = Object.entries(args)
+        .map(([key, value]) => `${key}: "${value}"`)
+        .join(', ');
+    const functionHandler = {};
+    let handlerString = '';
+    if (handler) {
+        const functionName = `Fn${(0, utils_1.capitalize)(typeName)}`;
+        const eventVersion = handler.eventVersion;
+        handlerString = `, handler: { functionName: "${functionName}", eventVersion: "${eventVersion}" }`;
+        functionHandler[functionName] = handler;
+    }
+    const toolsString = tools?.length
+        ? `, tools: [${getConversationToolsString(tools)}]`
+        : '';
+    const conversationDirective = `@conversation(${argsString}${handlerString}${toolsString})`;
+    const field = `${typeName}(conversationId: ID!, content: [ContentBlockInput], aiContext: AWSJSON, toolConfiguration: ToolConfigurationInput): ConversationMessage ${conversationDirective} @aws_cognito_user_pools`;
+    return { field, functionHandler };
+};
+exports.createConversationField = createConversationField;
+const isRef = (query) => query?.data?.type === 'ref';
+const getConversationToolsString = (tools) => tools
+    .map((tool) => {
+    const { query, description } = tool;
+    if (!isRef(query)) {
+        throw new Error(`Unexpected query was found in tool ${tool}.`);
+    }
+    // TODO: add validation for query / auth (cup) / etc
+    const queryName = query.data.link;
+    return `{ name: "${queryName}", description: "${description}" }`;
+})
+    .join(', ');
+//# sourceMappingURL=ConversationSchemaProcessor.js.map

package/dist/cjs/ai/ConversationSchemaProcessor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ConversationSchemaProcessor.js","sources":["../../../src/ai/ConversationSchemaProcessor.ts"],"sourcesContent":["\"use strict\";\nObject.defineProperty(exports, \"__esModule\", { value: true });\nexports.createConversationField = void 0;\nconst utils_1 = require(\"../runtime/utils\");\nconst createConversationField = (typeDef, typeName) => {\n    const { aiModel, systemPrompt, handler, tools } = typeDef;\n    const args = {\n        aiModel: aiModel.resourcePath,\n        // This is done to escape newlines in potentially multi-line system prompts\n        // e.g.\n        // realtorChat: a.conversation({\n        //   aiModel: a.ai.model('Claude 3 Haiku'),\n        //   systemPrompt: `You are a helpful real estate assistant\n        //   Respond in the poetic form of haiku.`,\n        // }),\n        //\n        // It doesn't affect non multi-line string inputs for system prompts\n        systemPrompt: systemPrompt.replace(/\\r?\\n/g, '\\\\n'),\n    };\n    const argsString = Object.entries(args)\n        .map(([key, value]) => `${key}: \"${value}\"`)\n        .join(', ');\n    const functionHandler = {};\n    let handlerString = '';\n    if (handler) {\n        const functionName = `Fn${(0, utils_1.capitalize)(typeName)}`;\n        const eventVersion = handler.eventVersion;\n        handlerString = `, handler: { functionName: \"${functionName}\", eventVersion: \"${eventVersion}\" }`;\n        functionHandler[functionName] = handler;\n    }\n    const toolsString = tools?.length\n        ? `, tools: [${getConversationToolsString(tools)}]`\n        : '';\n    const conversationDirective = `@conversation(${argsString}${handlerString}${toolsString})`;\n    const field = `${typeName}(conversationId: ID!, content: [ContentBlockInput], aiContext: AWSJSON, toolConfiguration: ToolConfigurationInput): ConversationMessage ${conversationDirective} @aws_cognito_user_pools`;\n    return { field, functionHandler };\n};\nexports.createConversationField = createConversationField;\nconst isRef = (query) => query?.data?.type === 'ref';\nconst getConversationToolsString = (tools) => tools\n    .map((tool) => {\n    const { query, description } = tool;\n    if (!isRef(query)) {\n        throw new Error(`Unexpected query was found in tool ${tool}.`);\n    }\n    // TODO: add validation for query / auth (cup) / etc\n    const queryName = query.data.link;\n    return `{ name: \"${queryName}\", description: \"${description}\" }`;\n})\n    .join(', ');\n"],"names":[],"mappings":";;AACA,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,YAAY,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;AAC9D,OAAO,CAAC,uBAAuB,GAAG,KAAK,CAAC,CAAC;AACzC,MAAM,OAAO,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAAC;AAC5C,MAAM,uBAAuB,GAAG,CAAC,OAAO,EAAE,QAAQ,KAAK;AACvD,IAAI,MAAM,EAAE,OAAO,EAAE,YAAY,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC;AAC9D,IAAI,MAAM,IAAI,GAAG;AACjB,QAAQ,OAAO,EAAE,OAAO,CAAC,YAAY;AACrC;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,QAAQ,YAAY,EAAE,YAAY,CAAC,OAAO,CAAC,QAAQ,EAAE,KAAK,CAAC;AAC3D,KAAK,CAAC;AACN,IAAI,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC;AAC3C,SAAS,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,KAAK,CAAC,EAAE,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC;AACpD,SAAS,IAAI,CAAC,IAAI,CAAC,CAAC;AACpB,IAAI,MAAM,eAAe,GAAG,EAAE,CAAC;AAC/B,IAAI,IAAI,aAAa,GAAG,EAAE,CAAC;AAC3B,IAAI,IAAI,OAAO,EAAE;AACjB,QAAQ,MAAM,YAAY,GAAG,CAAC,EAAE,EAAE,IAAI,OAAO,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC;AACtE,QAAQ,MAAM,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;AAClD,QAAQ,aAAa,GAAG,CAAC,4BAA4B,EAAE,YAAY,CAAC,kBAAkB,EAAE,YAAY,CAAC,GAAG,CAAC,CAAC;AAC1G,QAAQ,eAAe,CAAC,YAAY,CAAC,GAAG,OAAO,CAAC;AAChD,KAAK;AACL,IAAI,MAAM,WAAW,GAAG,KAAK,EAAE,MAAM;AACrC,UAAU,CAAC,UAAU,EAAE,0BAA0B,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;AAC3D,UAAU,EAAE,CAAC;AACb,IAAI,MAAM,qBAAqB,GAAG,CAAC,cAAc,EAAE,UAAU,CAAC,EAAE,aAAa,CAAC,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC;AAC/F,IAAI,MAAM,KAAK,GAAG,CAAC,EAAE,QAAQ,CAAC,wIAAwI,EAAE,qBAAqB,CAAC,wBAAwB,CAAC,CAAC;AACxN,IAAI,OAAO,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC;AACtC,CAAC,CAAC;AACF,OAAO,CAAC,uBAAuB,GAAG,uBAAuB,CAAC;AAC1D,MAAM,KAAK,GAAG,CAAC,KAAK,KAAK,KAAK,EAAE,IAAI,EAAE,IAAI,KAAK,KAAK,CAAC;AACrD,MAAM,0BAA0B,GAAG,CAAC,KAAK,KAAK,KAAK;AACnD,KAAK,GAAG,CAAC,CAAC,IAAI,KAAK;AACnB,IAAI,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,GAAG,IAAI,CAAC;AACxC,IAAI,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE;AACvB,QAAQ,MAAM,IAAI,KAAK,CAAC,CAAC,mCAAmC,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;AACvE,KAAK;AACL;AACA,IAAI,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC;AACtC,IAAI,OAAO,CAAC,SAAS,EAAE,SAAS,CAAC,iBAAiB,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC;AACrE,CAAC,CAAC;AACF,KAAK,IAAI,CAAC,IAAI,CAAC;;"}

package/dist/esm/SchemaProcessor.mjs

@@ -7,7 +7,8 @@ import { getHandlerData } from './Handler.mjs';
 import * as os from 'os';
 import * as path from 'path';
 import { brandName } from './ai/ConversationType.mjs';
-import { createConversationField, conversationTypes } from './ai/ConversationSchemaTypes.mjs';
+import { CONVERSATION_SCHEMA_GRAPHQL_TYPES } from './ai/ConversationSchemaGraphQLTypes.mjs';
+import { createConversationField } from './ai/ConversationSchemaProcessor.mjs';
 
 function isInternalModel(model) {
     if (model.data &&
@@ -1014,7 +1015,7 @@ const schemaPreprocessor = (schema) => {
     };
     gqlModels.push(...generateCustomOperationTypes(customOperations));
     if (shouldAddConversationTypes) {
-        gqlModels.push(...conversationTypes);
+        gqlModels.push(CONVERSATION_SCHEMA_GRAPHQL_TYPES);
     }
     const processedSchema = gqlModels.join('\n\n');
     return {