Enhance AIChat with dynamic assistant update and reset functionality

YaoApp · Feb 9, 2025 · 65f1548 · 65f1548
1 parent cf29dc1
commit 65f1548
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 1 deletion.
diff --git a/packages/xgen/layouts/components/Neo/components/AIChat/index.tsx b/packages/xgen/layouts/components/Neo/components/AIChat/index.tsx
@@ -66,6 +66,7 @@ const AIChat = (props: AIChatProps) => {
 		cancel,
 		uploadFile,
 		attachments,
+		updateAssistant,
 		resetAssistant,
 		removeAttachment,
 		addAttachment,
@@ -231,6 +232,13 @@ const AIChat = (props: AIChatProps) => {
 			})
 		}
 
+		// Reset the assistant
+		if (options?.assistant) {
+			updateAssistant(options?.assistant)
+		} else {
+			resetAssistant()
+		}
+
 		// Focus using the new method
 		setTimeout(() => {
 			focusRef.current?.()
@@ -567,7 +575,10 @@ const AIChat = (props: AIChatProps) => {
 							<Assistant
 								assistant={assistant || {}}
 								loading={loadingChat}
-								onDelete={resetAssistant}
+								onDelete={() => {
+									resetAssistant()
+									focusRef.current?.() // focus on the input
+								}}
 							/>
 							{/* Current Page Info */}
 							{showCurrentPage && currentPage && (

diff --git a/packages/xgen/layouts/components/Neo/hooks/useAIChat.ts b/packages/xgen/layouts/components/Neo/hooks/useAIChat.ts
@@ -1400,6 +1400,7 @@ export default ({ chat_id, upload_options = {} }: Args) => {
 		generatePrompts,
 		titleGenerating,
 		resetAssistant,
+		updateAssistant,
 		setTitleGenerating,
 		getAssistants,
 		findAssistant,

diff --git a/packages/xgen/types/app.ts b/packages/xgen/types/app.ts
@@ -338,6 +338,7 @@ export declare namespace App {
 	interface NewChatOptions {
 		content?: string
 		chat_id?: string
+		assistant?: AssistantSummary
 		attachments?: ChatAttachment[]
 		placeholder?: ChatPlaceholder
 	}