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

Fix: adjusted knowledgetable to add rag config as individual column

Browse Source
This commit is contained in:
Maytown 2025-05-12 12:47:22 +02:00
parent 5a2de12f4b
commit 0b9ed1ee42
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
backend/open_webui/models/knowledge.py
View File

@ -35,6 +35,7 @@ class Knowledge(Base):
data = Column(JSON, nullable=True) data = Column(JSON, nullable=True)
meta = Column(JSON, nullable=True) meta = Column(JSON, nullable=True)
rag_config = Column(JSON, nullable=True) # Configuration for RAG (Retrieval-Augmented Generation) model.
access_control = Column(JSON, nullable=True) # Controls data access levels. access_control = Column(JSON, nullable=True) # Controls data access levels.
# Defines access control rules for this entry. # Defines access control rules for this entry.
@ -68,6 +69,7 @@ class KnowledgeModel(BaseModel):
data: Optional[dict] = None data: Optional[dict] = None
meta: Optional[dict] = None meta: Optional[dict] = None
rag_config: Optional[dict] = None # Configuration for RAG (Retrieval-Augmented Generation) model.
access_control: Optional[dict] = None access_control: Optional[dict] = None
@ -97,7 +99,7 @@ class KnowledgeForm(BaseModel):
description: str description: str
data: Optional[dict] = None data: Optional[dict] = None
access_control: Optional[dict] = None access_control: Optional[dict] = None
rag_config: Optional[dict] = None rag_config: Optional[dict] = {}
class KnowledgeTable: class KnowledgeTable:
@ -107,7 +109,6 @@ class KnowledgeTable:
with get_db() as db: with get_db() as db:
knowledge_data = { knowledge_data = {
**form_data.model_dump(), **form_data.model_dump(),
"data": {"rag_config": form_data.rag_config},
"id": str(uuid.uuid4()), "id": str(uuid.uuid4()),
"user_id": user_id, "user_id": user_id,
"created_at": int(time.time()), "created_at": int(time.time()),