Chat.svelte 55 KB

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