From 3a0a9e2d8f7b0bae4a1e157f981ec35754511a28 Mon Sep 17 00:00:00 2001
From: takatost <takatost@users.noreply.github.com>
Date: Sat, 19 Aug 2023 21:31:40 +0800
Subject: [PATCH] fix: embedding get price definition missing (#922)

---
 api/core/indexing_runner.py                       | 2 +-
 api/core/model_providers/models/embedding/base.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

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.