2025-12-20 17:08:54 +08:00
|
|
|
from pydantic import BaseModel
|
|
|
|
|
from typing import List, Optional
|
|
|
|
|
|
|
|
|
|
class RegisterRequest(BaseModel):
|
|
|
|
|
url: str
|
|
|
|
|
|
|
|
|
|
class PendingRequest(BaseModel):
|
|
|
|
|
task_id: int
|
|
|
|
|
limit: Optional[int] = 10
|
|
|
|
|
|
|
|
|
|
class AddUrlsRequest(BaseModel):
|
|
|
|
|
task_id: int
|
|
|
|
|
urls: List[str]
|
|
|
|
|
|
2025-12-22 22:08:51 +08:00
|
|
|
# schemas.py
|
2025-12-20 17:08:54 +08:00
|
|
|
class CrawlResult(BaseModel):
|
2025-12-22 22:08:51 +08:00
|
|
|
source_url: str
|
|
|
|
|
chunk_index: int # 新增字段
|
2025-12-20 17:08:54 +08:00
|
|
|
title: Optional[str] = None
|
|
|
|
|
content: Optional[str] = None
|
|
|
|
|
embedding: Optional[List[float]] = None
|
|
|
|
|
|
|
|
|
|
class SaveResultsRequest(BaseModel):
|
|
|
|
|
task_id: int
|
|
|
|
|
results: List[CrawlResult]
|
