Ned Halksworth
86 lines
1.9 KiB
Python
86 lines
1.9 KiB
Python
"""Pydantic models for sFetch's API."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any, Literal
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class SearchResult(BaseModel):
|
|
id: int
|
|
url: str
|
|
title: str
|
|
snippet: str
|
|
indexed_at: str
|
|
|
|
|
|
class ImageResult(BaseModel):
|
|
id: int
|
|
url: str
|
|
page_url: str
|
|
alt_text: str
|
|
indexed_at: str
|
|
|
|
|
|
class VideoResult(BaseModel):
|
|
id: int
|
|
url: str
|
|
page_url: str
|
|
title: str
|
|
indexed_at: str
|
|
|
|
|
|
class SearchResponse(BaseModel):
|
|
query: str
|
|
type: str = "web"
|
|
total: int
|
|
results: list[SearchResult] | list[ImageResult] | list[VideoResult]
|
|
|
|
|
|
class CrawlRequest(BaseModel):
|
|
seed_urls: list[str] = Field(min_length=1)
|
|
max_depth: int = Field(default=2, ge=0, le=5)
|
|
max_pages_per_domain: int = Field(default=50, ge=1, le=500)
|
|
same_domain_only: bool = True
|
|
|
|
|
|
class AIMessage(BaseModel):
|
|
role: Literal["system", "user", "assistant", "tool"]
|
|
content: str = ""
|
|
thinking: str | None = None
|
|
tool_name: str | None = None
|
|
tool_calls: list[dict[str, Any]] | None = None
|
|
|
|
|
|
class AIChatRequest(BaseModel):
|
|
model: str | None = None
|
|
messages: list[AIMessage] = Field(min_length=1)
|
|
think: bool | str | None = None
|
|
use_web_search: bool = False
|
|
web_result_limit: int = Field(default=5, ge=1, le=10)
|
|
|
|
|
|
class AISearchRequest(BaseModel):
|
|
query: str = Field(min_length=1)
|
|
model: str | None = None
|
|
include_web: bool = True
|
|
local_result_limit: int = Field(default=5, ge=1, le=10)
|
|
web_result_limit: int = Field(default=5, ge=1, le=10)
|
|
think: bool | str | None = None
|
|
|
|
|
|
class AISource(BaseModel):
|
|
title: str
|
|
url: str
|
|
source_type: Literal["local", "web"]
|
|
content: str = ""
|
|
|
|
|
|
class AIAnswerResponse(BaseModel):
|
|
model: str
|
|
content: str
|
|
thinking: str | None = None
|
|
sources: list[AISource] = Field(default_factory=list)
|
|
configured: bool = True
|