From 2478d2d099eafa7f4852112f394da875324192eb Mon Sep 17 00:00:00 2001
From: Sensational Code <sensational.code@gmail.com>
Date: Mon, 16 Dec 2024 10:48:45 -0800
Subject: [PATCH] Save agent prompt to workflow

---
 .../ManagementPortal/pages/agents/create.vue  | 46 +++++++++++--------
 1 file changed, 27 insertions(+), 19 deletions(-)

diff --git a/src/ui/ManagementPortal/pages/agents/create.vue b/src/ui/ManagementPortal/pages/agents/create.vue
index 72a4b9cca..4ffef8bc3 100644
--- a/src/ui/ManagementPortal/pages/agents/create.vue
+++ b/src/ui/ManagementPortal/pages/agents/create.vue
@@ -1145,10 +1145,12 @@ export default {
 				this.workflowMainAIModel = null;
 			}
 
-			if (!this.selectedWorkflow?.type) {
-				this.showWorkflowConfiguration = false;
-				this.selectedWorkflow = null;
-			}
+			this.showWorkflowConfiguration = true;
+
+			// if (!this.selectedWorkflow?.type) {
+			// 	this.showWorkflowConfiguration = false;
+			// 	this.selectedWorkflow = null;
+			// }
 		},
 
 		workflowMainAIModel() {
@@ -1196,10 +1198,10 @@ export default {
 
 			this.loadingStatusText = 'Retrieving workflows...';
 			this.workflowOptions = [
-				{
-					type: null,
-					workflow_name: 'None',
-				},
+				// {
+				// 	type: null,
+				// 	workflow_name: 'None',
+				// },
 				{
 					type: 'langgraph-react-agent-workflow',
 					workflow_name: 'LangGraph ReAct Agent Workflow',
@@ -1486,10 +1488,6 @@ export default {
 				this.text_embedding_profile_object_id = this.selectedTextEmbeddingProfile?.object_id ?? '';
 			}
 
-			if (this.systemPrompt === '') {
-				errors.push('Please provide a system prompt.');
-			}
-
 			if (!this.orchestration_settings.orchestrator) {
 				errors.push('Please select an orchestrator.');
 			}
@@ -1498,10 +1496,18 @@ export default {
 				errors.push('Please select an AI model for the orchestrator.');
 			}
 
-			if (this.selectedWorkflow && !this.workflowMainAIModel) {
+			if (!this.selectedWorkflow) {
+				errors.push('Please select a workflow.');
+			}
+
+			if (!this.workflowMainAIModel) {
 				errors.push('Please select an AI model for the workflow.');
 			}
 
+			if (this.systemPrompt === '') {
+				errors.push('Please provide a system prompt.');
+			}
+
 			// if (!this.selectedDataSource) {
 			// 	errors.push('Please select a data source.');
 			// }
@@ -1616,12 +1622,14 @@ export default {
 								},
 							},
 
-							// [promptObjectId]: {
-							// 	object_id: promptObjectId,
-							// 	properties: {
-							// 		object_role: 'main_prompt',
-							// 	},
-							// },
+							...(promptObjectId ? {
+								[promptObjectId]: {
+									object_id: promptObjectId,
+									properties: {
+										object_role: 'main_prompt',
+									},
+								},
+							} : {}),
 						},
 					};
 				}