zhichyu commited on
Commit
b371a08
·
1 Parent(s): 9958ac6

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": 0,
448
- "to_page": -1,
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