mirror of
https://github.com/onyx-dot-app/onyx.git
synced 2026-10-23 03:42:59 -05: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]
|