From 1fdad50daceac81bba4c289b23b065c9104de2a3 Mon Sep 17 00:00:00 2001
From: Zhichang Yu <yuzhichang@gmail.com>
Date: Wed, 18 Dec 2024 14:42:33 +0800
Subject: [PATCH] Fix raptor (#4089)

### 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 | 4 ++--
 api/db/services/task_service.py     | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/api/db/services/document_service.py b/api/db/services/document_service.py
index edbae80cf..ba146fa4f 100644
--- a/api/db/services/document_service.py
+++ b/api/db/services/document_service.py
@@ -444,8 +444,8 @@ def queue_raptor_tasks(doc):
         return {
             "id": get_uuid(),
             "doc_id": doc["id"],
-            "from_page": 0,
-            "to_page": -1,
+            "from_page": 100000000,
+            "to_page": 100000000,
             "progress_msg": "Start to do RAPTOR (Recursive Abstractive Processing for Tree-Organized Retrieval)."
         }
 
diff --git a/api/db/services/task_service.py b/api/db/services/task_service.py
index aa9ae8ac7..e03c12623 100644
--- a/api/db/services/task_service.py
+++ b/api/db/services/task_service.py
@@ -202,7 +202,7 @@ class TaskService(CommonService):
 
 def queue_tasks(doc: dict, bucket: str, name: str):
     def new_task():
-        return {"id": get_uuid(), "doc_id": doc["id"], "progress": 0.0}
+        return {"id": get_uuid(), "doc_id": doc["id"], "progress": 0.0, "from_page": 0, "to_page": 100000000}
 
     tsks = []