get embeddings

doc_similarity/answer_similarity.py

@@ -53,7 +53,10 @@ if __name__ == "__main__":
         content = item["_source"]["answer"]
         content_level = str(item["_source"]["content_level"])
         level_dict[content_level].append(id)
-        embedding_dict[id] = bc.encode([content]).tolist()[0]
+        try:
+            embedding_dict[id] = bc.encode([content]).tolist()[0]
+        except Exception as e:
+            pass
 
     # redis_client_db.hmset("answer:level_dict", json.dumps(level_dict))
 

doc_similarity/diary_similarity.py

@@ -26,7 +26,10 @@ if __name__ == "__main__":
             content = item["_source"]["answer"]
             content_level = str(item["_source"]["content_level"])
             level_dict[content_level].append(id)
-            embedding_dict[id] = bc.encode([content]).tolist()[0]
+            try:
+                embedding_dict[id] = bc.encode([content]).tolist()[0]
+            except Exception as e:
+                pass
 
     # redis_client_db.hmset("diary:level_dict", json.dumps(level_dict))