mirror of
https://github.com/open-webui/open-webui
synced 2025-06-23 02:16:52 +00:00
Feature: added indivdiual rag config update function
This commit is contained in:
parent
171c18cff8
commit
1ae3873c55
@ -219,12 +219,22 @@ class KnowledgeTable:
|
|||||||
except Exception:
|
except Exception:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
def get_knowledge_by_collection_name(self, name: str) -> Optional[KnowledgeModel]:
|
def update_rag_config_by_id(
|
||||||
|
self, id: str, rag_config: dict
|
||||||
|
) -> Optional[KnowledgeModel]:
|
||||||
try:
|
try:
|
||||||
with get_db() as db:
|
with get_db() as db:
|
||||||
knowledge = db.query(Knowledge).filter_by(name=name).first()
|
knowledge = self.get_knowledge_by_id(id=id)
|
||||||
return KnowledgeModel.model_validate(knowledge) if knowledge else None
|
db.query(Knowledge).filter_by(id=id).update(
|
||||||
except Exception:
|
{
|
||||||
|
"rag_config": rag_config,
|
||||||
|
"updated_at": int(time.time()),
|
||||||
|
}
|
||||||
|
)
|
||||||
|
db.commit()
|
||||||
|
return self.get_knowledge_by_id(id=id)
|
||||||
|
except Exception as e:
|
||||||
|
log.exception(e)
|
||||||
return None
|
return None
|
||||||
|
|
||||||
Knowledges = KnowledgeTable()
|
Knowledges = KnowledgeTable()
|
||||||
|
Loading…
Reference in New Issue
Block a user