From cefa92cfde94bc7d4d65e57692e26ee69df74fdc Mon Sep 17 00:00:00 2001 From: Brandon Rohrer Date: Thu, 14 Dec 2023 11:21:25 -0500 Subject: [PATCH] Update app.py --- app/enrichment/app.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/app/enrichment/app.py b/app/enrichment/app.py index 47969d6d2..1d5b76329 100644 --- a/app/enrichment/app.py +++ b/app/enrichment/app.py @@ -347,7 +347,10 @@ def poll_queue() -> None: # create embedding embedding = embed_texts(target_embeddings_model, [text]) - embedding_data = embedding['data'] + if 'data' in embedding: + embedding_data = embedding['data'] + else: + raise ValueError(embedding['message']) tag_list = get_tags_and_upload_to_cosmos(blob_service_client, chunk_dict["file_name"])