Merge pull request #319 from sayari-analytics/bug/masoud/app-2124-fix-sources-in-tests

[APP-2124] Use ontoloty source to get data

Author: mtomran

examples/smoke-test.py

@@ -8,11 +8,14 @@ def main():
     client = load_env_vars_and_authenticate()
 
     # List and fetch sources
-    sources = client.source.list_sources().data
+    sources = client.ontology.get_sources().data
     print(f"Found {len(sources)} sources")
 
-    first_source = client.source.get_source(sources[0].id)
+    # Get first two sources by ID (using multiple IDs works correctly)
+    filtered_sources = client.ontology.get_sources(id=[sources[0].id, sources[1].id])
+    first_source = filtered_sources.data[0]
     print(f"First source is: {first_source.label}")
+    print(f"Retrieved {len(filtered_sources.data)} sources via filtered API call")
 
     # Search for an entity
     search_term = "victoria beckham limited"

tests/custom/test_sdk.py

@@ -45,7 +45,7 @@ def test_sources(setup_connection):
     client = setup_connection
 
     # list sources
-    sources = client.source.list_sources()
+    sources = client.ontology.get_sources()
 
     # We should have 250 sources as of 12/19/2023
     assert len(sources.data) >= 250