forked from open-webui/open-webui
		
	refac: reset vector db
This commit is contained in:
		
							parent
							
								
									8d108060d5
								
							
						
					
					
						commit
						c2d6d3230b
					
				
					 2 changed files with 99 additions and 34 deletions
				
			
		|  | @ -2,7 +2,6 @@ | ||||||
| 	import fileSaver from 'file-saver'; | 	import fileSaver from 'file-saver'; | ||||||
| 	const { saveAs } = fileSaver; | 	const { saveAs } = fileSaver; | ||||||
| 
 | 
 | ||||||
| 	import { resetVectorDB } from '$lib/apis/rag'; |  | ||||||
| 	import { chats, user } from '$lib/stores'; | 	import { chats, user } from '$lib/stores'; | ||||||
| 
 | 
 | ||||||
| 	import { | 	import { | ||||||
|  | @ -330,38 +329,6 @@ | ||||||
| 					{$i18n.t('Export All Chats (All Users)')} | 					{$i18n.t('Export All Chats (All Users)')} | ||||||
| 				</div> | 				</div> | ||||||
| 			</button> | 			</button> | ||||||
| 
 |  | ||||||
| 			<hr class=" dark:border-gray-700" /> |  | ||||||
| 
 |  | ||||||
| 			<button |  | ||||||
| 				class=" flex rounded-md py-2 px-3.5 w-full hover:bg-gray-200 dark:hover:bg-gray-800 transition" |  | ||||||
| 				on:click={() => { |  | ||||||
| 					const res = resetVectorDB(localStorage.token).catch((error) => { |  | ||||||
| 						toast.error(error); |  | ||||||
| 						return null; |  | ||||||
| 					}); |  | ||||||
| 
 |  | ||||||
| 					if (res) { |  | ||||||
| 						toast.success($i18n.t('Success')); |  | ||||||
| 					} |  | ||||||
| 				}} |  | ||||||
| 			> |  | ||||||
| 				<div class=" self-center mr-3"> |  | ||||||
| 					<svg |  | ||||||
| 						xmlns="http://www.w3.org/2000/svg" |  | ||||||
| 						viewBox="0 0 16 16" |  | ||||||
| 						fill="currentColor" |  | ||||||
| 						class="w-4 h-4" |  | ||||||
| 					> |  | ||||||
| 						<path |  | ||||||
| 							fill-rule="evenodd" |  | ||||||
| 							d="M3.5 2A1.5 1.5 0 0 0 2 3.5v9A1.5 1.5 0 0 0 3.5 14h9a1.5 1.5 0 0 0 1.5-1.5v-7A1.5 1.5 0 0 0 12.5 4H9.621a1.5 1.5 0 0 1-1.06-.44L7.439 2.44A1.5 1.5 0 0 0 6.38 2H3.5Zm6.75 7.75a.75.75 0 0 0 0-1.5h-4.5a.75.75 0 0 0 0 1.5h4.5Z" |  | ||||||
| 							clip-rule="evenodd" |  | ||||||
| 						/> |  | ||||||
| 					</svg> |  | ||||||
| 				</div> |  | ||||||
| 				<div class=" self-center text-sm font-medium">{$i18n.t('Reset Vector Storage')}</div> |  | ||||||
| 			</button> |  | ||||||
| 		{/if} | 		{/if} | ||||||
| 	</div> | 	</div> | ||||||
| </div> | </div> | ||||||
|  |  | ||||||
|  | @ -5,8 +5,10 @@ | ||||||
| 		updateRAGConfig, | 		updateRAGConfig, | ||||||
| 		getQuerySettings, | 		getQuerySettings, | ||||||
| 		scanDocs, | 		scanDocs, | ||||||
| 		updateQuerySettings | 		updateQuerySettings, | ||||||
|  | 		resetVectorDB | ||||||
| 	} from '$lib/apis/rag'; | 	} from '$lib/apis/rag'; | ||||||
|  | 
 | ||||||
| 	import { documents } from '$lib/stores'; | 	import { documents } from '$lib/stores'; | ||||||
| 	import { onMount, getContext } from 'svelte'; | 	import { onMount, getContext } from 'svelte'; | ||||||
| 	import { toast } from 'svelte-sonner'; | 	import { toast } from 'svelte-sonner'; | ||||||
|  | @ -17,6 +19,8 @@ | ||||||
| 
 | 
 | ||||||
| 	let loading = false; | 	let loading = false; | ||||||
| 
 | 
 | ||||||
|  | 	let showResetConfirm = false; | ||||||
|  | 
 | ||||||
| 	let chunkSize = 0; | 	let chunkSize = 0; | ||||||
| 	let chunkOverlap = 0; | 	let chunkOverlap = 0; | ||||||
| 	let pdfExtractImages = true; | 	let pdfExtractImages = true; | ||||||
|  | @ -231,6 +235,100 @@ | ||||||
| 				/> | 				/> | ||||||
| 			</div> | 			</div> | ||||||
| 		</div> | 		</div> | ||||||
|  | 
 | ||||||
|  | 		<hr class=" dark:border-gray-700" /> | ||||||
|  | 
 | ||||||
|  | 		{#if showResetConfirm} | ||||||
|  | 			<div class="flex justify-between rounded-md items-center py-2 px-3.5 w-full transition"> | ||||||
|  | 				<div class="flex items-center space-x-3"> | ||||||
|  | 					<svg | ||||||
|  | 						xmlns="http://www.w3.org/2000/svg" | ||||||
|  | 						viewBox="0 0 16 16" | ||||||
|  | 						fill="currentColor" | ||||||
|  | 						class="w-4 h-4" | ||||||
|  | 					> | ||||||
|  | 						<path d="M2 3a1 1 0 0 1 1-1h10a1 1 0 0 1 1 1v1a1 1 0 0 1-1 1H3a1 1 0 0 1-1-1V3Z" /> | ||||||
|  | 						<path | ||||||
|  | 							fill-rule="evenodd" | ||||||
|  | 							d="M13 6H3v6a2 2 0 0 0 2 2h6a2 2 0 0 0 2-2V6ZM5.72 7.47a.75.75 0 0 1 1.06 0L8 8.69l1.22-1.22a.75.75 0 1 1 1.06 1.06L9.06 9.75l1.22 1.22a.75.75 0 1 1-1.06 1.06L8 10.81l-1.22 1.22a.75.75 0 0 1-1.06-1.06l1.22-1.22-1.22-1.22a.75.75 0 0 1 0-1.06Z" | ||||||
|  | 							clip-rule="evenodd" | ||||||
|  | 						/> | ||||||
|  | 					</svg> | ||||||
|  | 					<span>{$i18n.t('Are you sure?')}</span> | ||||||
|  | 				</div> | ||||||
|  | 
 | ||||||
|  | 				<div class="flex space-x-1.5 items-center"> | ||||||
|  | 					<button | ||||||
|  | 						class="hover:text-white transition" | ||||||
|  | 						on:click={() => { | ||||||
|  | 							const res = resetVectorDB(localStorage.token).catch((error) => { | ||||||
|  | 								toast.error(error); | ||||||
|  | 								return null; | ||||||
|  | 							}); | ||||||
|  | 
 | ||||||
|  | 							if (res) { | ||||||
|  | 								toast.success($i18n.t('Success')); | ||||||
|  | 							} | ||||||
|  | 
 | ||||||
|  | 							showResetConfirm = false; | ||||||
|  | 						}} | ||||||
|  | 					> | ||||||
|  | 						<svg | ||||||
|  | 							xmlns="http://www.w3.org/2000/svg" | ||||||
|  | 							viewBox="0 0 20 20" | ||||||
|  | 							fill="currentColor" | ||||||
|  | 							class="w-4 h-4" | ||||||
|  | 						> | ||||||
|  | 							<path | ||||||
|  | 								fill-rule="evenodd" | ||||||
|  | 								d="M16.704 4.153a.75.75 0 01.143 1.052l-8 10.5a.75.75 0 01-1.127.075l-4.5-4.5a.75.75 0 011.06-1.06l3.894 3.893 7.48-9.817a.75.75 0 011.05-.143z" | ||||||
|  | 								clip-rule="evenodd" | ||||||
|  | 							/> | ||||||
|  | 						</svg> | ||||||
|  | 					</button> | ||||||
|  | 					<button | ||||||
|  | 						class="hover:text-white transition" | ||||||
|  | 						on:click={() => { | ||||||
|  | 							showResetConfirm = false; | ||||||
|  | 						}} | ||||||
|  | 					> | ||||||
|  | 						<svg | ||||||
|  | 							xmlns="http://www.w3.org/2000/svg" | ||||||
|  | 							viewBox="0 0 20 20" | ||||||
|  | 							fill="currentColor" | ||||||
|  | 							class="w-4 h-4" | ||||||
|  | 						> | ||||||
|  | 							<path | ||||||
|  | 								d="M6.28 5.22a.75.75 0 00-1.06 1.06L8.94 10l-3.72 3.72a.75.75 0 101.06 1.06L10 11.06l3.72 3.72a.75.75 0 101.06-1.06L11.06 10l3.72-3.72a.75.75 0 00-1.06-1.06L10 8.94 6.28 5.22z" | ||||||
|  | 							/> | ||||||
|  | 						</svg> | ||||||
|  | 					</button> | ||||||
|  | 				</div> | ||||||
|  | 			</div> | ||||||
|  | 		{:else} | ||||||
|  | 			<button | ||||||
|  | 				class=" flex rounded-md py-2 px-3.5 w-full hover:bg-gray-200 dark:hover:bg-gray-800 transition" | ||||||
|  | 				on:click={() => { | ||||||
|  | 					showResetConfirm = true; | ||||||
|  | 				}} | ||||||
|  | 			> | ||||||
|  | 				<div class=" self-center mr-3"> | ||||||
|  | 					<svg | ||||||
|  | 						xmlns="http://www.w3.org/2000/svg" | ||||||
|  | 						viewBox="0 0 16 16" | ||||||
|  | 						fill="currentColor" | ||||||
|  | 						class="w-4 h-4" | ||||||
|  | 					> | ||||||
|  | 						<path | ||||||
|  | 							fill-rule="evenodd" | ||||||
|  | 							d="M3.5 2A1.5 1.5 0 0 0 2 3.5v9A1.5 1.5 0 0 0 3.5 14h9a1.5 1.5 0 0 0 1.5-1.5v-7A1.5 1.5 0 0 0 12.5 4H9.621a1.5 1.5 0 0 1-1.06-.44L7.439 2.44A1.5 1.5 0 0 0 6.38 2H3.5Zm6.75 7.75a.75.75 0 0 0 0-1.5h-4.5a.75.75 0 0 0 0 1.5h4.5Z" | ||||||
|  | 							clip-rule="evenodd" | ||||||
|  | 						/> | ||||||
|  | 					</svg> | ||||||
|  | 				</div> | ||||||
|  | 				<div class=" self-center text-sm font-medium">{$i18n.t('Reset Vector Storage')}</div> | ||||||
|  | 			</button> | ||||||
|  | 		{/if} | ||||||
| 	</div> | 	</div> | ||||||
| 
 | 
 | ||||||
| 	<div class="flex justify-end pt-3 text-sm font-medium"> | 	<div class="flex justify-end pt-3 text-sm font-medium"> | ||||||
|  |  | ||||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue
	
	 Timothy J. Baek
						Timothy J. Baek