AI/open-webui
1
0
Fork 1
mirror of https://git.mirrors.martin98.com/https://github.com/open-webui/open-webui synced 2025-08-19 20:29:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy J. Baek 2024-10-01 13:13:39 -07:00
parent 5c9dd25459
commit a0fb4a9b84
2 changed files with 13 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
backend/open_webui/apps/retrieval/main.py
View File

@ -729,13 +729,9 @@ def process_file(
text_content = " ".join([doc.page_content for doc in docs]) text_content = " ".join([doc.page_content for doc in docs])
log.debug(f"text_content: {text_content}") log.debug(f"text_content: {text_content}")
Files.update_files_metadata_by_id( Files.update_files_data_by_id(
form_data.file_id, form_data.file_id,
{ {"content": text_content},
"content": {
"text": text_content,
}
},
) )
try: try:

11
backend/open_webui/apps/webui/models/files.py
View File

@ -112,6 +112,17 @@ class FilesTable:
for file in db.query(File).filter_by(user_id=user_id).all() for file in db.query(File).filter_by(user_id=user_id).all()
] ]
def update_files_data_by_id(self, id: str, data: dict) -> Optional[FileModel]:
with get_db() as db:
try:
file = db.query(File).filter_by(id=id).first()
file.data = {**file.data, **data}
db.commit()
return FileModel.model_validate(file)
except Exception:
return None
def update_files_metadata_by_id(self, id: str, meta: dict) -> Optional[FileModel]: def update_files_metadata_by_id(self, id: str, meta: dict) -> Optional[FileModel]:
with get_db() as db: with get_db() as db:
try: try: