From 065ba14709ab4fdb03455e630b59ee74a89bdfe6 Mon Sep 17 00:00:00 2001 From: mspronesti Date: Thu, 5 Oct 2023 00:06:24 +0200 Subject: [PATCH] fix ruff lint --- pandasai/helpers/openai_info.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pandasai/helpers/openai_info.py b/pandasai/helpers/openai_info.py index 66cb1cfed..581f6a97f 100644 --- a/pandasai/helpers/openai_info.py +++ b/pandasai/helpers/openai_info.py @@ -102,7 +102,9 @@ def __call__(self, response: OpenAIObject) -> None: model_name = response.model if model_name in MODEL_COST_PER_1K_TOKENS: - prompt_cost = get_openai_token_cost_for_model(model_name, usage.prompt_tokens) + prompt_cost = get_openai_token_cost_for_model( + model_name, usage.prompt_tokens + ) completion_cost = get_openai_token_cost_for_model( model_name, usage.completion_tokens, is_completion=True )