8ae1e64987
Co-authored-by: Charles Packer <packercharles@gmail.com> Co-authored-by: Shubham Naik <shubham.naik10@gmail.com> Co-authored-by: Shubham Naik <shub@memgpt.ai>
12 lines
357 B
Python
12 lines
357 B
Python
from typing import List, Literal
|
|
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class EmbeddingResponse(BaseModel):
|
|
"""OpenAI embedding response model: https://platform.openai.com/docs/api-reference/embeddings/object"""
|
|
|
|
index: int # the index of the embedding in the list of embeddings
|
|
embedding: List[float]
|
|
object: Literal["embedding"] = "embedding"
|