mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-16 23:41:43 +00:00
fix RerankResponse type
This commit is contained in:
parent
e095daf2e4
commit
d1342c5991
1 changed files with 13 additions and 0 deletions
|
|
@ -23,3 +23,16 @@ class RerankResponse(BaseModel):
|
|||
id: str
|
||||
results: List[dict] # Contains index and relevance_score
|
||||
meta: dict # Contains api_version and billed_units
|
||||
_hidden_params: Optional[dict] = {}
|
||||
|
||||
class Config:
|
||||
underscore_attrs_are_private = True
|
||||
|
||||
def __getitem__(self, key):
|
||||
return self.__dict__[key]
|
||||
|
||||
def get(self, key, default=None):
|
||||
return self.__dict__.get(key, default)
|
||||
|
||||
def __contains__(self, key):
|
||||
return key in self.__dict__
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue