mirror of
https://github.com/infiniflow/ragflow.git
synced 2025-12-08 20:42:30 +08:00
change chunk.status to chunk.available (#2646)
### What problem does this PR solve? #1102 ### Type of change - [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
committed by
GitHub
parent
e82e8fde13
commit
c103dd2746
@ -609,8 +609,8 @@ def set(tenant_id):
|
||||
d["content_sm_ltks"] = rag_tokenizer.fine_grained_tokenize(d["content_ltks"])
|
||||
d["important_kwd"] = req["important_keywords"]
|
||||
d["important_tks"] = rag_tokenizer.tokenize(" ".join(req["important_keywords"]))
|
||||
if "available_int" in req:
|
||||
d["available_int"] = req["available_int"]
|
||||
if "available" in req:
|
||||
d["available_int"] = req["available"]
|
||||
|
||||
try:
|
||||
tenant_id = DocumentService.get_tenant_id(req["document_id"])
|
||||
|
||||
@ -11,7 +11,7 @@ class Chunk(Base):
|
||||
self.knowledgebase_id = None
|
||||
self.document_name = ""
|
||||
self.document_id = ""
|
||||
self.status = "1"
|
||||
self.available = 1
|
||||
for k in list(res_dict.keys()):
|
||||
if k not in self.__dict__:
|
||||
res_dict.pop(k)
|
||||
@ -39,7 +39,7 @@ class Chunk(Base):
|
||||
"content": self.content,
|
||||
"important_keywords": self.important_keywords,
|
||||
"document_id": self.document_id,
|
||||
"status": self.status,
|
||||
"available": self.available,
|
||||
})
|
||||
res = res.json()
|
||||
if res.get("retmsg") == "success":
|
||||
|
||||
@ -151,14 +151,12 @@ class TestDocument(TestSdk):
|
||||
name3 = 'westworld.pdf'
|
||||
path = 'test_data/westworld.pdf'
|
||||
|
||||
|
||||
# Create a document in the dataset using the file path
|
||||
rag.create_document(ds, name=name3, blob=open(path, "rb").read())
|
||||
|
||||
# Retrieve the document by name
|
||||
doc = rag.get_document(name="westworld.pdf")
|
||||
|
||||
|
||||
# Initiate asynchronous parsing
|
||||
doc.async_parse()
|
||||
|
||||
@ -279,7 +277,17 @@ class TestDocument(TestSdk):
|
||||
assert isinstance(chunk, Chunk), "Chunk was not added to chunk list"
|
||||
chunk.content = "ragflow123"
|
||||
res = chunk.save()
|
||||
assert res is True, f"Failed to update chunk, error: {res}"
|
||||
assert res is True, f"Failed to update chunk content, error: {res}"
|
||||
|
||||
def test_update_chunk_available(self):
|
||||
rag = RAGFlow(API_KEY, HOST_ADDRESS)
|
||||
doc = rag.get_document(name='story.txt')
|
||||
chunk = doc.add_chunk(content="ragflow")
|
||||
assert chunk is not None, "Chunk is None"
|
||||
assert isinstance(chunk, Chunk), "Chunk was not added to chunk list"
|
||||
chunk.available = 0
|
||||
res = chunk.save()
|
||||
assert res is True, f"Failed to update chunk status, error: {res}"
|
||||
|
||||
def test_retrieval_chunks(self):
|
||||
rag = RAGFlow(API_KEY, HOST_ADDRESS)
|
||||
|
||||
Reference in New Issue
Block a user