Feat/delete single dataset retrival (#6570)

This commit is contained in:
Jyong 2024-07-24 12:50:11 +08:00 • committed by GitHub
commit e4bb943fe5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
22 changed files with 651 additions and 115 deletions

View file

@ -0,0 +1,23 @@
from pydantic import BaseModel
class VectorSetting(BaseModel):
vector_weight: float
embedding_provider_name: str
embedding_model_name: str
class KeywordSetting(BaseModel):
keyword_weight: float
class Weights(BaseModel):
"""Model for weighted rerank."""
weight_type: str
vector_setting: VectorSetting
keyword_setting: KeywordSetting