From 70fffb48d2b7427284e402bdd5f4f8595492b81c Mon Sep 17 00:00:00 2001 From: Auto Pipeline Date: Wed, 18 Mar 2026 10:24:28 +0000 Subject: [PATCH] feat(gooddata-sdk): [AUTO] add filter_context and date_attribute to search_ai object_types docstring Update search_ai() docstring to include the newly added 'filter_context' and 'date_attribute' enum values in object_types parameter, reflecting the LX-2171 OpenAPI spec changes. Add corresponding test for filter_context object type. Co-Authored-By: Claude Sonnet 4.6 --- .../src/gooddata_sdk/compute/service.py | 2 +- .../tests/compute/test_compute_service.py | 20 +++++++++++++++++++ 2 files changed, 21 insertions(+), 1 deletion(-) diff --git a/packages/gooddata-sdk/src/gooddata_sdk/compute/service.py b/packages/gooddata-sdk/src/gooddata_sdk/compute/service.py index 6163798b9..ddaf455f4 100644 --- a/packages/gooddata-sdk/src/gooddata_sdk/compute/service.py +++ b/packages/gooddata-sdk/src/gooddata_sdk/compute/service.py @@ -290,7 +290,7 @@ def search_ai( deep_search (bool): turn on deep search - if true, content of complex objects will be searched as well limit (Optional[int]): maximum number of results to return. Defaults to None. object_types (Optional[list[str]]): list of object types to search for. Enum items: "attribute", "metric", "fact", - "label", "date", "dataset", "visualization" and "dashboard". Defaults to None. + "label", "date", "dataset", "visualization", "dashboard", "date_attribute" and "filter_context". Defaults to None. relevant_score_threshold (Optional[float]): minimum relevance score threshold for results. Defaults to None. title_to_descriptor_ratio (Optional[float]): ratio of title score to descriptor score. Defaults to None. diff --git a/packages/gooddata-sdk/tests/compute/test_compute_service.py b/packages/gooddata-sdk/tests/compute/test_compute_service.py index f91dfa29a..7fb1cfd61 100644 --- a/packages/gooddata-sdk/tests/compute/test_compute_service.py +++ b/packages/gooddata-sdk/tests/compute/test_compute_service.py @@ -238,3 +238,23 @@ def test_build_exec_def_from_chat_result(test_config): finally: sdk.catalog_workspace.delete_workspace(test_workspace_id) sdk.compute.reset_ai_chat_history(test_workspace_id) + + +@gd_vcr.use_cassette(str(_fixtures_dir / "ai_search_filter_context.yaml")) +def test_search_ai_filter_context_object_type(test_config): + """Test AI search with filter_context object type.""" + sdk = GoodDataSdk.create(host_=test_config["host"], token_=test_config["token"]) + path = _current_dir / "load" / "ai" + test_workspace_id = test_config["workspace_test"] + + try: + _setup_test_workspace(sdk, test_workspace_id, path) + result = sdk.compute.search_ai( + workspace_id=test_workspace_id, + question="filter context", + object_types=["filter_context"], + ) + assert result is not None + assert hasattr(result, "results") + finally: + sdk.catalog_workspace.delete_workspace(test_workspace_id)