import uuid
from typing import List, Optional

from pydantic import BaseModel


class UploadKBResponse(BaseModel):
    kb_id: uuid.UUID
    name: str
    chunks_stored: int


class UploadKBRequest(BaseModel):
    name: str
    description: Optional[str] = None


class TokenizeRequest(BaseModel):
    text: str


class TokenizeResponse(BaseModel):
    input_ids: List[int]
    attention_mask: List[int]


class SemanticSearchRequest(BaseModel):
    embedding: List[float]
    top_k: Optional[int] = 3


class SemanticSearchResult(BaseModel):
    chunk_id: str
    kb_id: str
    text: str
    image_url: str | None = None 
    score: float

class ManualTextRequest(BaseModel):
    kb_id: uuid.UUID
    text: str