mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-07-17 05:07:23 +08:00
Refactor: reformat all code for lefthook using ruff and gofmt (#16585)
This commit is contained in:
@@ -16,6 +16,7 @@
|
||||
|
||||
from .base import Base
|
||||
|
||||
|
||||
class ChunkUpdateError(Exception):
|
||||
def __init__(self, code=None, message=None, details=None):
|
||||
self.code = code
|
||||
@@ -23,6 +24,7 @@ class ChunkUpdateError(Exception):
|
||||
self.details = details
|
||||
super().__init__(message)
|
||||
|
||||
|
||||
class Chunk(Base):
|
||||
def __init__(self, rag, res_dict):
|
||||
self.id = ""
|
||||
@@ -48,17 +50,12 @@ class Chunk(Base):
|
||||
res_dict.pop(k)
|
||||
super().__init__(rag, res_dict)
|
||||
|
||||
#for backward compatibility
|
||||
# for backward compatibility
|
||||
if not self.document_name:
|
||||
self.document_name = self.document_keyword
|
||||
|
||||
|
||||
def update(self, update_message: dict):
|
||||
res = self.patch(f"/datasets/{self.dataset_id}/documents/{self.document_id}/chunks/{self.id}", update_message)
|
||||
res = res.json()
|
||||
if res.get("code") != 0:
|
||||
raise ChunkUpdateError(
|
||||
code=res.get("code"),
|
||||
message=res.get("message"),
|
||||
details=res.get("details")
|
||||
)
|
||||
raise ChunkUpdateError(code=res.get("code"), message=res.get("message"), details=res.get("details"))
|
||||
|
||||
Reference in New Issue
Block a user