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
@ -218,13 +218,23 @@ class KnowledgeTable:
|
||||
return True
|
||||
except Exception:
|
||||
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:
|
||||
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:
|
||||
knowledge = self.get_knowledge_by_id(id=id)
|
||||
db.query(Knowledge).filter_by(id=id).update(
|
||||
{
|
||||
"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
|
||||
|
||||
Knowledges = KnowledgeTable()
|
||||
|
Loading…
Reference in New Issue
Block a user