estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix: embedding get price definition missing (#922)

Browse Source
This commit is contained in:
takatost
2023-08-19 21:31:40 +08:00
committed by GitHub
parent 0a0d63457d
commit 3a0a9e2d8f
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
api/core/indexing_runner.py
View File

@@ -379,7 +379,7 @@ class IndexingRunner:
return { return {
"total_segments": total_segments, "total_segments": total_segments,
"tokens": tokens, "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(), "currency": embedding_model.get_currency(),
"preview": preview_texts "preview": preview_texts
} }

2
api/core/model_providers/models/embedding/base.py
View File

@@ -50,7 +50,7 @@ class BaseEmbedding(BaseProviderModel):
logger.debug(f"model: {self.name} price_config: {self._price_config}") logger.debug(f"model: {self.name} price_config: {self._price_config}")
return 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. calc tokens total price.