23 lines
504 B
Python
23 lines
504 B
Python
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]
|
|
|
|
class CrawlResult(BaseModel):
|
|
url: str
|
|
title: Optional[str] = None
|
|
content: Optional[str] = None
|
|
embedding: Optional[List[float]] = None
|
|
|
|
class SaveResultsRequest(BaseModel):
|
|
task_id: int
|
|
results: List[CrawlResult] |