Handle db subcommand.

Handle dms subcommand for DMS tools.

Handle file subcommand.

Handle import subcommand for adding DMS tables to Data Center.

List DMS databases and (optionally) their tables.

Handle reports subcommand.

List Data Agent workspaces.

Get the client configuration.

Get the client configuration.

Create a new session or reuse an existing one asynchronously. Args: session_id: Optional existing session ID to reuse.

Create a new Data Agent session asynchronously. Args: database_id: Optional database ID to bind to the session. title

Delete an uploaded file. Args: file_id: The file ID to delete. client_token: Optional idempotency token, auto-generat

Delete a file asynchronously. Args: file_id: The file ID to delete. client_token: Optional idempotency token.

Delete an uploaded file asynchronously. Args: file_id: The file ID to delete. session_id: Optional session ID used to

Get detailed information about a custom agent asynchronously. Args: custom_agent_id: The custom agent ID. workspace_i

Get the status of a session asynchronously. Args: session_id: The session ID. agent_id: The agent ID (optional).

Download a file by ID. Args: file_id: The file ID to download. save_path: Path to save the downloaded file.

Download a file from URL asynchronously. Args: download_url: URL to download from. save_path: Path to save the downlo

Notify file upload completion asynchronously. Args: file_id: The file ID from upload signature. filename: The origina

Create configuration from a dictionary. Args: config_dict: Dictionary containing configuration values. Returns:

Get chat content from the Data Agent. Args: agent_id: The agent ID. session_id: The session ID. checkpoin

Get chat content from the Data Agent asynchronously. Args: agent_id: The agent ID. session_id: The session ID.

Get the file type from path. Args: file_path: Path to the file. Returns: File type string or None if unsuppo

Get OSS upload signature asynchronously. Args: filename: Name of the file to upload. file_size: Size of the file in b

Get complete response asynchronously. Collects all `data` events (excluding tool_call_choices and metric_agent_config), which carry t

Get session info from cache. Args: session_id: The session ID. Returns: SessionInfo if found, None otherwise

Get session info from cache. Args: session_id: The session ID. Returns: SessionInfo if found, None otherwise

Check if there are more pages.

List custom agents asynchronously. Args: workspace_id: Optional workspace ID to filter agents. search_key: Fuzzy sear

List tables inside a DMS Data Center database (legacy file-based). Args: instance_name: The InstanceName returned by list_file_da

List files asynchronously. Args: session_id: The session ID. file_category: Optional filter, e.g. "WebReport" for age

Async search for database instances in DMS. Args: search_key: Optional search key. db_type: Optional database

List agent-generated reports (WebReport) for a session. Args: session_id: The session ID. Returns: List of F

List Data Agent sessions. For API_KEY authentication, uses startTime/endTime parameters. For AK/SK authentication, uses createStartTi

List all cached sessions. Returns: List of SessionInfo objects.

List all cached sessions. Returns: List of SessionInfo objects.

List Data Agent workspaces asynchronously. Args: workspace_type: Workspace type filter ("MY" or "ALL"). workspace_nam

Push a notification message to the specified session or active session. Args: session_id: The target session ID (optional). messa

Refresh session status from API. Args: session_id: The session ID. Returns: Updated SessionInfo. Ra

Refresh session status from API asynchronously. Args: session_id: The session ID. Returns: Updated SessionIn

Remove session from cache. Args: session_id: The session ID to remove.

Remove session from cache. Args: session_id: The session ID to remove.

Decorator to retry API calls on transient errors with exponential backoff.

Run a subprocess and create both progress.log and progress.jsonl files.

Send a message to the Data Agent asynchronously. Args: agent_id: The agent ID. session_id: The session ID.

Send a query and wait for complete response. Args: session: Active session to use. query: Natural language query.

Send a query and wait for complete response asynchronously. Args: session: Active session to use. query: Natural lang

Send a query and get detailed result with content blocks. Args: session: Active session to use. query: Natural langua

Send a query and get detailed result asynchronously. Args: session: Active session to use. query: Natural language qu

Send query and stream response content blocks. Args: session: Active session to use. query: Natural language query.

Send query and stream response content blocks asynchronously. Args: session: Active session to use. query: Natural la

Convert configuration to dictionary (excluding secrets). Returns: Dictionary with non-sensitive configuration values.

Calculate total number of pages.

Upload a file for analysis asynchronously. Args: file_path: Path to the local file. timeout: Upload timeout in second