Chat.svelte 52 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023
  1. <script lang="ts">
  2. import { v4 as uuidv4 } from 'uuid';
  3. import { toast } from 'svelte-sonner';
  4. import mermaid from 'mermaid';
  5. import { PaneGroup, Pane, PaneResizer } from 'paneforge';
  6. import { getContext, onDestroy, onMount, tick } from 'svelte';
  7. const i18n: Writable<i18nType> = getContext('i18n');
  8. import { goto } from '$app/navigation';
  9. import { page } from '$app/stores';
  10. import { get, type Unsubscriber, type Writable } from 'svelte/store';
  11. import type { i18n as i18nType } from 'i18next';
  12. import { WEBUI_BASE_URL } from '$lib/constants';
  13. import {
  14. chatId,
  15. chats,
  16. config,
  17. type Model,
  18. models,
  19. tags as allTags,
  20. settings,
  21. showSidebar,
  22. WEBUI_NAME,
  23. banners,
  24. user,
  25. socket,
  26. showControls,
  27. showCallOverlay,
  28. currentChatPage,
  29. temporaryChatEnabled,
  30. mobile,
  31. showOverview,
  32. chatTitle,
  33. showArtifacts,
  34. tools
  35. } from '$lib/stores';
  36. import {
  37. convertMessagesToHistory,
  38. copyToClipboard,
  39. getMessageContentParts,
  40. extractSentencesForAudio,
  41. promptTemplate,
  42. splitStream
  43. } from '$lib/utils';
  44. import { generateChatCompletion } from '$lib/apis/ollama';
  45. import {
  46. addTagById,
  47. createNewChat,
  48. deleteTagById,
  49. deleteTagsById,
  50. getAllTags,
  51. getChatById,
  52. getChatList,
  53. getTagsById,
  54. updateChatById
  55. } from '$lib/apis/chats';
  56. import { generateOpenAIChatCompletion } from '$lib/apis/openai';
  57. import { processWeb, processWebSearch, processYoutubeVideo } from '$lib/apis/retrieval';
  58. import { createOpenAITextStream } from '$lib/apis/streaming';
  59. import { queryMemory } from '$lib/apis/memories';
  60. import { getAndUpdateUserLocation, getUserSettings } from '$lib/apis/users';
  61. import {
  62. chatCompleted,
  63. generateQueries,
  64. chatAction,
  65. generateMoACompletion,
  66. stopTask
  67. } from '$lib/apis';
  68. import Banner from '../common/Banner.svelte';
  69. import MessageInput from '$lib/components/chat/MessageInput.svelte';
  70. import Messages from '$lib/components/chat/Messages.svelte';
  71. import Navbar from '$lib/components/layout/Navbar.svelte';
  72. import ChatControls from './ChatControls.svelte';
  73. import EventConfirmDialog from '../common/ConfirmDialog.svelte';
  74. import Placeholder from './Placeholder.svelte';
  75. import { getTools } from '$lib/apis/tools';
  76. export let chatIdProp = '';
  77. let loaded = false;
  78. const eventTarget = new EventTarget();
  79. let controlPane;
  80. let controlPaneComponent;
  81. let autoScroll = true;
  82. let processing = '';
  83. let messagesContainerElement: HTMLDivElement;
  84. let navbarElement;
  85. let showEventConfirmation = false;
  86. let eventConfirmationTitle = '';
  87. let eventConfirmationMessage = '';
  88. let eventConfirmationInput = false;
  89. let eventConfirmationInputPlaceholder = '';
  90. let eventConfirmationInputValue = '';
  91. let eventCallback = null;
  92. let chatIdUnsubscriber: Unsubscriber | undefined;
  93. let selectedModels = [''];
  94. let atSelectedModel: Model | undefined;
  95. let selectedModelIds = [];
  96. $: selectedModelIds = atSelectedModel !== undefined ? [atSelectedModel.id] : selectedModels;
  97. let selectedToolIds = [];
  98. let webSearchEnabled = false;
  99. let chat = null;
  100. let tags = [];
  101. let history = {
  102. messages: {},
  103. currentId: null
  104. };
  105. let taskId = null;
  106. // Chat Input
  107. let prompt = '';
  108. let chatFiles = [];
  109. let files = [];
  110. let params = {};
  111. $: if (chatIdProp) {
  112. (async () => {
  113. console.log(chatIdProp);
  114. if (chatIdProp && (await loadChat())) {
  115. await tick();
  116. loaded = true;
  117. window.setTimeout(() => scrollToBottom(), 0);
  118. const chatInput = document.getElementById('chat-input');
  119. chatInput?.focus();
  120. } else {
  121. await goto('/');
  122. }
  123. })();
  124. }
  125. $: if (selectedModels && chatIdProp !== '') {
  126. saveSessionSelectedModels();
  127. }
  128. const saveSessionSelectedModels = () => {
  129. if (selectedModels.length === 0 || (selectedModels.length === 1 && selectedModels[0] === '')) {
  130. return;
  131. }
  132. sessionStorage.selectedModels = JSON.stringify(selectedModels);
  133. console.log('saveSessionSelectedModels', selectedModels, sessionStorage.selectedModels);
  134. };
  135. $: if (selectedModels) {
  136. setToolIds();
  137. }
  138. const setToolIds = async () => {
  139. if (!$tools) {
  140. tools.set(await getTools(localStorage.token));
  141. }
  142. if (selectedModels.length !== 1) {
  143. return;
  144. }
  145. const model = $models.find((m) => m.id === selectedModels[0]);
  146. if (model) {
  147. selectedToolIds = (model?.info?.meta?.toolIds ?? []).filter((id) =>
  148. $tools.find((t) => t.id === id)
  149. );
  150. }
  151. };
  152. const showMessage = async (message) => {
  153. const _chatId = JSON.parse(JSON.stringify($chatId));
  154. let _messageId = JSON.parse(JSON.stringify(message.id));
  155. let messageChildrenIds = history.messages[_messageId].childrenIds;
  156. while (messageChildrenIds.length !== 0) {
  157. _messageId = messageChildrenIds.at(-1);
  158. messageChildrenIds = history.messages[_messageId].childrenIds;
  159. }
  160. history.currentId = _messageId;
  161. await tick();
  162. await tick();
  163. await tick();
  164. const messageElement = document.getElementById(`message-${message.id}`);
  165. if (messageElement) {
  166. messageElement.scrollIntoView({ behavior: 'smooth' });
  167. }
  168. await tick();
  169. saveChatHandler(_chatId);
  170. };
  171. const chatEventHandler = async (event, cb) => {
  172. console.log(event);
  173. if (event.chat_id === $chatId) {
  174. await tick();
  175. let message = history.messages[event.message_id];
  176. if (message) {
  177. const type = event?.data?.type ?? null;
  178. const data = event?.data?.data ?? null;
  179. if (type === 'status') {
  180. if (message?.statusHistory) {
  181. message.statusHistory.push(data);
  182. } else {
  183. message.statusHistory = [data];
  184. }
  185. } else if (type === 'source' || type === 'citation') {
  186. if (data?.type === 'code_execution') {
  187. // Code execution; update existing code execution by ID, or add new one.
  188. if (!message?.code_executions) {
  189. message.code_executions = [];
  190. }
  191. const existingCodeExecutionIndex = message.code_executions.findIndex(
  192. (execution) => execution.id === data.id
  193. );
  194. if (existingCodeExecutionIndex !== -1) {
  195. message.code_executions[existingCodeExecutionIndex] = data;
  196. } else {
  197. message.code_executions.push(data);
  198. }
  199. message.code_executions = message.code_executions;
  200. } else {
  201. // Regular source.
  202. if (message?.sources) {
  203. message.sources.push(data);
  204. } else {
  205. message.sources = [data];
  206. }
  207. }
  208. } else if (type === 'chat-completion') {
  209. chatCompletionEventHandler(data, message, event.chat_id);
  210. } else if (type === 'chat-title') {
  211. chatTitle.set(data);
  212. currentChatPage.set(1);
  213. await chats.set(await getChatList(localStorage.token, $currentChatPage));
  214. } else if (type === 'chat-tags') {
  215. chat = await getChatById(localStorage.token, $chatId);
  216. allTags.set(await getAllTags(localStorage.token));
  217. } else if (type === 'message') {
  218. message.content += data.content;
  219. } else if (type === 'replace') {
  220. message.content = data.content;
  221. } else if (type === 'action') {
  222. if (data.action === 'continue') {
  223. const continueButton = document.getElementById('continue-response-button');
  224. if (continueButton) {
  225. continueButton.click();
  226. }
  227. }
  228. } else if (type === 'confirmation') {
  229. eventCallback = cb;
  230. eventConfirmationInput = false;
  231. showEventConfirmation = true;
  232. eventConfirmationTitle = data.title;
  233. eventConfirmationMessage = data.message;
  234. } else if (type === 'execute') {
  235. eventCallback = cb;
  236. try {
  237. // Use Function constructor to evaluate code in a safer way
  238. const asyncFunction = new Function(`return (async () => { ${data.code} })()`);
  239. const result = await asyncFunction(); // Await the result of the async function
  240. if (cb) {
  241. cb(result);
  242. }
  243. } catch (error) {
  244. console.error('Error executing code:', error);
  245. }
  246. } else if (type === 'input') {
  247. eventCallback = cb;
  248. eventConfirmationInput = true;
  249. showEventConfirmation = true;
  250. eventConfirmationTitle = data.title;
  251. eventConfirmationMessage = data.message;
  252. eventConfirmationInputPlaceholder = data.placeholder;
  253. eventConfirmationInputValue = data?.value ?? '';
  254. } else {
  255. console.log('Unknown message type', data);
  256. }
  257. history.messages[event.message_id] = message;
  258. }
  259. }
  260. };
  261. const onMessageHandler = async (event: {
  262. origin: string;
  263. data: { type: string; text: string };
  264. }) => {
  265. if (event.origin !== window.origin) {
  266. return;
  267. }
  268. // Replace with your iframe's origin
  269. if (event.data.type === 'input:prompt') {
  270. console.debug(event.data.text);
  271. const inputElement = document.getElementById('chat-input');
  272. if (inputElement) {
  273. prompt = event.data.text;
  274. inputElement.focus();
  275. }
  276. }
  277. if (event.data.type === 'action:submit') {
  278. console.debug(event.data.text);
  279. if (prompt !== '') {
  280. await tick();
  281. submitPrompt(prompt);
  282. }
  283. }
  284. if (event.data.type === 'input:prompt:submit') {
  285. console.debug(event.data.text);
  286. if (prompt !== '') {
  287. await tick();
  288. submitPrompt(event.data.text);
  289. }
  290. }
  291. };
  292. onMount(async () => {
  293. console.log('mounted');
  294. window.addEventListener('message', onMessageHandler);
  295. $socket?.on('chat-events', chatEventHandler);
  296. if (!$chatId) {
  297. chatIdUnsubscriber = chatId.subscribe(async (value) => {
  298. if (!value) {
  299. await initNewChat();
  300. }
  301. });
  302. } else {
  303. if ($temporaryChatEnabled) {
  304. await goto('/');
  305. }
  306. }
  307. showControls.subscribe(async (value) => {
  308. if (controlPane && !$mobile) {
  309. try {
  310. if (value) {
  311. controlPaneComponent.openPane();
  312. } else {
  313. controlPane.collapse();
  314. }
  315. } catch (e) {
  316. // ignore
  317. }
  318. }
  319. if (!value) {
  320. showCallOverlay.set(false);
  321. showOverview.set(false);
  322. showArtifacts.set(false);
  323. }
  324. });
  325. const chatInput = document.getElementById('chat-input');
  326. chatInput?.focus();
  327. chats.subscribe(() => {});
  328. });
  329. onDestroy(() => {
  330. chatIdUnsubscriber?.();
  331. window.removeEventListener('message', onMessageHandler);
  332. $socket?.off('chat-events');
  333. });
  334. // File upload functions
  335. const uploadGoogleDriveFile = async (fileData) => {
  336. console.log('Starting uploadGoogleDriveFile with:', {
  337. id: fileData.id,
  338. name: fileData.name,
  339. url: fileData.url,
  340. headers: {
  341. Authorization: `Bearer ${token}`
  342. }
  343. });
  344. // Validate input
  345. if (!fileData?.id || !fileData?.name || !fileData?.url || !fileData?.headers?.Authorization) {
  346. throw new Error('Invalid file data provided');
  347. }
  348. const tempItemId = uuidv4();
  349. const fileItem = {
  350. type: 'file',
  351. file: '',
  352. id: null,
  353. url: fileData.url,
  354. name: fileData.name,
  355. collection_name: '',
  356. status: 'uploading',
  357. error: '',
  358. itemId: tempItemId,
  359. size: 0
  360. };
  361. try {
  362. files = [...files, fileItem];
  363. console.log('Processing web file with URL:', fileData.url);
  364. // Configure fetch options with proper headers
  365. const fetchOptions = {
  366. headers: {
  367. Authorization: fileData.headers.Authorization,
  368. Accept: '*/*'
  369. },
  370. method: 'GET'
  371. };
  372. // Attempt to fetch the file
  373. console.log('Fetching file content from Google Drive...');
  374. const fileResponse = await fetch(fileData.url, fetchOptions);
  375. if (!fileResponse.ok) {
  376. const errorText = await fileResponse.text();
  377. throw new Error(`Failed to fetch file (${fileResponse.status}): ${errorText}`);
  378. }
  379. // Get content type from response
  380. const contentType = fileResponse.headers.get('content-type') || 'application/octet-stream';
  381. console.log('Response received with content-type:', contentType);
  382. // Convert response to blob
  383. console.log('Converting response to blob...');
  384. const fileBlob = await fileResponse.blob();
  385. if (fileBlob.size === 0) {
  386. throw new Error('Retrieved file is empty');
  387. }
  388. console.log('Blob created:', {
  389. size: fileBlob.size,
  390. type: fileBlob.type || contentType
  391. });
  392. // Create File object with proper MIME type
  393. const file = new File([fileBlob], fileData.name, {
  394. type: fileBlob.type || contentType
  395. });
  396. console.log('File object created:', {
  397. name: file.name,
  398. size: file.size,
  399. type: file.type
  400. });
  401. if (file.size === 0) {
  402. throw new Error('Created file is empty');
  403. }
  404. // Upload file to server
  405. console.log('Uploading file to server...');
  406. const uploadedFile = await uploadFile(localStorage.token, file);
  407. if (!uploadedFile) {
  408. throw new Error('Server returned null response for file upload');
  409. }
  410. console.log('File uploaded successfully:', uploadedFile);
  411. // Update file item with upload results
  412. fileItem.status = 'uploaded';
  413. fileItem.file = uploadedFile;
  414. fileItem.id = uploadedFile.id;
  415. fileItem.size = file.size;
  416. fileItem.collection_name = uploadedFile?.meta?.collection_name;
  417. fileItem.url = `${WEBUI_API_BASE_URL}/files/${uploadedFile.id}`;
  418. files = files;
  419. toast.success($i18n.t('File uploaded successfully'));
  420. } catch (e) {
  421. console.error('Error uploading file:', e);
  422. files = files.filter((f) => f.itemId !== tempItemId);
  423. toast.error(
  424. $i18n.t('Error uploading file: {{error}}', {
  425. error: e.message || 'Unknown error'
  426. })
  427. );
  428. }
  429. };
  430. const handleGoogleDrivePicker = async () => {
  431. try {
  432. const fileData = await createPicker();
  433. if (fileData) {
  434. await uploadGoogleDriveFile(fileData);
  435. }
  436. } catch (error) {
  437. toast.error('Error accessing Google Drive: ' + error.message);
  438. }
  439. };
  440. const uploadWeb = async (url) => {
  441. console.log(url);
  442. const fileItem = {
  443. type: 'doc',
  444. name: url,
  445. collection_name: '',
  446. status: 'uploading',
  447. url: url,
  448. error: ''
  449. };
  450. try {
  451. files = [...files, fileItem];
  452. const res = await processWeb(localStorage.token, '', url);
  453. if (res) {
  454. fileItem.status = 'uploaded';
  455. fileItem.collection_name = res.collection_name;
  456. fileItem.file = {
  457. ...res.file,
  458. ...fileItem.file
  459. };
  460. files = files;
  461. }
  462. } catch (e) {
  463. // Remove the failed doc from the files array
  464. files = files.filter((f) => f.name !== url);
  465. toast.error(JSON.stringify(e));
  466. }
  467. };
  468. const uploadYoutubeTranscription = async (url) => {
  469. console.log(url);
  470. const fileItem = {
  471. type: 'doc',
  472. name: url,
  473. collection_name: '',
  474. status: 'uploading',
  475. context: 'full',
  476. url: url,
  477. error: ''
  478. };
  479. try {
  480. files = [...files, fileItem];
  481. const res = await processYoutubeVideo(localStorage.token, url);
  482. if (res) {
  483. fileItem.status = 'uploaded';
  484. fileItem.collection_name = res.collection_name;
  485. fileItem.file = {
  486. ...res.file,
  487. ...fileItem.file
  488. };
  489. files = files;
  490. }
  491. } catch (e) {
  492. // Remove the failed doc from the files array
  493. files = files.filter((f) => f.name !== url);
  494. toast.error(e);
  495. }
  496. };
  497. //////////////////////////
  498. // Web functions
  499. //////////////////////////
  500. const initNewChat = async () => {
  501. if ($page.url.searchParams.get('models')) {
  502. selectedModels = $page.url.searchParams.get('models')?.split(',');
  503. } else if ($page.url.searchParams.get('model')) {
  504. const urlModels = $page.url.searchParams.get('model')?.split(',');
  505. if (urlModels.length === 1) {
  506. const m = $models.find((m) => m.id === urlModels[0]);
  507. if (!m) {
  508. const modelSelectorButton = document.getElementById('model-selector-0-button');
  509. if (modelSelectorButton) {
  510. modelSelectorButton.click();
  511. await tick();
  512. const modelSelectorInput = document.getElementById('model-search-input');
  513. if (modelSelectorInput) {
  514. modelSelectorInput.focus();
  515. modelSelectorInput.value = urlModels[0];
  516. modelSelectorInput.dispatchEvent(new Event('input'));
  517. }
  518. }
  519. } else {
  520. selectedModels = urlModels;
  521. }
  522. } else {
  523. selectedModels = urlModels;
  524. }
  525. } else {
  526. if (sessionStorage.selectedModels) {
  527. selectedModels = JSON.parse(sessionStorage.selectedModels);
  528. sessionStorage.removeItem('selectedModels');
  529. } else {
  530. if ($settings?.models) {
  531. selectedModels = $settings?.models;
  532. } else if ($config?.default_models) {
  533. console.log($config?.default_models.split(',') ?? '');
  534. selectedModels = $config?.default_models.split(',');
  535. }
  536. }
  537. }
  538. selectedModels = selectedModels.filter((modelId) => $models.map((m) => m.id).includes(modelId));
  539. if (selectedModels.length === 0 || (selectedModels.length === 1 && selectedModels[0] === '')) {
  540. if ($models.length > 0) {
  541. selectedModels = [$models[0].id];
  542. } else {
  543. selectedModels = [''];
  544. }
  545. }
  546. await showControls.set(false);
  547. await showCallOverlay.set(false);
  548. await showOverview.set(false);
  549. await showArtifacts.set(false);
  550. if ($page.url.pathname.includes('/c/')) {
  551. window.history.replaceState(history.state, '', `/`);
  552. }
  553. autoScroll = true;
  554. await chatId.set('');
  555. await chatTitle.set('');
  556. history = {
  557. messages: {},
  558. currentId: null
  559. };
  560. chatFiles = [];
  561. params = {};
  562. if ($page.url.searchParams.get('youtube')) {
  563. uploadYoutubeTranscription(
  564. `https://www.youtube.com/watch?v=${$page.url.searchParams.get('youtube')}`
  565. );
  566. }
  567. if ($page.url.searchParams.get('web-search') === 'true') {
  568. webSearchEnabled = true;
  569. }
  570. if ($page.url.searchParams.get('tools')) {
  571. selectedToolIds = $page.url.searchParams
  572. .get('tools')
  573. ?.split(',')
  574. .map((id) => id.trim())
  575. .filter((id) => id);
  576. } else if ($page.url.searchParams.get('tool-ids')) {
  577. selectedToolIds = $page.url.searchParams
  578. .get('tool-ids')
  579. ?.split(',')
  580. .map((id) => id.trim())
  581. .filter((id) => id);
  582. }
  583. if ($page.url.searchParams.get('call') === 'true') {
  584. showCallOverlay.set(true);
  585. showControls.set(true);
  586. }
  587. if ($page.url.searchParams.get('q')) {
  588. prompt = $page.url.searchParams.get('q') ?? '';
  589. if (prompt) {
  590. await tick();
  591. submitPrompt(prompt);
  592. }
  593. }
  594. selectedModels = selectedModels.map((modelId) =>
  595. $models.map((m) => m.id).includes(modelId) ? modelId : ''
  596. );
  597. const userSettings = await getUserSettings(localStorage.token);
  598. if (userSettings) {
  599. settings.set(userSettings.ui);
  600. } else {
  601. settings.set(JSON.parse(localStorage.getItem('settings') ?? '{}'));
  602. }
  603. const chatInput = document.getElementById('chat-input');
  604. setTimeout(() => chatInput?.focus(), 0);
  605. };
  606. const loadChat = async () => {
  607. chatId.set(chatIdProp);
  608. chat = await getChatById(localStorage.token, $chatId).catch(async (error) => {
  609. await goto('/');
  610. return null;
  611. });
  612. if (chat) {
  613. tags = await getTagsById(localStorage.token, $chatId).catch(async (error) => {
  614. return [];
  615. });
  616. const chatContent = chat.chat;
  617. if (chatContent) {
  618. console.log(chatContent);
  619. selectedModels =
  620. (chatContent?.models ?? undefined) !== undefined
  621. ? chatContent.models
  622. : [chatContent.models ?? ''];
  623. history =
  624. (chatContent?.history ?? undefined) !== undefined
  625. ? chatContent.history
  626. : convertMessagesToHistory(chatContent.messages);
  627. chatTitle.set(chatContent.title);
  628. const userSettings = await getUserSettings(localStorage.token);
  629. if (userSettings) {
  630. await settings.set(userSettings.ui);
  631. } else {
  632. await settings.set(JSON.parse(localStorage.getItem('settings') ?? '{}'));
  633. }
  634. params = chatContent?.params ?? {};
  635. chatFiles = chatContent?.files ?? [];
  636. autoScroll = true;
  637. await tick();
  638. if (history.currentId) {
  639. history.messages[history.currentId].done = true;
  640. }
  641. await tick();
  642. return true;
  643. } else {
  644. return null;
  645. }
  646. }
  647. };
  648. const scrollToBottom = async () => {
  649. await tick();
  650. if (messagesContainerElement) {
  651. messagesContainerElement.scrollTop = messagesContainerElement.scrollHeight;
  652. }
  653. };
  654. const createMessagesList = (responseMessageId) => {
  655. if (responseMessageId === null) {
  656. return [];
  657. }
  658. const message = history.messages[responseMessageId];
  659. if (message?.parentId) {
  660. return [...createMessagesList(message.parentId), message];
  661. } else {
  662. return [message];
  663. }
  664. };
  665. const chatCompletedHandler = async (chatId, modelId, responseMessageId, messages) => {
  666. await mermaid.run({
  667. querySelector: '.mermaid'
  668. });
  669. const res = await chatCompleted(localStorage.token, {
  670. model: modelId,
  671. messages: messages.map((m) => ({
  672. id: m.id,
  673. role: m.role,
  674. content: m.content,
  675. info: m.info ? m.info : undefined,
  676. timestamp: m.timestamp,
  677. ...(m.sources ? { sources: m.sources } : {})
  678. })),
  679. chat_id: chatId,
  680. session_id: $socket?.id,
  681. id: responseMessageId
  682. }).catch((error) => {
  683. toast.error(error);
  684. messages.at(-1).error = { content: error };
  685. return null;
  686. });
  687. if (res !== null) {
  688. // Update chat history with the new messages
  689. for (const message of res.messages) {
  690. history.messages[message.id] = {
  691. ...history.messages[message.id],
  692. ...(history.messages[message.id].content !== message.content
  693. ? { originalContent: history.messages[message.id].content }
  694. : {}),
  695. ...message
  696. };
  697. }
  698. }
  699. await tick();
  700. if ($chatId == chatId) {
  701. if (!$temporaryChatEnabled) {
  702. chat = await updateChatById(localStorage.token, chatId, {
  703. models: selectedModels,
  704. messages: messages,
  705. history: history,
  706. params: params,
  707. files: chatFiles
  708. });
  709. currentChatPage.set(1);
  710. await chats.set(await getChatList(localStorage.token, $currentChatPage));
  711. }
  712. }
  713. };
  714. const chatActionHandler = async (chatId, actionId, modelId, responseMessageId, event = null) => {
  715. const messages = createMessagesList(responseMessageId);
  716. const res = await chatAction(localStorage.token, actionId, {
  717. model: modelId,
  718. messages: messages.map((m) => ({
  719. id: m.id,
  720. role: m.role,
  721. content: m.content,
  722. info: m.info ? m.info : undefined,
  723. timestamp: m.timestamp,
  724. ...(m.sources ? { sources: m.sources } : {})
  725. })),
  726. ...(event ? { event: event } : {}),
  727. chat_id: chatId,
  728. session_id: $socket?.id,
  729. id: responseMessageId
  730. }).catch((error) => {
  731. toast.error(error);
  732. messages.at(-1).error = { content: error };
  733. return null;
  734. });
  735. if (res !== null) {
  736. // Update chat history with the new messages
  737. for (const message of res.messages) {
  738. history.messages[message.id] = {
  739. ...history.messages[message.id],
  740. ...(history.messages[message.id].content !== message.content
  741. ? { originalContent: history.messages[message.id].content }
  742. : {}),
  743. ...message
  744. };
  745. }
  746. }
  747. if ($chatId == chatId) {
  748. if (!$temporaryChatEnabled) {
  749. chat = await updateChatById(localStorage.token, chatId, {
  750. models: selectedModels,
  751. messages: messages,
  752. history: history,
  753. params: params,
  754. files: chatFiles
  755. });
  756. currentChatPage.set(1);
  757. await chats.set(await getChatList(localStorage.token, $currentChatPage));
  758. }
  759. }
  760. };
  761. const getChatEventEmitter = async (modelId: string, chatId: string = '') => {
  762. return setInterval(() => {
  763. $socket?.emit('usage', {
  764. action: 'chat',
  765. model: modelId,
  766. chat_id: chatId
  767. });
  768. }, 1000);
  769. };
  770. const createMessagePair = async (userPrompt) => {
  771. prompt = '';
  772. if (selectedModels.length === 0) {
  773. toast.error($i18n.t('Model not selected'));
  774. } else {
  775. const modelId = selectedModels[0];
  776. const model = $models.filter((m) => m.id === modelId).at(0);
  777. const messages = createMessagesList(history.currentId);
  778. const parentMessage = messages.length !== 0 ? messages.at(-1) : null;
  779. const userMessageId = uuidv4();
  780. const responseMessageId = uuidv4();
  781. const userMessage = {
  782. id: userMessageId,
  783. parentId: parentMessage ? parentMessage.id : null,
  784. childrenIds: [responseMessageId],
  785. role: 'user',
  786. content: userPrompt ? userPrompt : `[PROMPT] ${userMessageId}`,
  787. timestamp: Math.floor(Date.now() / 1000)
  788. };
  789. const responseMessage = {
  790. id: responseMessageId,
  791. parentId: userMessageId,
  792. childrenIds: [],
  793. role: 'assistant',
  794. content: `[RESPONSE] ${responseMessageId}`,
  795. done: true,
  796. model: modelId,
  797. modelName: model.name ?? model.id,
  798. modelIdx: 0,
  799. timestamp: Math.floor(Date.now() / 1000)
  800. };
  801. if (parentMessage) {
  802. parentMessage.childrenIds.push(userMessageId);
  803. history.messages[parentMessage.id] = parentMessage;
  804. }
  805. history.messages[userMessageId] = userMessage;
  806. history.messages[responseMessageId] = responseMessage;
  807. history.currentId = responseMessageId;
  808. await tick();
  809. if (autoScroll) {
  810. scrollToBottom();
  811. }
  812. if (messages.length === 0) {
  813. await initChatHandler();
  814. } else {
  815. await saveChatHandler($chatId);
  816. }
  817. }
  818. };
  819. const chatCompletionEventHandler = async (data, message, chatId) => {
  820. const { id, done, choices, sources, selectedModelId, error, usage } = data;
  821. if (error) {
  822. await handleOpenAIError(error, message);
  823. }
  824. if (sources) {
  825. message.sources = sources;
  826. // Only remove status if it was initially set
  827. if (model?.info?.meta?.knowledge ?? false) {
  828. message.statusHistory = message.statusHistory.filter(
  829. (status) => status.action !== 'knowledge_search'
  830. );
  831. }
  832. }
  833. if (choices) {
  834. const value = choices[0]?.delta?.content ?? '';
  835. if (message.content == '' && value == '\n') {
  836. console.log('Empty response');
  837. } else {
  838. message.content += value;
  839. if (navigator.vibrate && ($settings?.hapticFeedback ?? false)) {
  840. navigator.vibrate(5);
  841. }
  842. // Emit chat event for TTS
  843. const messageContentParts = getMessageContentParts(
  844. message.content,
  845. $config?.audio?.tts?.split_on ?? 'punctuation'
  846. );
  847. messageContentParts.pop();
  848. // dispatch only last sentence and make sure it hasn't been dispatched before
  849. if (
  850. messageContentParts.length > 0 &&
  851. messageContentParts[messageContentParts.length - 1] !== message.lastSentence
  852. ) {
  853. message.lastSentence = messageContentParts[messageContentParts.length - 1];
  854. eventTarget.dispatchEvent(
  855. new CustomEvent('chat', {
  856. detail: {
  857. id: message.id,
  858. content: messageContentParts[messageContentParts.length - 1]
  859. }
  860. })
  861. );
  862. }
  863. }
  864. }
  865. if (selectedModelId) {
  866. message.selectedModelId = selectedModelId;
  867. message.arena = true;
  868. }
  869. if (usage) {
  870. message.usage = usage;
  871. }
  872. if (done) {
  873. message.done = true;
  874. if ($settings.notificationEnabled && !document.hasFocus()) {
  875. new Notification(`${message.model}`, {
  876. body: message.content,
  877. icon: `${WEBUI_BASE_URL}/static/favicon.png`
  878. });
  879. }
  880. if ($settings.responseAutoCopy) {
  881. copyToClipboard(message.content);
  882. }
  883. if ($settings.responseAutoPlayback && !$showCallOverlay) {
  884. await tick();
  885. document.getElementById(`speak-button-${message.id}`)?.click();
  886. }
  887. // Emit chat event for TTS
  888. let lastMessageContentPart =
  889. getMessageContentParts(message.content, $config?.audio?.tts?.split_on ?? 'punctuation')?.at(
  890. -1
  891. ) ?? '';
  892. if (lastMessageContentPart) {
  893. eventTarget.dispatchEvent(
  894. new CustomEvent('chat', {
  895. detail: { id: message.id, content: lastMessageContentPart }
  896. })
  897. );
  898. }
  899. eventTarget.dispatchEvent(
  900. new CustomEvent('chat:finish', {
  901. detail: {
  902. id: message.id,
  903. content: message.content
  904. }
  905. })
  906. );
  907. history.messages[message.id] = message;
  908. await chatCompletedHandler(chatId, message.model, message.id, createMessagesList(message.id));
  909. }
  910. history.messages[message.id] = message;
  911. console.log(data);
  912. if (autoScroll) {
  913. scrollToBottom();
  914. }
  915. };
  916. //////////////////////////
  917. // Chat functions
  918. //////////////////////////
  919. const submitPrompt = async (userPrompt, { _raw = false } = {}) => {
  920. console.log('submitPrompt', userPrompt, $chatId);
  921. const messages = createMessagesList(history.currentId);
  922. const _selectedModels = selectedModels.map((modelId) =>
  923. $models.map((m) => m.id).includes(modelId) ? modelId : ''
  924. );
  925. if (JSON.stringify(selectedModels) !== JSON.stringify(_selectedModels)) {
  926. selectedModels = _selectedModels;
  927. }
  928. if (userPrompt === '') {
  929. toast.error($i18n.t('Please enter a prompt'));
  930. return;
  931. }
  932. if (selectedModels.includes('')) {
  933. toast.error($i18n.t('Model not selected'));
  934. return;
  935. }
  936. if (messages.length != 0 && messages.at(-1).done != true) {
  937. // Response not done
  938. return;
  939. }
  940. if (messages.length != 0 && messages.at(-1).error) {
  941. // Error in response
  942. toast.error($i18n.t(`Oops! There was an error in the previous response.`));
  943. return;
  944. }
  945. if (
  946. files.length > 0 &&
  947. files.filter((file) => file.type !== 'image' && file.status === 'uploading').length > 0
  948. ) {
  949. toast.error(
  950. $i18n.t(`Oops! There are files still uploading. Please wait for the upload to complete.`)
  951. );
  952. return;
  953. }
  954. if (
  955. ($config?.file?.max_count ?? null) !== null &&
  956. files.length + chatFiles.length > $config?.file?.max_count
  957. ) {
  958. toast.error(
  959. $i18n.t(`You can only chat with a maximum of {{maxCount}} file(s) at a time.`, {
  960. maxCount: $config?.file?.max_count
  961. })
  962. );
  963. return;
  964. }
  965. prompt = '';
  966. await tick();
  967. // Reset chat input textarea
  968. const chatInputElement = document.getElementById('chat-input');
  969. if (chatInputElement) {
  970. chatInputElement.style.height = '';
  971. }
  972. const _files = JSON.parse(JSON.stringify(files));
  973. chatFiles.push(..._files.filter((item) => ['doc', 'file', 'collection'].includes(item.type)));
  974. chatFiles = chatFiles.filter(
  975. // Remove duplicates
  976. (item, index, array) =>
  977. array.findIndex((i) => JSON.stringify(i) === JSON.stringify(item)) === index
  978. );
  979. files = [];
  980. prompt = '';
  981. // Create user message
  982. let userMessageId = uuidv4();
  983. let userMessage = {
  984. id: userMessageId,
  985. parentId: messages.length !== 0 ? messages.at(-1).id : null,
  986. childrenIds: [],
  987. role: 'user',
  988. content: userPrompt,
  989. files: _files.length > 0 ? _files : undefined,
  990. timestamp: Math.floor(Date.now() / 1000), // Unix epoch
  991. models: selectedModels
  992. };
  993. // Add message to history and Set currentId to messageId
  994. history.messages[userMessageId] = userMessage;
  995. history.currentId = userMessageId;
  996. // Append messageId to childrenIds of parent message
  997. if (messages.length !== 0) {
  998. history.messages[messages.at(-1).id].childrenIds.push(userMessageId);
  999. }
  1000. // Wait until history/message have been updated
  1001. await tick();
  1002. // focus on chat input
  1003. const chatInput = document.getElementById('chat-input');
  1004. chatInput?.focus();
  1005. saveSessionSelectedModels();
  1006. await sendPrompt(userPrompt, userMessageId, { newChat: true });
  1007. };
  1008. const sendPrompt = async (
  1009. prompt: string,
  1010. parentId: string,
  1011. { modelId = null, modelIdx = null, newChat = false } = {}
  1012. ) => {
  1013. // Create new chat if newChat is true and first user message
  1014. if (
  1015. newChat &&
  1016. history.messages[history.currentId].parentId === null &&
  1017. history.messages[history.currentId].role === 'user'
  1018. ) {
  1019. await initChatHandler();
  1020. } else {
  1021. await saveChatHandler($chatId);
  1022. }
  1023. // If modelId is provided, use it, else use selected model
  1024. let selectedModelIds = modelId
  1025. ? [modelId]
  1026. : atSelectedModel !== undefined
  1027. ? [atSelectedModel.id]
  1028. : selectedModels;
  1029. // Create response messages for each selected model
  1030. const responseMessageIds: Record<PropertyKey, string> = {};
  1031. for (const [_modelIdx, modelId] of selectedModelIds.entries()) {
  1032. const model = $models.filter((m) => m.id === modelId).at(0);
  1033. if (model) {
  1034. let responseMessageId = uuidv4();
  1035. let responseMessage = {
  1036. parentId: parentId,
  1037. id: responseMessageId,
  1038. childrenIds: [],
  1039. role: 'assistant',
  1040. content: '',
  1041. model: model.id,
  1042. modelName: model.name ?? model.id,
  1043. modelIdx: modelIdx ? modelIdx : _modelIdx,
  1044. userContext: null,
  1045. timestamp: Math.floor(Date.now() / 1000) // Unix epoch
  1046. };
  1047. // Add message to history and Set currentId to messageId
  1048. history.messages[responseMessageId] = responseMessage;
  1049. history.currentId = responseMessageId;
  1050. // Append messageId to childrenIds of parent message
  1051. if (parentId !== null) {
  1052. history.messages[parentId].childrenIds = [
  1053. ...history.messages[parentId].childrenIds,
  1054. responseMessageId
  1055. ];
  1056. }
  1057. responseMessageIds[`${modelId}-${modelIdx ? modelIdx : _modelIdx}`] = responseMessageId;
  1058. }
  1059. }
  1060. await tick();
  1061. // Save chat after all messages have been created
  1062. await saveChatHandler($chatId);
  1063. const _chatId = JSON.parse(JSON.stringify($chatId));
  1064. await Promise.all(
  1065. selectedModelIds.map(async (modelId, _modelIdx) => {
  1066. console.log('modelId', modelId);
  1067. const model = $models.filter((m) => m.id === modelId).at(0);
  1068. if (model) {
  1069. const messages = createMessagesList(parentId);
  1070. // If there are image files, check if model is vision capable
  1071. const hasImages = messages.some((message) =>
  1072. message.files?.some((file) => file.type === 'image')
  1073. );
  1074. if (hasImages && !(model.info?.meta?.capabilities?.vision ?? true)) {
  1075. toast.error(
  1076. $i18n.t('Model {{modelName}} is not vision capable', {
  1077. modelName: model.name ?? model.id
  1078. })
  1079. );
  1080. }
  1081. let responseMessageId =
  1082. responseMessageIds[`${modelId}-${modelIdx ? modelIdx : _modelIdx}`];
  1083. let responseMessage = history.messages[responseMessageId];
  1084. let userContext = null;
  1085. if ($settings?.memory ?? false) {
  1086. if (userContext === null) {
  1087. const res = await queryMemory(localStorage.token, prompt).catch((error) => {
  1088. toast.error(error);
  1089. return null;
  1090. });
  1091. if (res) {
  1092. if (res.documents[0].length > 0) {
  1093. userContext = res.documents[0].reduce((acc, doc, index) => {
  1094. const createdAtTimestamp = res.metadatas[0][index].created_at;
  1095. const createdAtDate = new Date(createdAtTimestamp * 1000)
  1096. .toISOString()
  1097. .split('T')[0];
  1098. return `${acc}${index + 1}. [${createdAtDate}]. ${doc}\n`;
  1099. }, '');
  1100. }
  1101. console.log(userContext);
  1102. }
  1103. }
  1104. }
  1105. responseMessage.userContext = userContext;
  1106. const chatEventEmitter = await getChatEventEmitter(model.id, _chatId);
  1107. scrollToBottom();
  1108. if (webSearchEnabled) {
  1109. await getWebSearchResults(model.id, parentId, responseMessageId);
  1110. }
  1111. await sendPromptSocket(model, responseMessageId, _chatId);
  1112. if (chatEventEmitter) clearInterval(chatEventEmitter);
  1113. } else {
  1114. toast.error($i18n.t(`Model {{modelId}} not found`, { modelId }));
  1115. }
  1116. })
  1117. );
  1118. currentChatPage.set(1);
  1119. chats.set(await getChatList(localStorage.token, $currentChatPage));
  1120. };
  1121. const sendPromptSocket = async (model, responseMessageId, _chatId) => {
  1122. const responseMessage = history.messages[responseMessageId];
  1123. const userMessage = history.messages[responseMessage.parentId];
  1124. let files = JSON.parse(JSON.stringify(chatFiles));
  1125. if (model?.info?.meta?.knowledge ?? false) {
  1126. // Only initialize and add status if knowledge exists
  1127. responseMessage.statusHistory = [
  1128. {
  1129. action: 'knowledge_search',
  1130. description: $i18n.t(`Searching Knowledge for "{{searchQuery}}"`, {
  1131. searchQuery: userMessage.content
  1132. }),
  1133. done: false
  1134. }
  1135. ];
  1136. files.push(
  1137. ...model.info.meta.knowledge.map((item) => {
  1138. if (item?.collection_name) {
  1139. return {
  1140. id: item.collection_name,
  1141. name: item.name,
  1142. legacy: true
  1143. };
  1144. } else if (item?.collection_names) {
  1145. return {
  1146. name: item.name,
  1147. type: 'collection',
  1148. collection_names: item.collection_names,
  1149. legacy: true
  1150. };
  1151. } else {
  1152. return item;
  1153. }
  1154. })
  1155. );
  1156. history.messages[responseMessageId] = responseMessage;
  1157. }
  1158. files.push(
  1159. ...(userMessage?.files ?? []).filter((item) =>
  1160. ['doc', 'file', 'collection'].includes(item.type)
  1161. ),
  1162. ...(responseMessage?.files ?? []).filter((item) => ['web_search_results'].includes(item.type))
  1163. );
  1164. // Remove duplicates
  1165. files = files.filter(
  1166. (item, index, array) =>
  1167. array.findIndex((i) => JSON.stringify(i) === JSON.stringify(item)) === index
  1168. );
  1169. scrollToBottom();
  1170. eventTarget.dispatchEvent(
  1171. new CustomEvent('chat:start', {
  1172. detail: {
  1173. id: responseMessageId
  1174. }
  1175. })
  1176. );
  1177. await tick();
  1178. const stream =
  1179. model?.info?.params?.stream_response ??
  1180. $settings?.params?.stream_response ??
  1181. params?.stream_response ??
  1182. true;
  1183. const messages = [
  1184. params?.system || $settings.system || (responseMessage?.userContext ?? null)
  1185. ? {
  1186. role: 'system',
  1187. content: `${promptTemplate(
  1188. params?.system ?? $settings?.system ?? '',
  1189. $user.name,
  1190. $settings?.userLocation
  1191. ? await getAndUpdateUserLocation(localStorage.token)
  1192. : undefined
  1193. )}${
  1194. (responseMessage?.userContext ?? null)
  1195. ? `\n\nUser Context:\n${responseMessage?.userContext ?? ''}`
  1196. : ''
  1197. }`
  1198. }
  1199. : undefined,
  1200. ...createMessagesList(responseMessageId)
  1201. ]
  1202. .filter((message) => message?.content?.trim())
  1203. .map((message, idx, arr) => ({
  1204. role: message.role,
  1205. ...((message.files?.filter((file) => file.type === 'image').length > 0 ?? false) &&
  1206. message.role === 'user'
  1207. ? {
  1208. content: [
  1209. {
  1210. type: 'text',
  1211. text: message?.merged?.content ?? message.content
  1212. },
  1213. ...message.files
  1214. .filter((file) => file.type === 'image')
  1215. .map((file) => ({
  1216. type: 'image_url',
  1217. image_url: {
  1218. url: file.url
  1219. }
  1220. }))
  1221. ]
  1222. }
  1223. : {
  1224. content: message?.merged?.content ?? message.content
  1225. })
  1226. }));
  1227. const res = await generateOpenAIChatCompletion(
  1228. localStorage.token,
  1229. {
  1230. stream: stream,
  1231. model: model.id,
  1232. messages: messages,
  1233. params: {
  1234. ...$settings?.params,
  1235. ...params,
  1236. format: $settings.requestFormat ?? undefined,
  1237. keep_alive: $settings.keepAlive ?? undefined,
  1238. stop:
  1239. (params?.stop ?? $settings?.params?.stop ?? undefined)
  1240. ? (params?.stop.split(',').map((token) => token.trim()) ?? $settings.params.stop).map(
  1241. (str) => decodeURIComponent(JSON.parse('"' + str.replace(/\"/g, '\\"') + '"'))
  1242. )
  1243. : undefined
  1244. },
  1245. tool_ids: selectedToolIds.length > 0 ? selectedToolIds : undefined,
  1246. files: files.length > 0 ? files : undefined,
  1247. session_id: $socket?.id,
  1248. chat_id: $chatId,
  1249. id: responseMessageId,
  1250. ...(!$temporaryChatEnabled && messages.length == 1 && selectedModels[0] === model.id
  1251. ? {
  1252. background_tasks: {
  1253. title_generation: $settings?.title?.auto ?? true,
  1254. tags_generation: $settings?.autoTags ?? true
  1255. }
  1256. }
  1257. : {}),
  1258. ...(stream && (model.info?.meta?.capabilities?.usage ?? false)
  1259. ? {
  1260. stream_options: {
  1261. include_usage: true
  1262. }
  1263. }
  1264. : {})
  1265. },
  1266. `${WEBUI_BASE_URL}/api`
  1267. ).catch((error) => {
  1268. responseMessage.error = {
  1269. content: error
  1270. };
  1271. responseMessage.done = true;
  1272. return null;
  1273. });
  1274. console.log(res);
  1275. if (res) {
  1276. taskId = res.task_id;
  1277. }
  1278. await tick();
  1279. scrollToBottom();
  1280. };
  1281. const handleOpenAIError = async (error, responseMessage) => {
  1282. let errorMessage = '';
  1283. let innerError;
  1284. if (error) {
  1285. innerError = error;
  1286. }
  1287. console.error(innerError);
  1288. if ('detail' in innerError) {
  1289. toast.error(innerError.detail);
  1290. errorMessage = innerError.detail;
  1291. } else if ('error' in innerError) {
  1292. if ('message' in innerError.error) {
  1293. toast.error(innerError.error.message);
  1294. errorMessage = innerError.error.message;
  1295. } else {
  1296. toast.error(innerError.error);
  1297. errorMessage = innerError.error;
  1298. }
  1299. } else if ('message' in innerError) {
  1300. toast.error(innerError.message);
  1301. errorMessage = innerError.message;
  1302. }
  1303. responseMessage.error = {
  1304. content: $i18n.t(`Uh-oh! There was an issue with the response.`) + '\n' + errorMessage
  1305. };
  1306. responseMessage.done = true;
  1307. if (responseMessage.statusHistory) {
  1308. responseMessage.statusHistory = responseMessage.statusHistory.filter(
  1309. (status) => status.action !== 'knowledge_search'
  1310. );
  1311. }
  1312. history.messages[responseMessage.id] = responseMessage;
  1313. };
  1314. const stopResponse = () => {
  1315. if (taskId) {
  1316. const res = stopTask(localStorage.token, taskId).catch((error) => {
  1317. return null;
  1318. });
  1319. if (res) {
  1320. taskId = null;
  1321. }
  1322. }
  1323. };
  1324. const submitMessage = async (parentId, prompt) => {
  1325. let userPrompt = prompt;
  1326. let userMessageId = uuidv4();
  1327. let userMessage = {
  1328. id: userMessageId,
  1329. parentId: parentId,
  1330. childrenIds: [],
  1331. role: 'user',
  1332. content: userPrompt,
  1333. models: selectedModels
  1334. };
  1335. if (parentId !== null) {
  1336. history.messages[parentId].childrenIds = [
  1337. ...history.messages[parentId].childrenIds,
  1338. userMessageId
  1339. ];
  1340. }
  1341. history.messages[userMessageId] = userMessage;
  1342. history.currentId = userMessageId;
  1343. await tick();
  1344. await sendPrompt(userPrompt, userMessageId);
  1345. };
  1346. const regenerateResponse = async (message) => {
  1347. console.log('regenerateResponse');
  1348. if (history.currentId) {
  1349. let userMessage = history.messages[message.parentId];
  1350. let userPrompt = userMessage.content;
  1351. if ((userMessage?.models ?? [...selectedModels]).length == 1) {
  1352. // If user message has only one model selected, sendPrompt automatically selects it for regeneration
  1353. await sendPrompt(userPrompt, userMessage.id);
  1354. } else {
  1355. // If there are multiple models selected, use the model of the response message for regeneration
  1356. // e.g. many model chat
  1357. await sendPrompt(userPrompt, userMessage.id, {
  1358. modelId: message.model,
  1359. modelIdx: message.modelIdx
  1360. });
  1361. }
  1362. }
  1363. };
  1364. const continueResponse = async () => {
  1365. console.log('continueResponse');
  1366. const _chatId = JSON.parse(JSON.stringify($chatId));
  1367. if (history.currentId && history.messages[history.currentId].done == true) {
  1368. const responseMessage = history.messages[history.currentId];
  1369. responseMessage.done = false;
  1370. await tick();
  1371. const model = $models
  1372. .filter((m) => m.id === (responseMessage?.selectedModelId ?? responseMessage.model))
  1373. .at(0);
  1374. if (model) {
  1375. await sendPromptSocket(model, responseMessage.id, _chatId);
  1376. }
  1377. }
  1378. };
  1379. const mergeResponses = async (messageId, responses, _chatId) => {
  1380. console.log('mergeResponses', messageId, responses);
  1381. const message = history.messages[messageId];
  1382. const mergedResponse = {
  1383. status: true,
  1384. content: ''
  1385. };
  1386. message.merged = mergedResponse;
  1387. history.messages[messageId] = message;
  1388. try {
  1389. const [res, controller] = await generateMoACompletion(
  1390. localStorage.token,
  1391. message.model,
  1392. history.messages[message.parentId].content,
  1393. responses
  1394. );
  1395. if (res && res.ok && res.body) {
  1396. const textStream = await createOpenAITextStream(res.body, $settings.splitLargeChunks);
  1397. for await (const update of textStream) {
  1398. const { value, done, sources, error, usage } = update;
  1399. if (error || done) {
  1400. break;
  1401. }
  1402. if (mergedResponse.content == '' && value == '\n') {
  1403. continue;
  1404. } else {
  1405. mergedResponse.content += value;
  1406. history.messages[messageId] = message;
  1407. }
  1408. if (autoScroll) {
  1409. scrollToBottom();
  1410. }
  1411. }
  1412. await saveChatHandler(_chatId);
  1413. } else {
  1414. console.error(res);
  1415. }
  1416. } catch (e) {
  1417. console.error(e);
  1418. }
  1419. };
  1420. const getWebSearchResults = async (
  1421. model: string,
  1422. parentId: string,
  1423. responseMessageId: string
  1424. ) => {
  1425. // TODO: move this to the backend
  1426. const responseMessage = history.messages[responseMessageId];
  1427. const userMessage = history.messages[parentId];
  1428. const messages = createMessagesList(history.currentId);
  1429. responseMessage.statusHistory = [
  1430. {
  1431. done: false,
  1432. action: 'web_search',
  1433. description: $i18n.t('Generating search query')
  1434. }
  1435. ];
  1436. history.messages[responseMessageId] = responseMessage;
  1437. const prompt = userMessage.content;
  1438. let queries = await generateQueries(
  1439. localStorage.token,
  1440. model,
  1441. messages.filter((message) => message?.content?.trim()),
  1442. prompt
  1443. ).catch((error) => {
  1444. console.log(error);
  1445. return [prompt];
  1446. });
  1447. if (queries.length === 0) {
  1448. responseMessage.statusHistory.push({
  1449. done: true,
  1450. error: true,
  1451. action: 'web_search',
  1452. description: $i18n.t('No search query generated')
  1453. });
  1454. history.messages[responseMessageId] = responseMessage;
  1455. return;
  1456. }
  1457. const searchQuery = queries[0];
  1458. responseMessage.statusHistory.push({
  1459. done: false,
  1460. action: 'web_search',
  1461. description: $i18n.t(`Searching "{{searchQuery}}"`, { searchQuery })
  1462. });
  1463. history.messages[responseMessageId] = responseMessage;
  1464. const results = await processWebSearch(localStorage.token, searchQuery).catch((error) => {
  1465. console.log(error);
  1466. toast.error(error);
  1467. return null;
  1468. });
  1469. if (results) {
  1470. responseMessage.statusHistory.push({
  1471. done: true,
  1472. action: 'web_search',
  1473. description: $i18n.t('Searched {{count}} sites', { count: results.filenames.length }),
  1474. query: searchQuery,
  1475. urls: results.filenames
  1476. });
  1477. if (responseMessage?.files ?? undefined === undefined) {
  1478. responseMessage.files = [];
  1479. }
  1480. responseMessage.files.push({
  1481. collection_name: results.collection_name,
  1482. name: searchQuery,
  1483. type: 'web_search_results',
  1484. urls: results.filenames
  1485. });
  1486. history.messages[responseMessageId] = responseMessage;
  1487. } else {
  1488. responseMessage.statusHistory.push({
  1489. done: true,
  1490. error: true,
  1491. action: 'web_search',
  1492. description: 'No search results found'
  1493. });
  1494. history.messages[responseMessageId] = responseMessage;
  1495. }
  1496. };
  1497. const initChatHandler = async () => {
  1498. if (!$temporaryChatEnabled) {
  1499. chat = await createNewChat(localStorage.token, {
  1500. id: $chatId,
  1501. title: $i18n.t('New Chat'),
  1502. models: selectedModels,
  1503. system: $settings.system ?? undefined,
  1504. params: params,
  1505. history: history,
  1506. messages: createMessagesList(history.currentId),
  1507. tags: [],
  1508. timestamp: Date.now()
  1509. });
  1510. currentChatPage.set(1);
  1511. await chats.set(await getChatList(localStorage.token, $currentChatPage));
  1512. await chatId.set(chat.id);
  1513. window.history.replaceState(history.state, '', `/c/${chat.id}`);
  1514. } else {
  1515. await chatId.set('local');
  1516. }
  1517. await tick();
  1518. };
  1519. const saveChatHandler = async (_chatId) => {
  1520. if ($chatId == _chatId) {
  1521. if (!$temporaryChatEnabled) {
  1522. chat = await updateChatById(localStorage.token, _chatId, {
  1523. models: selectedModels,
  1524. history: history,
  1525. messages: createMessagesList(history.currentId),
  1526. params: params,
  1527. files: chatFiles
  1528. });
  1529. currentChatPage.set(1);
  1530. await chats.set(await getChatList(localStorage.token, $currentChatPage));
  1531. }
  1532. }
  1533. };
  1534. </script>
  1535. <svelte:head>
  1536. <title>
  1537. {$chatTitle
  1538. ? `${$chatTitle.length > 30 ? `${$chatTitle.slice(0, 30)}...` : $chatTitle} | ${$WEBUI_NAME}`
  1539. : `${$WEBUI_NAME}`}
  1540. </title>
  1541. </svelte:head>
  1542. <audio id="audioElement" src="" style="display: none;" />
  1543. <EventConfirmDialog
  1544. bind:show={showEventConfirmation}
  1545. title={eventConfirmationTitle}
  1546. message={eventConfirmationMessage}
  1547. input={eventConfirmationInput}
  1548. inputPlaceholder={eventConfirmationInputPlaceholder}
  1549. inputValue={eventConfirmationInputValue}
  1550. on:confirm={(e) => {
  1551. if (e.detail) {
  1552. eventCallback(e.detail);
  1553. } else {
  1554. eventCallback(true);
  1555. }
  1556. }}
  1557. on:cancel={() => {
  1558. eventCallback(false);
  1559. }}
  1560. />
  1561. {#if !chatIdProp || (loaded && chatIdProp)}
  1562. <div
  1563. class="h-screen max-h-[100dvh] {$showSidebar
  1564. ? 'md:max-w-[calc(100%-260px)]'
  1565. : ''} w-full max-w-full flex flex-col"
  1566. id="chat-container"
  1567. >
  1568. {#if $settings?.backgroundImageUrl ?? null}
  1569. <div
  1570. class="absolute {$showSidebar
  1571. ? 'md:max-w-[calc(100%-260px)] md:translate-x-[260px]'
  1572. : ''} top-0 left-0 w-full h-full bg-cover bg-center bg-no-repeat"
  1573. style="background-image: url({$settings.backgroundImageUrl}) "
  1574. />
  1575. <div
  1576. class="absolute top-0 left-0 w-full h-full bg-gradient-to-t from-white to-white/85 dark:from-gray-900 dark:to-[#171717]/90 z-0"
  1577. />
  1578. {/if}
  1579. <Navbar
  1580. bind:this={navbarElement}
  1581. chat={{
  1582. id: $chatId,
  1583. chat: {
  1584. title: $chatTitle,
  1585. models: selectedModels,
  1586. system: $settings.system ?? undefined,
  1587. params: params,
  1588. history: history,
  1589. timestamp: Date.now()
  1590. }
  1591. }}
  1592. title={$chatTitle}
  1593. bind:selectedModels
  1594. shareEnabled={!!history.currentId}
  1595. {initNewChat}
  1596. />
  1597. <PaneGroup direction="horizontal" class="w-full h-full">
  1598. <Pane defaultSize={50} class="h-full flex w-full relative">
  1599. {#if $banners.length > 0 && !history.currentId && !$chatId && selectedModels.length <= 1}
  1600. <div class="absolute top-12 left-0 right-0 w-full z-30">
  1601. <div class=" flex flex-col gap-1 w-full">
  1602. {#each $banners.filter( (b) => (b.dismissible ? !JSON.parse(localStorage.getItem('dismissedBannerIds') ?? '[]').includes(b.id) : true) ) as banner}
  1603. <Banner
  1604. {banner}
  1605. on:dismiss={(e) => {
  1606. const bannerId = e.detail;
  1607. localStorage.setItem(
  1608. 'dismissedBannerIds',
  1609. JSON.stringify(
  1610. [
  1611. bannerId,
  1612. ...JSON.parse(localStorage.getItem('dismissedBannerIds') ?? '[]')
  1613. ].filter((id) => $banners.find((b) => b.id === id))
  1614. )
  1615. );
  1616. }}
  1617. />
  1618. {/each}
  1619. </div>
  1620. </div>
  1621. {/if}
  1622. <div class="flex flex-col flex-auto z-10 w-full">
  1623. {#if $settings?.landingPageMode === 'chat' || createMessagesList(history.currentId).length > 0}
  1624. <div
  1625. class=" pb-2.5 flex flex-col justify-between w-full flex-auto overflow-auto h-0 max-w-full z-10 scrollbar-hidden"
  1626. id="messages-container"
  1627. bind:this={messagesContainerElement}
  1628. on:scroll={(e) => {
  1629. autoScroll =
  1630. messagesContainerElement.scrollHeight - messagesContainerElement.scrollTop <=
  1631. messagesContainerElement.clientHeight + 5;
  1632. }}
  1633. >
  1634. <div class=" h-full w-full flex flex-col">
  1635. <Messages
  1636. chatId={$chatId}
  1637. bind:history
  1638. bind:autoScroll
  1639. bind:prompt
  1640. {selectedModels}
  1641. {sendPrompt}
  1642. {showMessage}
  1643. {submitMessage}
  1644. {continueResponse}
  1645. {regenerateResponse}
  1646. {mergeResponses}
  1647. {chatActionHandler}
  1648. bottomPadding={files.length > 0}
  1649. />
  1650. </div>
  1651. </div>
  1652. <div class=" pb-[1rem]">
  1653. <MessageInput
  1654. {history}
  1655. {selectedModels}
  1656. bind:files
  1657. bind:prompt
  1658. bind:autoScroll
  1659. bind:selectedToolIds
  1660. bind:webSearchEnabled
  1661. bind:atSelectedModel
  1662. transparentBackground={$settings?.backgroundImageUrl ?? false}
  1663. {stopResponse}
  1664. {createMessagePair}
  1665. on:upload={async (e) => {
  1666. const { type, data } = e.detail;
  1667. if (type === 'web') {
  1668. await uploadWeb(data);
  1669. } else if (type === 'youtube') {
  1670. await uploadYoutubeTranscription(data);
  1671. } else if (type === 'google-drive') {
  1672. await uploadGoogleDriveFile(data);
  1673. }
  1674. }}
  1675. on:submit={async (e) => {
  1676. if (e.detail) {
  1677. await tick();
  1678. submitPrompt(
  1679. ($settings?.richTextInput ?? true)
  1680. ? e.detail.replaceAll('\n\n', '\n')
  1681. : e.detail
  1682. );
  1683. }
  1684. }}
  1685. />
  1686. <div
  1687. class="absolute bottom-1 text-xs text-gray-500 text-center line-clamp-1 right-0 left-0"
  1688. >
  1689. <!-- {$i18n.t('LLMs can make mistakes. Verify important information.')} -->
  1690. </div>
  1691. </div>
  1692. {:else}
  1693. <div class="overflow-auto w-full h-full flex items-center">
  1694. <Placeholder
  1695. {history}
  1696. {selectedModels}
  1697. bind:files
  1698. bind:prompt
  1699. bind:autoScroll
  1700. bind:selectedToolIds
  1701. bind:webSearchEnabled
  1702. bind:atSelectedModel
  1703. transparentBackground={$settings?.backgroundImageUrl ?? false}
  1704. {stopResponse}
  1705. {createMessagePair}
  1706. on:upload={async (e) => {
  1707. const { type, data } = e.detail;
  1708. if (type === 'web') {
  1709. await uploadWeb(data);
  1710. } else if (type === 'youtube') {
  1711. await uploadYoutubeTranscription(data);
  1712. }
  1713. }}
  1714. on:submit={async (e) => {
  1715. if (e.detail) {
  1716. await tick();
  1717. submitPrompt(
  1718. ($settings?.richTextInput ?? true)
  1719. ? e.detail.replaceAll('\n\n', '\n')
  1720. : e.detail
  1721. );
  1722. }
  1723. }}
  1724. />
  1725. </div>
  1726. {/if}
  1727. </div>
  1728. </Pane>
  1729. <ChatControls
  1730. bind:this={controlPaneComponent}
  1731. bind:history
  1732. bind:chatFiles
  1733. bind:params
  1734. bind:files
  1735. bind:pane={controlPane}
  1736. chatId={$chatId}
  1737. modelId={selectedModelIds?.at(0) ?? null}
  1738. models={selectedModelIds.reduce((a, e, i, arr) => {
  1739. const model = $models.find((m) => m.id === e);
  1740. if (model) {
  1741. return [...a, model];
  1742. }
  1743. return a;
  1744. }, [])}
  1745. {submitPrompt}
  1746. {stopResponse}
  1747. {showMessage}
  1748. {eventTarget}
  1749. />
  1750. </PaneGroup>
  1751. </div>
  1752. {/if}