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

enh: add/remove file from knowledge

Browse Source
This commit is contained in:
Timothy J. Baek
2024-10-03 06:46:20 -07:00
parent 1c01e52f7c
commit 78413d0c2e
2 changed files with 135 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
backend/open_webui/apps/webui/models/knowledge.py
View File

@@ -126,28 +126,10 @@ class KnowledgeTable:
) -> Optional[KnowledgeModel]:
try:
with get_db() as db:
knowledge = self.get_knowledge_by_id(id=id)
db.query(Knowledge).filter_by(id=id).update(
{
**({"name": form_data.name} if form_data.name else {}),
**(
{"description": form_data.description}
if form_data.description
else {}
),
**(
{
"data": (
form_data.data
if overwrite
else {
**(self.get_knowledge_by_id(id=id)).data,
**form_data.data,
}
)
}
if form_data.data
else {}
),
**form_data.model_dump(exclude_none=True),
"updated_at": int(time.time()),
}
)