UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-08 01:30:40 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

feat: inject search result doc in the response, not the query

Browse Source 
this is to handle when we have multiple models selected or regenerate a
response, it'll only add it to the model's response and not add dupes on
the user message
This commit is contained in:
Jun Siang Cheah 2024-05-12 20:44:46 +08:00
parent 44c8b0bb83
commit f49e1afaa6
3 changed files with 96 additions and 55 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
src/lib/components/chat/Messages/ResponseMessage.svelte
View File

@ -369,8 +369,9 @@
class="prose chat-{message.role} w-full max-w-full dark:prose-invert prose-headings:my-0 prose-p:m-0 prose-p:-mb-6 prose-pre:my-0 prose-table:my-0 prose-blockquote:my-0 prose-img:my-0 prose-ul:-my-4 prose-ol:-my-4 prose-li:-my-3 prose-ul:-mb-6 prose-ol:-mb-8 prose-ol:p-0 prose-li:-mb-4 whitespace-pre-line"
>
<div>
{#if message.progress}
{#if message.progress || message.files}
<div class="my-2.5 w-full flex overflow-x-auto gap-2 flex-wrap">
{#if message.progress}
<div>
<button
class="h-16 flex items-center space-x-3 px-2.5 dark:bg-gray-600 rounded-xl border border-gray-200 dark:border-none text-left"
@ -422,6 +423,48 @@
</div>
</button>
</div>
{/if}
{#if message.files}
{#each message.files as file}
<div>
{#if file.type === 'websearch'}
<button
class="h-16 w-[15rem] flex items-center space-x-3 px-2.5 dark:bg-gray-600 rounded-xl border border-gray-200 dark:border-none text-left"
type="button"
>
<div class="p-2.5 bg-red-400 text-white rounded-lg">
<svg
xmlns="http://www.w3.org/2000/svg"
viewBox="0 0 24 24"
fill="currentColor"
class="w-6 h-6"
>
<path
d="M11.625 16.5a1.875 1.875 0 1 0 0-3.75 1.875 1.875 0 0 0 0 3.75Z"
/>
<path
fill-rule="evenodd"
d="M5.625 1.5H9a3.75 3.75 0 0 1 3.75 3.75v1.875c0 1.036.84 1.875 1.875 1.875H16.5a3.75 3.75 0 0 1 3.75 3.75v7.875c0 1.035-.84 1.875-1.875 1.875H5.625a1.875 1.875 0 0 1-1.875-1.875V3.375c0-1.036.84-1.875 1.875-1.875Zm6 16.5c.66 0 1.277-.19 1.797-.518l1.048 1.048a.75.75 0 0 0 1.06-1.06l-1.047-1.048A3.375 3.375 0 1 0 11.625 18Z"
clip-rule="evenodd"
/>
<path
d="M14.25 5.25a5.23 5.23 0 0 0-1.279-3.434 9.768 9.768 0 0 1 6.963 6.963A5.23 5.23 0 0 0 16.5 7.5h-1.875a.375.375 0 0 1-.375-.375V5.25Z"
/>
</svg>
</div>
<div class="flex flex-col justify-center -space-y-0.5">
<div class=" dark:text-gray-100 text-sm font-medium line-clamp-1">
{file.name}
</div>
<div class=" text-gray-500 text-sm">{$i18n.t('Search Results')}</div>
</div>
</button>
{/if}
</div>
{/each}
{/if}
</div>
{/if}

7
src/routes/(app)/+page.svelte
View File

@ -327,11 +327,10 @@
messages = messages;
return;
}
const parentMessage = history.messages[parentId];
if (!parentMessage.files) {
parentMessage.files = [];
if (!responseMessage.files) {
responseMessage.files = [];
}
parentMessage.files.push({
responseMessage.files.push({
collection_name: searchDocument!.collection_name,
name: searchQuery,
type: 'websearch',

7
src/routes/(app)/c/[id]/+page.svelte
View File

@ -332,11 +332,10 @@
messages = messages;
return;
}
const parentMessage = history.messages[parentId];
if (!parentMessage.files) {
parentMessage.files = [];
if (!responseMessage.files) {
responseMessage.files = [];
}
parentMessage.files.push({
responseMessage.files.push({
collection_name: searchDocument!.collection_name,
name: searchQuery,
type: 'websearch',