From d1342c59917acdc633938d1ff4e36a95638e7b23 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 6 Sep 2024 16:05:19 -0700 Subject: [PATCH] fix RerankResponse type --- litellm/rerank_api/types.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/litellm/rerank_api/types.py b/litellm/rerank_api/types.py index 605e25a2ecb..3b95597c310 100644 --- a/litellm/rerank_api/types.py +++ b/litellm/rerank_api/types.py @@ -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__