Skip to content

Commit 758a823

Browse files
author
Vinod Durairaj
committed
fix: apply black formatting to pass CI tests
1 parent 2696b9b commit 758a823

File tree

2 files changed

+14
-6
lines changed

2 files changed

+14
-6
lines changed

src/redshift_utils_mcp/__main__.py

Lines changed: 13 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -150,27 +150,35 @@ def main(
150150

151151
if missing_vars:
152152
# Use typer for more elegant error output
153-
typer.echo(typer.style("Error: Missing required configuration!", fg=typer.colors.RED, bold=True))
154-
typer.echo("\nPlease provide the following configuration items either via command-line arguments or environment variables:")
153+
typer.echo(
154+
typer.style(
155+
"Error: Missing required configuration!", fg=typer.colors.RED, bold=True
156+
)
157+
)
158+
typer.echo(
159+
"\nPlease provide the following configuration items either via command-line arguments or environment variables:"
160+
)
155161

156162
# Map description back to CLI option for clarity
157163
config_map = {
158164
"REDSHIFT_CLUSTER_ID": "--cluster-id",
159165
"REDSHIFT_DATABASE": "--database",
160166
"REDSHIFT_SECRET_ARN": "--secret-arn",
161-
"AWS_REGION/AWS_DEFAULT_REGION": "--region"
167+
"AWS_REGION/AWS_DEFAULT_REGION": "--region",
162168
}
163169

164170
for var_desc in missing_vars:
165-
env_var_key = var_desc.split(" or ")[0] # Get the ENV VAR part
171+
env_var_key = var_desc.split(" or ")[0] # Get the ENV VAR part
166172
cli_opt = config_map.get(env_var_key)
167173

168174
message = f" - {typer.style(env_var_key, fg=typer.colors.YELLOW)} (Environment Variable)"
169175
if cli_opt:
170176
message += f" or {typer.style(cli_opt, fg=typer.colors.CYAN)} (Command-Line Argument)"
171177
typer.echo(message)
172178

173-
typer.echo(f"\nFor example, set {typer.style('REDSHIFT_CLUSTER_ID', fg=typer.colors.YELLOW)} or use {typer.style('--cluster-id', fg=typer.colors.CYAN)}.")
179+
typer.echo(
180+
f"\nFor example, set {typer.style('REDSHIFT_CLUSTER_ID', fg=typer.colors.YELLOW)} or use {typer.style('--cluster-id', fg=typer.colors.CYAN)}."
181+
)
174182
raise typer.Exit(code=1)
175183

176184
logger.info("Configuration loaded. Starting MCP server...")

src/redshift_utils_mcp/utils/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
SqlExecutionError,
99
DataApiTimeoutError,
1010
get_data_api_config,
11-
execute_sql,
11+
execute_sql,
1212
)
1313

1414

0 commit comments

Comments
 (0)