10 lines
200 B
Python
10 lines
200 B
Python
# models.py
|
|
from pydantic import BaseModel
|
|
from typing import List
|
|
|
|
class EmbeddingInput(BaseModel):
|
|
embedding: List[float]
|
|
|
|
class SearchInput(BaseModel):
|
|
embedding: List[float]
|
|
k: int = 5 |