schemas.py

from typing import *
from pydantic import BaseModel

class SearchRequest(BaseModel):
    keyword: str
    threshold: int

class SearchResponse(BaseModel):
    results: List[Dict[str, Any]]

class ChatRequest(BaseModel):
    messages: List[Dict[str, str]]
    model: str

class ChatResponse(BaseModel):
    response: str