add playground flag

dcSpark · Oct 4, 2024 · 6e51159 · 6e51159
1 parent 9b83aa7
commit 6e51159
Show file tree

Hide file tree

Showing 6 changed files with 33 additions and 4 deletions.
diff --git a/apps/shinkai-desktop/src/components/chat/message-stream.tsx b/apps/shinkai-desktop/src/components/chat/message-stream.tsx
@@ -138,6 +138,7 @@ export const useWebSocketMessage = ({
     if (lastMessage?.data) {
       try {
         const parseData: WsMessage = JSON.parse(lastMessage.data);
+        console.log(parseData, 'parseData');
         if (parseData.message_type !== 'Stream' || parseData.inbox !== inboxId)
           return;
         isStreamingFinished.current = false;

diff --git a/apps/shinkai-desktop/src/components/playground/baml-editor.tsx b/apps/shinkai-desktop/src/components/playground/baml-editor.tsx
@@ -241,6 +241,7 @@ function BamlEditor() {
       content: escapedBamlInput,
       workflowCode,
       isHidden: true,
+      playground: true,
       chatConfig: {
         stream: false,
         custom_prompt: '',

diff --git a/apps/shinkai-desktop/src/components/playground/workflow-editor.tsx b/apps/shinkai-desktop/src/components/playground/workflow-editor.tsx
@@ -232,6 +232,7 @@ function WorkflowEditor() {
       isHidden: true,
       selectedVRFiles,
       selectedVRFolders,
+      playground: true,
     });
   };
   const { mutateAsync: createWorkflow, isPending: isCreateWorkflowPending } =

diff --git a/libs/shinkai-message-ts/src/api/jobs/types.ts b/libs/shinkai-message-ts/src/api/jobs/types.ts
@@ -79,9 +79,14 @@ export type CreateJobRequest = {
       local_vrpack: [];
       local_vrkai: [];
     };
-    associated_ui: {
-      Sheet: string;
-    } | null;
+    associated_ui:
+      | null
+      | {
+          Sheet: string | null;
+        }
+      | {
+          Playground: '';
+        };
     is_hidden: boolean;
   };
 };

diff --git a/libs/shinkai-node-state/src/v2/mutations/createJob/index.ts b/libs/shinkai-node-state/src/v2/mutations/createJob/index.ts
@@ -20,7 +20,27 @@ export const createJob = async ({
   selectedVRFiles,
   selectedVRFolders,
   chatConfig,
+  playground,
 }: CreateJobInput) => {
+  let associatedUI:
+    | null
+    | {
+        Sheet: string | null;
+      }
+    | {
+        Playground: '';
+      } = null;
+
+  if (playground) {
+    associatedUI = {
+      Playground: '',
+    };
+  } else if (sheetId) {
+    associatedUI = {
+      Sheet: sheetId,
+    };
+  }
+
   const { job_id: jobId } = await createJobApi(nodeAddress, token, {
     llm_provider: llmProvider,
     job_creation_info: {
@@ -31,7 +51,7 @@ export const createJob = async ({
         local_vrkai: [],
         network_folders: [],
       },
-      associated_ui: sheetId ? { Sheet: sheetId } : null,
+      associated_ui: associatedUI,
       is_hidden: isHidden,
     },
   });

diff --git a/libs/shinkai-node-state/src/v2/mutations/createJob/types.ts b/libs/shinkai-node-state/src/v2/mutations/createJob/types.ts
@@ -17,6 +17,7 @@ export type CreateJobInput = Token & {
   selectedVRFiles?: VectorFSItemScopeEntry[];
   selectedVRFolders?: VectorFSFolderScopeEntry[];
   chatConfig?: JobConfig;
+  playground?: boolean;
 };
 
 export type CreateJobOutput = {