Merge pull request #212 from Hikari-Haru/cost-fix

Cost fix

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,
                 )

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
-        ) + await self.usage_service.get_price(embedding_model.last_token_usage, True)
+            llm_predictor.last_token_usage, chatgpt=True
+        ) + await self.usage_service.get_price(embedding_model.last_token_usage, embeddings=True)
 
         if ctx:
             await self.try_edit(