diff --git a/api/core/indexing_runner.py b/api/core/indexing_runner.py index 2f47f52705..ef6e9001c7 100644 --- a/api/core/indexing_runner.py +++ b/api/core/indexing_runner.py @@ -379,7 +379,7 @@ class IndexingRunner: return { "total_segments": total_segments, "tokens": tokens, - "total_price": '{:f}'.format(embedding_model.get_token_price(tokens)), + "total_price": '{:f}'.format(embedding_model.calc_tokens_price(tokens)), "currency": embedding_model.get_currency(), "preview": preview_texts } diff --git a/api/core/model_providers/models/embedding/base.py b/api/core/model_providers/models/embedding/base.py index 1cbe56a9f3..25f30894b0 100644 --- a/api/core/model_providers/models/embedding/base.py +++ b/api/core/model_providers/models/embedding/base.py @@ -50,7 +50,7 @@ class BaseEmbedding(BaseProviderModel): logger.debug(f"model: {self.name} price_config: {self._price_config}") return self._price_config - def calc_tokens_price(self, tokens:int) -> decimal.Decimal: + def calc_tokens_price(self, tokens: int) -> decimal.Decimal: """ calc tokens total price.