Merge branch 'feat/document-segmentic' into deploy/dev

langgenius · Dec 28, 2023 · 582ca01 · 582ca01
2 parents ae9bce7 + 680d7b7
commit 582ca01
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/api/core/indexing_runner.py b/api/core/indexing_runner.py
@@ -297,13 +297,14 @@ def file_indexing_estimate(self, tenant_id: str, file_details: List[UploadFile],
                     "qa_preview": document_qa_list,
                     "preview": preview_texts
                 }
-        embedding_model_type_instance = cast(TextEmbeddingModel, embedding_model_instance.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 = cast(TextEmbeddingModel, embedding_model_instance.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
+            )
         return {
             "total_segments": total_segments,
             "tokens": tokens,