fix: embedding get price definition missing (#922)

3 years ago · 3a0a9e2d8f
parent 0a0d63457d
commit 3a0a9e2d8f
2 changed files with 2 additions and 2 deletions
--- 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
        }