NL: Check before accessing model (#4237)

datacommonsorg · May 14, 2024 · f47084d · f47084d
1 parent 40c44a6
commit f47084d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nl_server/embeddings_map.py b/nl_server/embeddings_map.py
@@ -112,6 +112,6 @@ def _set_embeddings(self, idx_name: str, idx_info: IndexConfig):
       raise e
 
     # if store successfully created, set it in embeddings_map
-    if store:
+    if store and idx_info.model in self.name_to_emb_model:
       self.embeddings_map[idx_name] = Embeddings(
           model=self.name_to_emb_model[idx_info.model], store=store)