62cb5f1bac
### What problem does this PR solve? ### Type of change - [x] New Feature (non-breaking change which adds functionality) --------- Co-authored-by: Kevin Hu <kevinhu.sh@gmail.com>
51 lines
1.6 KiB
Python
51 lines
1.6 KiB
Python
from .base import Base
|
|
|
|
|
|
class Chunk(Base):
|
|
def __init__(self, rag, res_dict):
|
|
self.id = ""
|
|
self.content = ""
|
|
self.important_keywords = []
|
|
self.create_time = ""
|
|
self.create_timestamp_flt = 0.0
|
|
self.knowledgebase_id = None
|
|
self.document_name = ""
|
|
self.document_id = ""
|
|
self.status = "1"
|
|
for k in list(res_dict.keys()):
|
|
if k not in self.__dict__:
|
|
res_dict.pop(k)
|
|
super().__init__(rag, res_dict)
|
|
|
|
def delete(self) -> bool:
|
|
"""
|
|
Delete the chunk in the document.
|
|
"""
|
|
res = self.post('/doc/chunk/rm',
|
|
{"doc_id": self.document_id, 'chunk_ids': [self.id]})
|
|
res = res.json()
|
|
if res.get("retmsg") == "success":
|
|
return True
|
|
raise Exception(res["retmsg"])
|
|
|
|
def save(self) -> bool:
|
|
"""
|
|
Save the document details to the server.
|
|
"""
|
|
res = self.post('/doc/chunk/set',
|
|
{"chunk_id": self.id,
|
|
"kb_id": self.knowledgebase_id,
|
|
"name": self.document_name,
|
|
"content_with_weight": self.content,
|
|
"important_kwd": self.important_keywords,
|
|
"create_time": self.create_time,
|
|
"create_timestamp_flt": self.create_timestamp_flt,
|
|
"doc_id": self.document_id,
|
|
"status": self.status,
|
|
})
|
|
res = res.json()
|
|
if res.get("retmsg") == "success":
|
|
return True
|
|
raise Exception(res["retmsg"])
|
|
|