pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refactor (#4303)

Browse Source 
### What problem does this PR solve?

### Type of change
- [x] Refactoring
This commit is contained in:
Kevin Hu
2024-12-31 14:31:31 +08:00
committed by GitHub
parent b52b0f68fc
commit 4ba4f622a5
6 changed files with 11 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
api/db/services/document_service.py
View File

@ -96,14 +96,11 @@ class DocumentService(CommonService):
def insert(cls, doc):
if not cls.save(**doc):
raise RuntimeError("Database error (Document)!")
e, doc = cls.get_by_id(doc["id"])
if not e:
raise RuntimeError("Database error (Document retrieval)!")
e, kb = KnowledgebaseService.get_by_id(doc.kb_id)
e, kb = KnowledgebaseService.get_by_id(doc["kb_id"])
if not KnowledgebaseService.update_by_id(
kb.id, {"doc_num": kb.doc_num + 1}):
raise RuntimeError("Database error (Knowledgebase)!")
return doc
return Document(**doc)
@classmethod
@DB.connection_context()