From 8cb547c9485a5d00e99b678c93735161767901aa Mon Sep 17 00:00:00 2001 From: Rene Teigen Date: Tue, 7 Mar 2023 21:17:40 +0000 Subject: [PATCH] fix cost reporting for query --- models/index_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/models/index_model.py b/models/index_model.py index 2cb8170..8f8e89d 100644 --- a/models/index_model.py +++ b/models/index_model.py @@ -940,9 +940,9 @@ class Index_handler: try: total_price = round( - await self.usage_service.get_price(llm_predictor.last_token_usage) + await self.usage_service.get_price(llm_predictor.last_token_usage, chatgpt=True) + await self.usage_service.get_price( - embedding_model.last_token_usage, True + embedding_model.last_token_usage, embeddings=True ), 6, )