Merge pull request #71 from iceljc/features/delete-logs-with-message

add delete log
SciSharp · Feb 28, 2024 · b30a2ee · b30a2ee
2 parents eae6d0b + 6dbb4a0
commit b30a2ee
Show file tree

Hide file tree

Showing 3 changed files with 44 additions and 2 deletions.
diff --git a/src/routes/chat/[agentId]/[conversationId]/chat-box.svelte b/src/routes/chat/[agentId]/[conversationId]/chat-box.svelte
@@ -93,6 +93,10 @@
 	let isLite = false;
 	let isFrame = false;
 
+	/** @type {any} */
+	let contentLogComponent;
+	/** @type {any} */
+	let stateLogComponent;
 	
 	onMount(async () => {
 		dialogs = await GetDialogs(params.conversationId);
@@ -532,10 +536,22 @@
 		const foundIdx = dialogs.findIndex(x => x.message_id === messageId);
 		if (foundIdx < 0) return false;
 		dialogs = dialogs.filter((x, idx) => idx < foundIdx);
+		truncateLogs(messageId);
 		refresh();
 		return true;
 	}
 
+	/** @param {string} messageId */
+	function truncateLogs(messageId) {
+		if (contentLogComponent && contentLogComponent.onDeleteMessage) {
+			contentLogComponent.onDeleteMessage(messageId);
+		}
+		
+		if (stateLogComponent && stateLogComponent.onDeleteMessage) {
+			stateLogComponent.onDeleteMessage(messageId);
+		}
+	}
+
 	/** @param {string} messageId */
 	function directToLog(messageId) {
 		if (!!!messageId || isLite) return;
@@ -610,7 +626,12 @@
 	<Splitpanes>
 		{#if isLoadStateLog}
 		<Pane size={30} minSize={20} maxSize={50} >
-			<StateLog stateLogs={stateLogs} closeWindow={toggleStateLog} cleanScreen={cleanStateLogScreen} />
+			<StateLog
+				bind:this={stateLogComponent}
+				stateLogs={stateLogs}
+				closeWindow={toggleStateLog}
+				cleanScreen={cleanStateLogScreen}
+			/>
 		</Pane>
 		{/if}
 		<Pane minSize={20}>
@@ -813,7 +834,12 @@
 		</Pane>
 		{#if isLoadContentLog}
 		<Pane size={30} minSize={20} maxSize={50}>
-			<ContentLog contentLogs={contentLogs} closeWindow={toggleContentLog} cleanScreen={cleanContentLogScreen} />
+			<ContentLog
+				bind:this={contentLogComponent}
+				contentLogs={contentLogs}
+				closeWindow={toggleContentLog}
+				cleanScreen={cleanContentLogScreen}
+			/>
 		</Pane>
 		{/if}
 	</Splitpanes>

diff --git a/src/routes/chat/[agentId]/[conversationId]/contentLogs/content-log.svelte b/src/routes/chat/[agentId]/[conversationId]/contentLogs/content-log.svelte
@@ -15,6 +15,11 @@
     /** @type {() => void} */
     export let cleanScreen;
 
+    export const onDeleteMessage = (/** @type {string} */ messageId) => {
+        const targetIdx = allLogs.findIndex(x => x.message_id === messageId);
+        allLogs = allLogs.filter((x, idx) => idx < targetIdx);
+    }
+
     // @ts-ignore
     let scrollbar;
     /** @type {import('$types').ConversationContentLogModel[]} */
@@ -67,6 +72,12 @@
         cleanLogs();
         cleanScreen && cleanScreen();
     }
+
+    // /** @param {string} messageId */
+    // function onDeleteMessage(messageId) {
+    //     const targetIdx = allLogs.findIndex(x => x.message_id === messageId);
+    //     allLogs = allLogs.filter((x, idx) => idx < targetIdx);
+    // }
 </script>
 
 <div class="chat-log">

diff --git a/src/routes/chat/[agentId]/[conversationId]/stateLogs/state-log.svelte b/src/routes/chat/[agentId]/[conversationId]/stateLogs/state-log.svelte
@@ -15,6 +15,11 @@
     /** @type {() => void} */
     export let cleanScreen;
 
+    export const onDeleteMessage = (/** @type {string} */ messageId) => {
+        const targetIdx = allLogs.findIndex(x => x.message_id === messageId);
+        allLogs = allLogs.filter((x, idx) => idx < targetIdx);
+    }
+
     // @ts-ignore
     let scrollbar;
     /** @type {any[]} */