Added condition

Author: dristy.cd

pebblo/app/service/doc_helper.py

@@ -245,18 +245,20 @@ def _create_data_source_findings(data_source_findings):
             for snippet in data.get("snippets"):
                 entity_details = {}
                 topic_details = {}
-                if data["labelName"] in snippet.get("entityDetails").keys():
-                    entity_details = {
-                        data["labelName"]: snippet.get("entityDetails")[
-                            data["labelName"]
-                        ]
-                    }
-                elif data["labelName"] in snippet.get("topicDetails").keys():
-                    topic_details = {
-                        data["labelName"]: snippet.get("topicDetails")[
-                            data["labelName"]
-                        ]
-                    }
+                if snippet.get("entityDetails"):
+                    if data["labelName"] in snippet.get("entityDetails").keys():
+                        entity_details = {
+                            data["labelName"]: snippet.get("entityDetails")[
+                                data["labelName"]
+                            ]
+                        }
+                if snippet.get("topicDetails"):
+                    if data["labelName"] in snippet.get("topicDetails").keys():
+                        topic_details = {
+                            data["labelName"]: snippet.get("topicDetails")[
+                                data["labelName"]
+                            ]
+                        }
                 snippet["entityDetails"] = entity_details
                 snippet["topicDetails"] = topic_details
         return data_source_findings