From a938e1f184f4a2bf4ede032111cfb29f6989683d Mon Sep 17 00:00:00 2001 From: takatost Date: Thu, 4 Jan 2024 13:28:52 +0800 Subject: [PATCH] fix: notion_indexing_estimate embedding_model_instance NPE (#1907) --- api/core/indexing_runner.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/api/core/indexing_runner.py b/api/core/indexing_runner.py index 1c87432c4c..f4ccb7d765 100644 --- a/api/core/indexing_runner.py +++ b/api/core/indexing_runner.py @@ -382,13 +382,15 @@ class IndexingRunner: ) total_segments += len(documents) - embedding_model_type_instance = embedding_model_instance.model_type_instance - embedding_model_type_instance = cast(TextEmbeddingModel, embedding_model_type_instance) + embedding_model_type_instance = None + if embedding_model_instance: + embedding_model_type_instance = embedding_model_instance.model_type_instance + embedding_model_type_instance = cast(TextEmbeddingModel, embedding_model_type_instance) for document in documents: if len(preview_texts) < 5: preview_texts.append(document.page_content) - if indexing_technique == 'high_quality' or embedding_model_instance: + if indexing_technique == 'high_quality' and embedding_model_type_instance: tokens += embedding_model_type_instance.get_num_tokens( model=embedding_model_instance.model, credentials=embedding_model_instance.credentials,