Skip to content

Commit cbb0717

Browse files
authored
fix: properly handle set_cloud_connection_table_prefix tool within safe mode (#847)
1 parent 70f55a3 commit cbb0717

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

airbyte/mcp/cloud_ops.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -941,6 +941,7 @@ def rename_cloud_connection(
941941

942942
@mcp_tool(
943943
domain="cloud",
944+
destructive=True,
944945
open_world=True,
945946
)
946947
def set_cloud_connection_table_prefix(
@@ -955,10 +956,14 @@ def set_cloud_connection_table_prefix(
955956
) -> str:
956957
"""Set the table prefix for a connection on Airbyte Cloud.
957958
959+
This is a destructive operation that can break downstream dependencies if the
960+
table prefix is changed incorrectly. Use with caution.
961+
958962
By default, the `AIRBYTE_CLIENT_ID`, `AIRBYTE_CLIENT_SECRET`, `AIRBYTE_WORKSPACE_ID`,
959963
and `AIRBYTE_API_ROOT` environment variables will be used to authenticate with the
960964
Airbyte Cloud API.
961965
"""
966+
check_guid_created_in_session(connection_id)
962967
workspace: CloudWorkspace = _get_cloud_workspace()
963968
connection = workspace.get_connection(connection_id=connection_id)
964969
connection.set_table_prefix(prefix=prefix)

0 commit comments

Comments
 (0)