faiss server
This commit is contained in:
10
faiss/models.py
Normal file
10
faiss/models.py
Normal file
@@ -0,0 +1,10 @@
|
||||
# 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
|
||||
Reference in New Issue
Block a user