fix cost reporting for query

1 year ago · 8cb547c948
parent 1a511dfe50
commit 8cb547c948
1 changed files with 2 additions and 2 deletions
--- 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,
                )