mirror of
https://github.com/danswer-ai/danswer.git
synced 2025-06-20 21:10:58 +02:00
31 lines
572 B
Python
31 lines
572 B
Python
from pydantic import BaseModel
|
|
|
|
|
|
class EmbedRequest(BaseModel):
|
|
# This already includes any prefixes, the text is just passed directly to the model
|
|
texts: list[str]
|
|
model_name: str
|
|
max_context_length: int
|
|
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]
|