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-09-29 23:08:55 +02:00
parent 677c36c3aa
commit 6afc686e17
6 changed files with 22 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
backend/open_webui/apps/retrieval/utils.py
View File

@ -319,7 +319,7 @@ def get_rag_context(
for file in files: for file in files:
if file.get("context") == "full": if file.get("context") == "full":
context = { context = {
"documents": [[file["content"]]], "documents": [[file.get("file").get("content")]],
"metadatas": [[{"file_id": file.get("id"), "name": file.get("name")}]], "metadatas": [[{"file_id": file.get("id"), "name": file.get("name")}]],
} }
else: else:

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

@ -150,7 +150,10 @@
if (res) { if (res) {
fileItem.status = 'processed'; fileItem.status = 'processed';
fileItem.collection_name = res.collection_name; fileItem.collection_name = res.collection_name;
fileItem.content = res.content; fileItem.file = {
...fileItem.file,
content: res.content
};
files = files; files = files;
} }

12
src/lib/components/chat/MessageInput/Commands.svelte
View File

@ -46,14 +46,17 @@
if (res) { if (res) {
fileItem.status = 'processed'; fileItem.status = 'processed';
fileItem.collection_name = res.collection_name; fileItem.collection_name = res.collection_name;
fileItem.content = res.content; fileItem.file = {
content: res.content,
...fileItem.file
};
files = files; files = files;
} }
} catch (e) { } catch (e) {
// Remove the failed doc from the files array // Remove the failed doc from the files array
files = files.filter((f) => f.name !== url); files = files.filter((f) => f.name !== url);
toast.error(e); toast.error(JSON.stringify(e));
} }
}; };
@ -76,7 +79,10 @@
if (res) { if (res) {
fileItem.status = 'processed'; fileItem.status = 'processed';
fileItem.collection_name = res.collection_name; fileItem.collection_name = res.collection_name;
fileItem.content = res.content; fileItem.file = {
content: res.content,
...fileItem.file
};
files = files; files = files;
} }
} catch (e) { } catch (e) {

2
src/lib/components/common/FileItem.svelte
View File

@ -33,7 +33,7 @@
class="h-14 {className} flex items-center space-x-3 {colorClassName} rounded-xl border border-gray-100 dark:border-gray-800 text-left" class="h-14 {className} flex items-center space-x-3 {colorClassName} rounded-xl border border-gray-100 dark:border-gray-800 text-left"
type="button" type="button"
on:click={async () => { on:click={async () => {
if (file.content) { if (file?.file?.content) {
showModal = !showModal; showModal = !showModal;
} else { } else {
if (url) { if (url) {

8
src/lib/components/common/FileItemModal.svelte
View File

@ -61,8 +61,10 @@
• •
{/if} {/if}
{#if file.content} {#if file?.file?.content}
<div class="capitalize shrink-0">{getLineCount(file.content)} extracted lines</div> <div class="capitalize shrink-0">
{getLineCount(file?.file?.content ?? '')} extracted lines
</div>
<div class="flex items-center gap-1 shrink-0"> <div class="flex items-center gap-1 shrink-0">
<Info /> <Info />
@ -100,7 +102,7 @@
</div> </div>
<div class="max-h-96 overflow-scroll scrollbar-hidden text-xs whitespace-pre-wrap"> <div class="max-h-96 overflow-scroll scrollbar-hidden text-xs whitespace-pre-wrap">
{file?.content ?? 'No content'} {file?.file?.content ?? 'No content'}
</div> </div>
</div> </div>
</Modal> </Modal>

3
src/lib/utils/index.ts
View File

@ -889,5 +889,6 @@ export const formatFileSize = (size) => {
}; };
export const getLineCount = (text) => { export const getLineCount = (text) => {
return text.split('\n').length; console.log(typeof text);
return text ? text.split('\n').length : 0;
}; };