Fix cost reporting in search too

2 years ago · a3d4e76c8a
parent 8cb547c948
commit a3d4e76c8a
1 changed files with 2 additions and 2 deletions
--- a/models/search_model.py
+++ b/models/search_model.py
@ -491,8 +491,8 @@ class Search:
            embedding_model.last_token_usage, embeddings=True
        )
        price += await self.usage_service.get_price(
-            llm_predictor.last_token_usage
+            llm_predictor.last_token_usage, chatgpt=True
-        ) + await self.usage_service.get_price(embedding_model.last_token_usage, True)
+        ) + await self.usage_service.get_price(embedding_model.last_token_usage, embeddings=True)
        if ctx:
            await self.try_edit(