mirror of
https://github.com/danswer-ai/danswer.git
synced 2025-05-13 21:30:21 +02:00
29 lines
456 B
Python
29 lines
456 B
Python
from pydantic import BaseModel
|
|
|
|
|
|
class EmbedRequest(BaseModel):
|
|
texts: list[str]
|
|
model_name: str
|
|
normalize_embeddings: bool
|
|
|
|
|
|
class EmbedResponse(BaseModel):
|
|
embeddings: list[list[float]]
|
|
|
|
|
|
class RerankRequest(BaseModel):
|
|
query: str
|
|
documents: list[str]
|
|
|
|
|
|
class RerankResponse(BaseModel):
|
|
scores: list[list[float]]
|
|
|
|
|
|
class IntentRequest(BaseModel):
|
|
query: str
|
|
|
|
|
|
class IntentResponse(BaseModel):
|
|
class_probs: list[float]
|