feat: cover search relationship RDF and report APIs

Author: ayush-shah

platform-api/src/openmetadata_demo_api/catalog.py

@@ -101,6 +101,19 @@
     "createStorageService",
     "createContainer",
 }
+SEARCH_RELATIONSHIP_PREFIXES = ("search/", "rdf/", "reports/")
+SEARCH_RELATIONSHIP_OPERATIONS = {
+    "searchSchemaEntityRelationship",
+    "searchEntityRelationship",
+    "searchEntityRelationshipWithDirection",
+}
+SEARCH_RELATIONSHIP_PREREQUISITES = {
+    "createDatabaseService",
+    "createDatabase",
+    "createDBSchema",
+    "createTable",
+    "createDashboardService",
+}
 CORE_SERVICE_PREREQUISITES = {
     "createApiService",
     "createDashboardService",
@@ -460,6 +473,17 @@ def scenarios() -> dict[str, Scenario]:
                 if _is_lineage_query_profile_usage_operation(operation)
             ),
         ),
+        "lineage-search-relationships": scenario_from_operations(
+            "lineage-search-relationships",
+            "Search, aggregation, indexing, RDF, relationship traversal, and reports",
+            tuple(
+                operation
+                for operation in all_operations
+                if str(operation["source"]["file"]).startswith(SEARCH_RELATIONSHIP_PREFIXES)
+                or operation["operation_id"] in SEARCH_RELATIONSHIP_OPERATIONS
+                or operation["operation_id"] in SEARCH_RELATIONSHIP_PREREQUISITES
+            ),
+        ),
     }
 
 

platform-api/src/openmetadata_demo_api/request_fixtures.py

@@ -63,6 +63,12 @@
             "assets": [DEMO_ENTITY_REFERENCE],
         },
         "AlertType": "Notification",
+        "AggregationRequest": {
+            "query": "",
+            "index": "table_search_index",
+            "fieldName": "service.name.keyword",
+            "size": 10,
+        },
         "AssetType": "Inline",
         "BulkColumnUpdateRequest": {
             "columnName": "demo",
@@ -199,6 +205,13 @@
             "stepName": "metadata",
             "message": "Dry-run progress",
         },
+        "PreviewSearchRequest": {
+            "query": "orders",
+            "index": "table_search_index",
+            "searchSettings": {},
+            "from": 0,
+            "size": 10,
+        },
         "RecognizerFeedback": {
             "entityLink": "<#E::table::demo.demo.demo.demo::columns::id>",
             "tagFQN": "PII.Sensitive",
@@ -221,6 +234,14 @@
         "ReportDataType": "entityReportData",
         "RestoreEntity": {"id": DEMO_ID},
         "SearchIndexSampleData": {"messages": ['{"id": 1}']},
+        "SparqlQuery": {
+            "query": (
+                "SELECT ?subject ?predicate ?object WHERE { ?subject ?predicate ?object } LIMIT 10"
+            ),
+            "format": "json",
+            "timeout": 30000,
+            "inference": "none",
+        },
         "String": "demo",
         "TableData": {"columns": ["id"], "rows": [[1]]},
         "TableJoins": {"startDate": "2026-01-01", "dayCount": 1},
@@ -250,6 +271,14 @@
             "description": "Updated by the dry-run demo",
         },
         "ValidateGlossaryTagsRequest": {"dryRun": True, "glossaryTags": []},
+        "VectorSearchRequest": {
+            "query": "orders",
+            "filters": {"entityType": ["table"]},
+            "size": 10,
+            "from": 0,
+            "k": 1000,
+            "threshold": 0.0,
+        },
         "VoteRequest": {"updatedVoteType": "votedUp"},
         "WebAnalyticEventData": {"timestamp": 1, "eventType": "PageView"},
         "WebAnalyticEventType": "PageView",
@@ -274,6 +303,10 @@
             "metadata.generated.schema.api.addTagToAssetsRequest",
             "AddTagToAssetsRequest",
         ),
+        "AggregationRequest": (
+            "metadata.generated.schema.search.aggregationRequest",
+            "AggregationRequest",
+        ),
         "BulkAssets": ("metadata.generated.schema.api.bulkAssets", "BulkAssets"),
         "BulkColumnUpdateRequest": (
             "metadata.generated.schema.api.data.bulkColumnUpdateRequest",
@@ -361,6 +394,10 @@
             "metadata.generated.schema.entity.services.ingestionPipelines.progressUpdate",
             "ProgressUpdate",
         ),
+        "PreviewSearchRequest": (
+            "metadata.generated.schema.api.search.previewSearchRequest",
+            "PreviewSearchRequest",
+        ),
         "RecognizerFeedback": (
             "metadata.generated.schema.type.recognizerFeedback",
             "RecognizerFeedback",
@@ -375,6 +412,10 @@
             "metadata.generated.schema.entity.data.searchIndex",
             "SearchIndexSampleData",
         ),
+        "SparqlQuery": (
+            "metadata.generated.schema.api.rdf.sparqlQuery",
+            "SparqlQuery",
+        ),
         "TableData": ("metadata.generated.schema.entity.data.table", "TableData"),
         "TableJoins": ("metadata.generated.schema.entity.data.table", "TableJoins"),
         "TableProfilerConfig": (
@@ -433,6 +474,15 @@
     }
 )
 
+JAVA_SOURCE_REQUESTS = MappingProxyType(
+    {
+        "VectorSearchRequest": (
+            "openmetadata-service/src/main/java/org/openmetadata/service/search/vector/"
+            "utils/DTOs.java"
+        )
+    }
+)
+
 
 def fixture_for(model_name: str) -> Any:
     return REQUEST_FIXTURES[model_name]

platform-api/tests/test_request_fixtures.py

@@ -2,6 +2,7 @@
 
 from openmetadata_demo_api.request_fixtures import (
     DEMO_ID,
+    JAVA_SOURCE_REQUESTS,
     PYDANTIC_PARAMETER_ENUMS,
     PYDANTIC_REQUEST_MODELS,
     REQUEST_FIXTURES,
@@ -31,6 +32,15 @@ def test_java_and_jakarta_request_fixtures_have_transport_valid_shapes() -> None
     ]
     assert REQUEST_FIXTURES["String"] == "demo"
     assert REQUEST_FIXTURES["UUID"] == DEMO_ID
+    assert JAVA_SOURCE_REQUESTS["VectorSearchRequest"].endswith("vector/utils/DTOs.java")
+    assert REQUEST_FIXTURES["VectorSearchRequest"] == {
+        "query": "orders",
+        "filters": {"entityType": ["table"]},
+        "size": 10,
+        "from": 0,
+        "k": 1000,
+        "threshold": 0.0,
+    }
 
 
 def test_required_parameter_enum_fixtures_are_valid_rc1_values() -> None:

platform-api/tests/test_runtime.py

@@ -1015,3 +1015,66 @@ def test_lineage_query_profile_usage_requests_are_validated_and_placeholder_free
         "CreateQueryCostRecordRequest",
         "CreateTableProfileRequest",
     } <= models
+
+
+def test_search_relationship_rdf_report_scenario_covers_source_routes() -> None:
+    prefixes = ("search/", "rdf/", "reports/")
+    relationship_operations = {
+        "searchSchemaEntityRelationship",
+        "searchEntityRelationship",
+        "searchEntityRelationshipWithDirection",
+    }
+    prerequisites = {
+        "createDatabaseService",
+        "createDatabase",
+        "createDBSchema",
+        "createTable",
+        "createDashboardService",
+    }
+    expected = {
+        operation["operation_id"]
+        for operation in load_operations()
+        if str(operation["source"]["file"]).startswith(prefixes)
+        or operation["operation_id"] in relationship_operations
+    }
+    scenario = scenarios()["lineage-search-relationships"]
+
+    assert {step.operation_id for step in scenario.operation_steps} == expected | prerequisites
+    assert len(scenario.run()) == len(expected | prerequisites)
+    assert {
+        "searchEntitiesWithQuery",
+        "previewSearch",
+        "aggregateSearchRequest",
+        "vectorSearch",
+        "querySparqlPost",
+        "getFullLineage",
+        "exportGlossaryAsOntology",
+        "createOrUpdateReport",
+    } <= expected
+    ordered = [step.operation_id for step in scenario.ordered_steps()]
+    assert ordered.index("createTable") < ordered.index("searchEntityRelationship")
+    assert ordered.index("createDashboardService") < ordered.index("createOrUpdateReport")
+
+
+def test_search_relationship_rdf_report_requests_are_validated_and_placeholder_free() -> None:
+    scenario = scenarios()["lineage-search-relationships"]
+
+    assert not {
+        str(value["_model"])
+        for step in scenario.operation_steps
+        for value in _nested_mappings(step.request)
+        if "_model" in value
+    }
+    vector = next(
+        step.request["body"]
+        for step in scenario.operation_steps
+        if step.operation_id == "vectorSearch"
+    )
+    assert vector["query"] == "orders"
+    assert vector["filters"]["entityType"] == ("table",)
+    assert (vector["size"], vector["from"], vector["k"], vector["threshold"]) == (
+        10,
+        0,
+        1000,
+        0.0,
+    )