From 67917dcf8bff2e52c3af8dd165bdecac0a80ee7d Mon Sep 17 00:00:00 2001 From: Adnane Guettaf Date: Fri, 12 Jul 2024 09:43:49 +0200 Subject: [PATCH] [fix] remove calculate tokens for some depricated endpoints --- edenai_apis/apis/google/google_helpers.py | 2 +- edenai_apis/apis/google/google_image_api.py | 6 +++--- edenai_apis/apis/google/google_text_api.py | 1 - 3 files changed, 4 insertions(+), 5 deletions(-) diff --git a/edenai_apis/apis/google/google_helpers.py b/edenai_apis/apis/google/google_helpers.py index 16987d78..c577f8e3 100644 --- a/edenai_apis/apis/google/google_helpers.py +++ b/edenai_apis/apis/google/google_helpers.py @@ -621,7 +621,7 @@ def gemini_request(payload: dict, model: str, api_key: str): message=original_response["error"]["message"], code=response.status_code, ) - calculate_usage_tokens(original_response=original_response) + # calculate_usage_tokens(original_response=original_response) return original_response diff --git a/edenai_apis/apis/google/google_image_api.py b/edenai_apis/apis/google/google_image_api.py index 95d63e63..e0f03d22 100644 --- a/edenai_apis/apis/google/google_image_api.py +++ b/edenai_apis/apis/google/google_image_api.py @@ -478,7 +478,7 @@ def _gemini_pro_vision_qa( message="Internal Server Error", code=500, ) from exc - calculate_usage_tokens(original_response=original_response) + # calculate_usage_tokens(original_response=original_response) answer = original_response["candidates"][0]["content"]["parts"][0]["text"] standardized_response = QuestionAnswerDataClass( @@ -570,8 +570,8 @@ def image__embeddings( raise ProviderException(message="No predictions found", code=400) items: Sequence[EmbeddingDataClass] = [] - - for prediction in original_response["predictions"] : + + for prediction in original_response["predictions"]: embedding = prediction.get("imageEmbedding") or [] items.append(EmbeddingDataClass(embedding=embedding)) diff --git a/edenai_apis/apis/google/google_text_api.py b/edenai_apis/apis/google/google_text_api.py index 2add2522..60945211 100644 --- a/edenai_apis/apis/google/google_text_api.py +++ b/edenai_apis/apis/google/google_text_api.py @@ -593,7 +593,6 @@ def _handle_non_streaming( raise ProviderException( "Provider did not return a valid JSON", code=response.status_code ) from exc - calculate_usage_tokens(original_response) generated_text = original_response["candidates"][0]["content"]["parts"][0][ "text" ]