INNER CODE UNIT · Python

RetrieveRequest

pathwaycom/llm-app · serviette/server/main.py:49

class RetrieveRequest(BaseModel):
    query: str
    k: int = Field(default=5, ge=1)


class RetrieveResult(BaseModel):
    text: str
    metadata: dict[str, Any] = Field(default_factory=dict)
    score: float


class RetrieveResponse(BaseModel):
    results: list[RetrieveResult]


class DocumentEntry(BaseModel):
    id: str
    name: str

View source record →