feat: Add sortByColumn functionality to List Operation

This change introduces the ability to sort results by a specific column in the List Operation.

PiperOrigin-RevId: 766468964
This commit is contained in:
Google Team Member 2025-06-02 21:22:49 -07:00 committed by Copybara-Service
parent eb2b9ef88f
commit af95dd2932
4 changed files with 15 additions and 6 deletions

View File

@ -252,6 +252,12 @@ class ConnectionsClient:
"Timeout in seconds for execution of custom query"
),
},
"sortByColumns": {
"type": "array",
"items": {"type": "string"},
"default": [],
"description": "Column to sort the results by",
},
"connectorOutputPayload": {"type": "object"},
"nextPageToken": {"type": "string"},
"execute-connector_Response": {
@ -665,6 +671,7 @@ class ConnectionsClient:
"serviceName": {"$ref": "#/components/schemas/serviceName"},
"host": {"$ref": "#/components/schemas/host"},
"entity": {"$ref": "#/components/schemas/entity"},
"sortByColumns": {"$ref": "#/components/schemas/sortByColumns"},
"dynamicAuthConfig": {
"$ref": "#/components/schemas/dynamicAuthConfig"
},

View File

@ -64,11 +64,7 @@ class IntegrationConnectorTool(BaseTool):
'dynamic_auth_config',
]
OPTIONAL_FIELDS = [
'page_size',
'page_token',
'filter',
]
OPTIONAL_FIELDS = ['page_size', 'page_token', 'filter', 'sortByColumns']
def __init__(
self,

View File

@ -511,6 +511,7 @@ class TestConnectionsClient:
assert schema["type"] == "object"
assert "properties" in schema
assert "filterClause" in schema["properties"]
assert "sortByColumns" in schema["properties"]
def test_action_request(self):
schema = ConnectionsClient.action_request("TestAction")

View File

@ -157,7 +157,11 @@ async def test_run_with_auth_async_none_token(
integration_tool_with_auth, mock_rest_api_tool
):
"""Tests run_async when auth credential token is None."""
input_args = {"user_id": "user456", "filter": "some_filter"}
input_args = {
"user_id": "user456",
"filter": "some_filter",
"sortByColumns": ["a", "b"],
}
expected_call_args = {
"user_id": "user456",
"filter": "some_filter",
@ -168,6 +172,7 @@ async def test_run_with_auth_async_none_token(
"entity": "TestEntity",
"operation": "LIST",
"action": "TestAction",
"sortByColumns": ["a", "b"],
}
with mock.patch(