Merge pull request #400 from microsoft/brrohrer/modelcheckerror

brrohrer/modelcheckerror
microsoft · Dec 15, 2023 · e495630 · e495630
2 parents 3e32c3b + cefa92c
commit e495630
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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"])