fixed bug and bumped version (#250)

Author: maartendecat

pyproject.toml

@@ -9,7 +9,7 @@ license = "Apache-2.0"
 name = "elimity-insights-client"
 readme = "README.md"
 repository = "https://github.com/elimity-com/insights-client-python"
-version = "10.2.1b0"
+version = "10.2.1b1"
 
 [tool.poetry.dependencies]
 python = "^3.7"

src/elimity_insights_client/api/entities/_entities.py

@@ -26,7 +26,9 @@ def entities(
     schemas = {
         source.id: source.domain_graph_schema
         for source in sos
-        if linked_source_ids is None or source.id in linked_source_ids
+        if source.id == entity_type.source_id
+        or linked_source_ids is None
+        or source.id in linked_source_ids
     }
     que = query(entity_type, schemas)
     queries = [que]