@aws-amplify/graphql-api-construct 1.15.1-ai-streaming.1 → 1.16.0-gen2-migration-1015.0

package/node_modules/@aws-amplify/graphql-conversation-transformer/lib/resolvers/assistant-streaming-mutation-resolver-fn.template.js

@@ -1,121 +0,0 @@
-import { util } from '@aws-appsync/utils';
-import * as ddb from '@aws-appsync/utils/dynamodb';
-
-/**
- * Sends a request to the attached data source
- * @param {import('@aws-appsync/utils').Context} ctx the context
- * @returns {*} the request
- */
-export function request(ctx) {
-  const {
-    // required
-    conversationId,
-    associatedUserMessageId,
-    contentBlockIndex,
-    // text chunk
-    contentBlockText,
-    contentBlockDeltaIndex,
-    // tool use
-    contentBlockToolUse,
-    // block complete
-    contentBlockDoneAtIndex,
-    // turn complete
-    stopReason,
-  } = ctx.args.input;
-
-  const { owner } = ctx.args;
-  const streamId = `${associatedUserMessageId}#stream`;
-
-  if (stopReason) {
-    // should we be writing the turn complete chunk here?
-    // probably not. It's just the marker for us to reconcile the chunks.
-    return ddb.get({ key: { id: streamId } });
-  }
-
-  // not using this yet. we'll need to enable ttl on the table first.
-  const now = util.time.nowEpochSeconds()
-  const ttl = now + 60 * 60 * 24; // 1 day
-
-  const event = {
-    contentBlockIndex,
-    contentBlockDeltaIndex,
-    contentBlockText,
-    contentBlockToolUse,
-    contentBlockDoneAtIndex,
-  };
-
-  // AppSync JS runtime doesn't support Object.fromEntries
-  const chunk = Object.keys(event)
-  .filter((k) => event[k] != null)
-  .reduce((a, k) => ({ ...a, [k]: event[k] }), {});
-
-  // TODO: Use expression names for all attributes.
-  // reference: https://docs.aws.amazon.com/amazondynamodb/latest/developerguide/ReservedWords.html
-  const setExpression = 'SET events = list_append(if_not_exists(events, :empty_list), :events), conversationId = :conversationId, #owner = :owner'
-  const id = streamId
-  return {
-    operation: 'UpdateItem',
-    key: util.dynamodb.toMapValues({ id }),
-    update: {
-      expression: setExpression,
-      expressionValues: util.dynamodb.toMapValues({
-        ':events': [chunk],
-        ':empty_list': [],
-        ':conversationId': conversationId,
-        ':owner': owner,
-      }),
-      expressionNames: {
-        '#owner': 'owner',
-      },
-    },
-  };
-}
-
-/**
- * Returns the resolver result
- * @param {import('@aws-appsync/utils').Context} ctx the context
- * @returns {*} the result
- */
-export function response(ctx) {
-  console.log('>>> response function ctx <<<', ctx);
-  if (ctx.error) {
-    util.error(ctx.error.message, ctx.error.type);
-  }
-
-  if (ctx.args.input.contentBlockToolUse || ctx.args.input.contentBlockText) {
-    console.log('>>> contentBlockToolUse <<<', ctx.args.input.contentBlockToolUse);
-    console.log('>>> contentBlockText <<<', ctx.args.input.contentBlockText);
-
-    const { conversationId } = ctx.result;
-    const { owner } = ctx.args;
-    const { contentBlockToolUse: toolUse, contentBlockText: text } = ctx.args.input;
-    const { createdAt, updatedAt } = ctx.stash.defaultValues;
-
-    const content = ctx.args.input.contentBlockToolUse ? [{ toolUse }] : [{ text }];
-
-    return {
-      __typename: 'ConversationMessageCustomChat',
-      id: `${ctx.args.input.associatedUserMessageId}#response`,
-      conversationId: conversationId,
-      owner: owner,
-      role: 'assistant',
-      createdAt,
-      updatedAt,
-      content,
-    }
-  }
-
-  if (ctx.args.input.contentBlockDoneAtIndex) {
-    console.log('>>> contentBlockDoneAtIndex <<<', ctx.args.input.contentBlockDoneAtIndex);
-    // Do we actually need this event? It's forcing us to return a value to the client here, which is awkward... maybe.
-    return {
-      id: `${ctx.args.input.associatedUserMessageId}#response`,
-      createdAt: ctx.stash.defaultValues.createdAt,
-      updatedAt: ctx.stash.defaultValues.updatedAt,
-      conversationId: ctx.args.input.conversationId,
-    }
-  }
-
-  console.log('>>> stopReason <<<', ctx.args.input.stopReason);
-  return ctx.result;
-}

package/node_modules/@aws-amplify/graphql-conversation-transformer/lib/resolvers/assistant-streaming-mutation-resolver.d.ts

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

package/node_modules/@aws-amplify/graphql-conversation-transformer/lib/resolvers/assistant-streaming-mutation-resolver.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"assistant-streaming-mutation-resolver.d.ts","sourceRoot":"","sources":["../../src/resolvers/assistant-streaming-mutation-resolver.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,uBAAuB,EAAE,MAAM,6CAA6C,CAAC;AAGtF,OAAO,EAAE,kCAAkC,EAAE,MAAM,qCAAqC,CAAC;AASzF,eAAO,MAAM,kCAAkC,WAAY,kCAAkC,KAAG,uBAY/F,CAAC"}

package/node_modules/@aws-amplify/graphql-conversation-transformer/lib/resolvers/assistant-streaming-mutation-resolver.js

@@ -1,25 +0,0 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.assistantStreamingMutationResolver = 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 graphql_transformer_common_1 = require("graphql-transformer-common");
-const assistantStreamingMutationResolver = (config) => {
-    let resolver = fs_1.default.readFileSync(path_1.default.join(__dirname, 'assistant-streaming-mutation-resolver-fn.template.js'), 'utf8');
-    const fieldName = (0, graphql_transformer_common_1.toUpper)(config.field.name.value);
-    const substitutions = {
-        CONVERSATION_MESSAGE_TYPE_NAME: `ConversationMessage${fieldName}`,
-    };
-    Object.entries(substitutions).forEach(([key, value]) => {
-        const replaced = resolver.replace(new RegExp(`\\[\\[${key}\\]\\]`, 'g'), value);
-        resolver = replaced;
-    });
-    const templateName = `Mutation.${config.field.name.value}.assistant-response-stream.js`;
-    return graphql_transformer_core_1.MappingTemplate.s3MappingFunctionCodeFromString(resolver, templateName);
-};
-exports.assistantStreamingMutationResolver = assistantStreamingMutationResolver;
-//# sourceMappingURL=assistant-streaming-mutation-resolver.js.map

package/node_modules/@aws-amplify/graphql-conversation-transformer/lib/resolvers/assistant-streaming-mutation-resolver.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"assistant-streaming-mutation-resolver.js","sourceRoot":"","sources":["../../src/resolvers/assistant-streaming-mutation-resolver.ts"],"names":[],"mappings":";;;;;;AAAA,oFAAwE;AAExE,4CAAoB;AACpB,gDAAwB;AAExB,2EAAqD;AAQ9C,MAAM,kCAAkC,GAAG,CAAC,MAA0C,EAA2B,EAAE;IACxH,IAAI,QAAQ,GAAG,YAAE,CAAC,YAAY,CAAC,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,sDAAsD,CAAC,EAAE,MAAM,CAAC,CAAC;IACrH,MAAM,SAAS,GAAG,IAAA,oCAAO,EAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACnD,MAAM,aAAa,GAAG;QACpB,8BAA8B,EAAE,sBAAsB,SAAS,EAAE;KAClE,CAAC;IACF,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;IACH,MAAM,YAAY,GAAG,YAAY,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,+BAA+B,CAAC;IACxF,OAAO,0CAAe,CAAC,+BAA+B,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;AACjF,CAAC,CAAC;AAZW,QAAA,kCAAkC,sCAY7C"}