|
@@ -274,6 +274,8 @@ class IndexingRunner:
|
|
|
tokens = 0
|
|
|
preview_texts = []
|
|
|
total_segments = 0
|
|
|
+ total_price = 0
|
|
|
+ currency = 'USD'
|
|
|
for file_detail in file_details:
|
|
|
|
|
|
processing_rule = DatasetProcessRule(
|
|
@@ -344,11 +346,13 @@ class IndexingRunner:
|
|
|
price_type=PriceType.INPUT,
|
|
|
tokens=tokens
|
|
|
)
|
|
|
+ total_price = '{:f}'.format(embedding_price_info.total_amount)
|
|
|
+ currency = embedding_price_info.currency
|
|
|
return {
|
|
|
"total_segments": total_segments,
|
|
|
"tokens": tokens,
|
|
|
- "total_price": '{:f}'.format(embedding_price_info.total_amount) if embedding_model_instance else 0,
|
|
|
- "currency": embedding_price_info.currency if embedding_model_instance else 'USD',
|
|
|
+ "total_price": total_price,
|
|
|
+ "currency": currency,
|
|
|
"preview": preview_texts
|
|
|
}
|
|
|
|
|
@@ -388,6 +392,8 @@ class IndexingRunner:
|
|
|
tokens = 0
|
|
|
preview_texts = []
|
|
|
total_segments = 0
|
|
|
+ total_price = 0
|
|
|
+ currency = 'USD'
|
|
|
for notion_info in notion_info_list:
|
|
|
workspace_id = notion_info['workspace_id']
|
|
|
data_source_binding = DataSourceBinding.query.filter(
|
|
@@ -470,20 +476,22 @@ class IndexingRunner:
|
|
|
"qa_preview": document_qa_list,
|
|
|
"preview": preview_texts
|
|
|
}
|
|
|
-
|
|
|
- embedding_model_type_instance = embedding_model_instance.model_type_instance
|
|
|
- embedding_model_type_instance = cast(TextEmbeddingModel, embedding_model_type_instance)
|
|
|
- embedding_price_info = embedding_model_type_instance.get_price(
|
|
|
- model=embedding_model_instance.model,
|
|
|
- credentials=embedding_model_instance.credentials,
|
|
|
- price_type=PriceType.INPUT,
|
|
|
- tokens=tokens
|
|
|
- )
|
|
|
+ if embedding_model_instance:
|
|
|
+ embedding_model_type_instance = embedding_model_instance.model_type_instance
|
|
|
+ embedding_model_type_instance = cast(TextEmbeddingModel, embedding_model_type_instance)
|
|
|
+ embedding_price_info = embedding_model_type_instance.get_price(
|
|
|
+ model=embedding_model_instance.model,
|
|
|
+ credentials=embedding_model_instance.credentials,
|
|
|
+ price_type=PriceType.INPUT,
|
|
|
+ tokens=tokens
|
|
|
+ )
|
|
|
+ total_price = '{:f}'.format(embedding_price_info.total_amount)
|
|
|
+ currency = embedding_price_info.currency
|
|
|
return {
|
|
|
"total_segments": total_segments,
|
|
|
"tokens": tokens,
|
|
|
- "total_price": '{:f}'.format(embedding_price_info.total_amount) if embedding_model_instance else 0,
|
|
|
- "currency": embedding_price_info.currency if embedding_model_instance else 'USD',
|
|
|
+ "total_price": total_price,
|
|
|
+ "currency": currency,
|
|
|
"preview": preview_texts
|
|
|
}
|
|
|
|