Fix raptor (#4089)
Browse files### What problem does this PR solve?
Fix raptor
### Type of change
- [x] Bug Fix (non-breaking change which fixes an issue)
api/db/services/document_service.py
CHANGED
@@ -444,8 +444,8 @@ def queue_raptor_tasks(doc):
|
|
444 |
return {
|
445 |
"id": get_uuid(),
|
446 |
"doc_id": doc["id"],
|
447 |
-
"from_page":
|
448 |
-
"to_page":
|
449 |
"progress_msg": "Start to do RAPTOR (Recursive Abstractive Processing for Tree-Organized Retrieval)."
|
450 |
}
|
451 |
|
|
|
444 |
return {
|
445 |
"id": get_uuid(),
|
446 |
"doc_id": doc["id"],
|
447 |
+
"from_page": 100000000,
|
448 |
+
"to_page": 100000000,
|
449 |
"progress_msg": "Start to do RAPTOR (Recursive Abstractive Processing for Tree-Organized Retrieval)."
|
450 |
}
|
451 |
|
api/db/services/task_service.py
CHANGED
@@ -202,7 +202,7 @@ class TaskService(CommonService):
|
|
202 |
|
203 |
def queue_tasks(doc: dict, bucket: str, name: str):
|
204 |
def new_task():
|
205 |
-
return {"id": get_uuid(), "doc_id": doc["id"], "progress": 0.0}
|
206 |
|
207 |
tsks = []
|
208 |
|
|
|
202 |
|
203 |
def queue_tasks(doc: dict, bucket: str, name: str):
|
204 |
def new_task():
|
205 |
+
return {"id": get_uuid(), "doc_id": doc["id"], "progress": 0.0, "from_page": 0, "to_page": 100000000}
|
206 |
|
207 |
tsks = []
|
208 |
|