schema.py

from dataclasses import dataclass
from typing import List


@dataclass
class EmbeddingResponse:
    embedding: List[float]


@dataclass
class SimilarityResponse:
    score: float