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

Function: Added get knowledge by collection name and adjusted insert new knowledge to insert rag config to DB

Browse Source
This commit is contained in:
Maytown 2025-05-02 09:16:40 +02:00
parent b46da1ae86
commit 45ee9c33ad
1 changed files with 21 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -99,20 +99,24 @@ class KnowledgeForm(BaseModel):
access_control: Optional[dict] = None access_control: Optional[dict] = None
class RAGConfigForm(BaseModel):
rag_config: Optional[dict] = None
class KnowledgeTable: class KnowledgeTable:
def insert_new_knowledge( def insert_new_knowledge(
self, user_id: str, form_data: KnowledgeForm self, user_id: str, form_data: KnowledgeForm, rag_data: RAGConfigForm
) -> Optional[KnowledgeModel]: ) -> Optional[KnowledgeModel]:
with get_db() as db: with get_db() as db:
knowledge = KnowledgeModel( knowledge_data = {
**{
**form_data.model_dump(), **form_data.model_dump(),
"data": {"rag_config": rag_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()),
"updated_at": int(time.time()), "updated_at": int(time.time()),
} }
) knowledge = KnowledgeModel(**knowledge_data)
try: try:
result = Knowledge(**knowledge.model_dump()) result = Knowledge(**knowledge.model_dump())
@ -217,5 +221,12 @@ class KnowledgeTable:
except Exception: except Exception:
return False return False
def get_knowledge_by_collection_name(self, name: str) -> Optional[KnowledgeModel]:
try:
with get_db() as db:
knowledge = db.query(Knowledge).filter_by(name=name).first()
return KnowledgeModel.model_validate(knowledge) if knowledge else None
except Exception:
return None
Knowledges = KnowledgeTable() Knowledges = KnowledgeTable()