add limit

doc_similarity/answer_similarity.py

@@ -42,9 +42,10 @@ if __name__ == "__main__":
     count = 0
     embedding_dict = {}
     for item in get_answer_info_from_es(["id", "answer", "content_level"]):
+        if count < 1000:
             count += 1
-        print(count)
             id = int(item["_id"])
+            print(count, id)
             content = item["_source"]["answer"]
             content_level = str(item["_source"]["content_level"])
             # print(id, content_level, content)