chore: adjust impl.

Author: sobolk

packages/amplify-graphql-conversation-transformer/src/graphql-types/message-model.ts

@@ -146,12 +146,13 @@ export const createAssistantResponseStreamingMutationInput = (messageModelName:
 };
 
 export const createAttachmentUploadUrlQueryInput = (messageModelName: string): InputObjectTypeDefinitionNode => {
-  const inputName = `Get${messageModelName}AttachmentUploadUrlInput`;
+  const inputName = `Get${messageModelName}AttachmentUrlsInput`;
   return {
     kind: 'InputObjectTypeDefinition',
     name: { kind: 'Name', value: inputName },
     fields: [
       makeInputValueDefinition('conversationId', makeNonNullType(makeNamedType('ID'))),
+      makeInputValueDefinition('attachmentKey', makeNonNullType(makeNamedType('String'))),
     ],
   };
 };
@@ -312,7 +313,7 @@ const constructConversationMessageModel = (
 };
 
 const STREAM_RESPONSE_TYPE_NAME = 'AmplifyAIConversationMessageStreamPart';
-const ATTACHMENT_URL_RESPONSE_TYPE_NAME = 'AmplifyAIAttachmentUrl';
+const ATTACHMENT_URL_RESPONSE_TYPE_NAME = 'AmplifyAIAttachmentUrls';
 
 export const constructStreamResponseType = (): ObjectTypeDefinitionNode => {
   return {

packages/amplify-graphql-conversation-transformer/src/resolvers/get-attachment-url-pipeline-definition.ts

@@ -83,5 +83,5 @@ function fieldName(config: ConversationDirectiveConfiguration): string {
  * Creates a template generator specific to the send message pipeline for a given slot name.
  */
 function templateGenerator(slotName: string) {
-  return createS3AssetMappingTemplateGenerator('Mutation', slotName, fieldName);
+  return createS3AssetMappingTemplateGenerator('Query', slotName, fieldName);
 }

packages/amplify-graphql-conversation-transformer/src/resolvers/templates/invoke-attachment-lambda-resolver-fn.template.js

@@ -5,6 +5,7 @@ export function request(ctx) {
 
   const payload = {
     conversationId: args.input.conversationId,
+    attachmentKey: args.input.attachmentKey,
   };
 
   return {

packages/amplify-graphql-conversation-transformer/src/transformer-steps/attachment-lambda/handler.ts

@@ -1,13 +1,15 @@
-import { S3Client, PutObjectCommand } from '@aws-sdk/client-s3';
+import { S3Client, GetObjectCommand, PutObjectCommand } from '@aws-sdk/client-s3';
 import { getSignedUrl } from '@aws-sdk/s3-request-presigner';
 
 /**
  * TODO
  */
 export const handler = async (event: {
   conversationId: string;
+  attachmentKey: string;
 }): Promise<{
-  url: string;
+  uploadUrl: string;
+  downloadUrl: string;
 }> => {
   console.log(JSON.stringify(event, null, 2));
   console.log(process.env);
@@ -19,11 +21,14 @@ export const handler = async (event: {
   const bucketName = process.env.S3_BUCKET_NAME;
 
   const client = new S3Client();
-  const attachmentKey = crypto.randomUUID();
-  const command = new PutObjectCommand({ Bucket: bucketName, Key: `${event.conversationId}/${attachmentKey}` });
-  const url = await getSignedUrl(client, command, { expiresIn: 3600 });
+  const putCommand = new PutObjectCommand({ Bucket: bucketName, Key: `${event.conversationId}/${event.attachmentKey}` });
+  const uploadUrl = await getSignedUrl(client, putCommand, { expiresIn: 3600 });
+
+  const getCommand = new GetObjectCommand({ Bucket: bucketName, Key: `${event.conversationId}/${event.attachmentKey}` });
+  const downloadUrl = await getSignedUrl(client, getCommand, { expiresIn: 3600 });
 
   return {
-    url,
+    uploadUrl,
+    downloadUrl,
   };
 };