- feature: added support to upload multiple files (#93)

* - feature: added support to upload multiple files * - refactor: adapted shinkai app to new methods
dcSpark · Nov 28, 2023 · 4b1a2e1 · 4b1a2e1
1 parent 47d4897
commit 4b1a2e1
Show file tree

Hide file tree

Showing 9 changed files with 15 additions and 13 deletions.
diff --git a/apps/shinkai-app/src/pages/Chat.tsx b/apps/shinkai-app/src/pages/Chat.tsx
@@ -112,7 +112,7 @@ const Chat: React.FC = () => {
         receiver,
         message: data.message,
         inboxId: deserializedId as string,
-        file,
+        files: [file],
         my_device_encryption_sk: auth.my_device_encryption_sk,
         my_device_identity_sk: auth.my_device_identity_sk,
         node_encryption_pk: auth.node_encryption_pk,

diff --git a/apps/shinkai-app/src/pages/CreateJob.tsx b/apps/shinkai-app/src/pages/CreateJob.tsx
@@ -66,7 +66,7 @@ const CreateJob: React.FC = () => {
       agentId: data.model,
       content: data.description,
       files_inbox: '',
-      file: undefined,
+      files: [],
       my_device_encryption_sk: auth.my_device_encryption_sk,
       my_device_identity_sk: auth.my_device_identity_sk,
       node_encryption_pk: auth.node_encryption_pk,

diff --git a/apps/shinkai-app/src/pages/JobChat.tsx b/apps/shinkai-app/src/pages/JobChat.tsx
@@ -162,7 +162,7 @@ const JobChat: React.FC = () => {
         receiver: auth.shinkai_identity,
         message: message_to_send,
         inboxId: deserializedId as string,
-        file: selectedFile,
+        files: [selectedFile],
         my_device_encryption_sk: auth.my_device_encryption_sk,
         my_device_identity_sk: auth.my_device_identity_sk,
         node_encryption_pk: auth.node_encryption_pk,

diff --git a/apps/shinkai-visor/src/components/create-job/create-job.tsx b/apps/shinkai-visor/src/components/create-job/create-job.tsx
@@ -135,7 +135,7 @@ export const CreateJob = () => {
       agentId: values.agent,
       content: content,
       files_inbox: '',
-      file: values.files[0],
+      files: values.files,
       my_device_encryption_sk: auth.my_device_encryption_sk,
       my_device_identity_sk: auth.my_device_identity_sk,
       node_encryption_pk: auth.node_encryption_pk,

diff --git a/libs/shinkai-message-ts/src/api/methods.ts b/libs/shinkai-message-ts/src/api/methods.ts
@@ -156,7 +156,7 @@ export const sendTextMessageWithFilesForInbox = async (
   receiver: string,
   text_message: string,
   job_inbox: string,
-  selectedFile: File,
+  files: File[],
   setupDetailsState: CredentialsPayload,
 ): Promise<{ inboxId: string; message: ShinkaiMessage }> => {
   try {
@@ -172,7 +172,9 @@ export const sendTextMessageWithFilesForInbox = async (
     );
 
     await fileUploader.createFolder();
-    await fileUploader.uploadEncryptedFile(selectedFile);
+    for (const fileToUpload of files) {
+      await fileUploader.uploadEncryptedFile(fileToUpload);
+    }
     const responseText = await fileUploader.finalizeAndSend(text_message);
     const message: ShinkaiMessage = JSON.parse(responseText);
 

diff --git a/libs/shinkai-node-state/src/lib/mutations/createJob/index.ts b/libs/shinkai-node-state/src/lib/mutations/createJob/index.ts
@@ -17,7 +17,7 @@ export const createJob = async ({
   agentId,
   content,
   files_inbox,
-  file,
+  files,
   my_device_encryption_sk,
   my_device_identity_sk,
   node_encryption_pk,
@@ -48,14 +48,14 @@ export const createJob = async ({
     },
   );
 
-  const response = file
+  const response = files?.length
     ? await sendTextMessageWithFilesForInbox(
         shinkaiIdentity,
         profile, // sender subidentity
         receiver,
         content,
         buildInboxIdFromJobId(jobId),
-        file,
+        files,
         {
           my_device_encryption_sk: my_device_encryption_sk,
           my_device_identity_sk: my_device_identity_sk,

diff --git a/libs/shinkai-node-state/src/lib/mutations/createJob/types.ts b/libs/shinkai-node-state/src/lib/mutations/createJob/types.ts
@@ -7,7 +7,7 @@ export type CreateJobInput = JobCredentialsPayload & {
   agentId: string;
   content: string;
   files_inbox: string;
-  file?: File;
+  files?: File[];
 };
 
 export type CreateJobOutput = {

diff --git a/libs/shinkai-node-state/src/lib/mutations/sendMesssageWithFilesToInbox/index.ts b/libs/shinkai-node-state/src/lib/mutations/sendMesssageWithFilesToInbox/index.ts
@@ -8,7 +8,7 @@ export const sendMessageWithFilesToInbox = async ({
   receiver,
   message,
   inboxId,
-  file,
+  files,
   my_device_encryption_sk,
   my_device_identity_sk,
   node_encryption_pk,
@@ -21,7 +21,7 @@ export const sendMessageWithFilesToInbox = async ({
     receiver,
     message,
     inboxId,
-    file,
+    files,
     {
       my_device_encryption_sk,
       my_device_identity_sk,

diff --git a/libs/shinkai-node-state/src/lib/mutations/sendMesssageWithFilesToInbox/types.ts b/libs/shinkai-node-state/src/lib/mutations/sendMesssageWithFilesToInbox/types.ts
@@ -14,5 +14,5 @@ export type SendMessageWithFilesToInboxInput = CredentialsPayload & {
   senderSubidentity: string;
   message: string;
   inboxId: string;
-  file: File;
+  files: File[];
 };