Prechádzať zdrojové kódy

打标查询默认limit设置为1

yuchengwei 2 mesiacov pred
rodič
commit
866bf2ae2c
1 zmenil súbory, kde vykonal 1 pridanie a 2 odobranie
  1. 1 2
      router/text_search.py

+ 1 - 2
router/text_search.py

@@ -11,7 +11,6 @@ router = APIRouter(prefix="/text", tags=["Text Search"])
 
 class TextSearchRequest(BaseModel):
     text: str
-    limit: int = 1
 
 @router.post("/search", response_model=StandardResponse)
 async def search_text(request: TextSearchRequest):
@@ -30,7 +29,7 @@ async def search_text(request: TextSearchRequest):
         for sentence in sentences:
             search_results = trunks_service.search_by_vector(
                 text=sentence,
-                limit=request.limit
+                limit=1
             )
             
             # 处理搜索结果