OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-26 18:26:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy J. Baek
2024-05-27 12:48:08 -07:00
parent 73178cf519
commit 4685f523b6
7 changed files with 45 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/lib/components/chat/Chat.svelte
View File

@@ -48,8 +48,7 @@
import { runWebSearch } from '$lib/apis/rag';
import Banner from '../common/Banner.svelte';
import { getUserSettings } from '$lib/apis/users';
const i18n: Writable<i18nType> = getContext('i18n');
export let chatIdProp = '';
@@ -408,7 +407,7 @@
responseMessage.userContext = userContext;
if (useWebSearch) {
await runWebSearchForPrompt(model.id, parentId, responseMessageId);
await getWebSearchResultsAsFiles(model.id, parentId, responseMessageId);
}
if (model?.owned_by === 'openai') {
@@ -425,10 +424,15 @@
await chats.set(await getChatList(localStorage.token));
};
const runWebSearchForPrompt = async (model: string, parentId: string, responseId: string) => {
const getWebSearchResultsAsFiles = async (
model: string,
parentId: string,
responseId: string
) => {
const responseMessage = history.messages[responseId];
responseMessage.progress = $i18n.t('Generating search query');
messages = messages;
const searchQuery = await generateChatSearchQuery(model, parentId);
if (!searchQuery) {
toast.warning($i18n.t('No search query generated'));
@@ -436,8 +440,10 @@
messages = messages;
return;
}
responseMessage.progress = $i18n.t("Searching the web for '{{searchQuery}}'", { searchQuery });
messages = messages;
const searchDocument = await runWebSearch(localStorage.token, searchQuery);
if (searchDocument === undefined) {
toast.warning($i18n.t('No search results found'));
@@ -445,9 +451,11 @@
messages = messages;
return;
}
if (!responseMessage.files) {
responseMessage.files = [];
}
responseMessage.files.push({
collection_name: searchDocument.collection_name,
name: searchQuery,
@@ -1157,6 +1165,6 @@
{messages}
{submitPrompt}
{stopResponse}
webSearchAvailable={$config.enable_websearch ?? false}
webSearchAvailable={$config?.features.enable_web_search ?? false}
/>
{/if}

49
src/lib/components/chat/MessageInput.svelte
View File

@@ -46,7 +46,6 @@
export let files = [];
export let fileUploadEnabled = true;
export let speechRecognitionEnabled = true;
export let webSearchAvailable = false;
@@ -779,37 +778,33 @@
{/if}
<div class=" flex">
{#if fileUploadEnabled}
<div class=" self-end mb-2 ml-1">
<Tooltip content={$i18n.t('Upload files')}>
<button
class="bg-gray-50 hover:bg-gray-100 text-gray-800 dark:bg-gray-850 dark:text-white dark:hover:bg-gray-800 transition rounded-full p-1.5"
type="button"
on:click={() => {
filesInputElement.click();
}}
<div class=" ml-1 flex items-center">
<Tooltip content={$i18n.t('Upload files')}>
<button
class="bg-gray-50 hover:bg-gray-100 text-gray-800 dark:bg-gray-850 dark:text-white dark:hover:bg-gray-800 transition rounded-full p-1.5"
type="button"
on:click={() => {
filesInputElement.click();
}}
>
<svg
xmlns="http://www.w3.org/2000/svg"
viewBox="0 0 16 16"
fill="currentColor"
class="size-5"
>
<svg
xmlns="http://www.w3.org/2000/svg"
viewBox="0 0 16 16"
fill="currentColor"
class="w-[1.2rem] h-[1.2rem]"
>
<path
d="M8.75 3.75a.75.75 0 0 0-1.5 0v3.5h-3.5a.75.75 0 0 0 0 1.5h3.5v3.5a.75.75 0 0 0 1.5 0v-3.5h3.5a.75.75 0 0 0 0-1.5h-3.5v-3.5Z"
/>
</svg>
</button>
</Tooltip>
</div>
{/if}
<path
d="M8.75 3.75a.75.75 0 0 0-1.5 0v3.5h-3.5a.75.75 0 0 0 0 1.5h3.5v3.5a.75.75 0 0 0 1.5 0v-3.5h3.5a.75.75 0 0 0 0-1.5h-3.5v-3.5Z"
/>
</svg>
</button>
</Tooltip>
</div>
<textarea
id="chat-textarea"
bind:this={chatTextAreaElement}
class="scrollbar-hidden bg-gray-50 dark:bg-gray-850 dark:text-gray-100 outline-none w-full py-3 px-3 {fileUploadEnabled
? ''
: ' pl-4'} rounded-xl resize-none h-[48px]"
class="scrollbar-hidden bg-gray-50 dark:bg-gray-850 dark:text-gray-100 outline-none w-full py-3 px-3 rounded-xl resize-none h-[48px]"
placeholder={chatInputPlaceholder !== ''
? chatInputPlaceholder
: isRecording