소스 검색

fix: embedding get price definition missing (#922)

tags/0.3.15
takatost 2 년 전
부모
커밋
3a0a9e2d8f
No account linked to committer's email address
2개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 1
    1
      api/core/indexing_runner.py
  2. 1
    1
      api/core/model_providers/models/embedding/base.py

+ 1
- 1
api/core/indexing_runner.py 파일 보기

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
} }

+ 1
- 1
api/core/model_providers/models/embedding/base.py 파일 보기

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.



Loading…
취소
저장