embeddings-RestAPI / schema /pydantic_schema.py
Chittrarasu's picture
deploy
21d405f
raw
history blame contribute delete
424 Bytes
from pydantic import BaseModel
from typing import List
# Request schema
class SMSRequest(BaseModel):
messages: List[str]
from pydantic import BaseModel
from typing import List
# Response schema for embedding vectors
class EmbeddingResponse(BaseModel):
message: str
embedding: List[float]
# Final response containing all embeddings
class EmbeddingsResponse(BaseModel):
embeddings: List[EmbeddingResponse]