lint and format

Author: StreetLamb

src/mcp_agent/cli/cloud/commands/auth/whoami/main.py

@@ -1,6 +1,5 @@
 """MCP Agent Cloud whoami command implementation."""
 
-
 from rich.console import Console
 from rich.panel import Panel
 from rich.table import Table

src/mcp_agent/cli/cloud/commands/logger/__init__.py

@@ -6,4 +6,4 @@
 
 from .tail.main import tail_logs
 
-__all__ = ["tail_logs"]
+__all__ = ["tail_logs"]

src/mcp_agent/cli/cloud/commands/logger/configure/__init__.py

@@ -2,4 +2,4 @@
 
 from .main import configure_logger
 
-__all__ = ["configure_logger"]
+__all__ = ["configure_logger"]

src/mcp_agent/cli/cloud/commands/logger/configure/main.py

@@ -32,10 +32,10 @@ def configure_logger(
     ),
 ) -> None:
     """Configure OTEL endpoint and headers for log collection.
-    
+
     This command allows you to configure the OpenTelemetry endpoint and headers
     that will be used for collecting logs from your deployed MCP apps.
-    
+
     Examples:
         mcp-agent cloud logger configure https://otel.example.com:4318/v1/logs
         mcp-agent cloud logger configure https://otel.example.com --headers "Authorization=Bearer token,X-Custom=value"
@@ -44,17 +44,19 @@ def configure_logger(
     if not endpoint and not test:
         console.print("[red]Error: Must specify endpoint or use --test[/red]")
         raise typer.Exit(1)
-    
+
     config_path = _find_config_file()
-    
+
     if test:
         if config_path and config_path.exists():
             config = _load_config(config_path)
             otel_config = config.get("otel", {})
             endpoint = otel_config.get("endpoint")
             headers_dict = otel_config.get("headers", {})
         else:
-            console.print("[yellow]No configuration file found. Use --endpoint to set up OTEL configuration.[/yellow]")
+            console.print(
+                "[yellow]No configuration file found. Use --endpoint to set up OTEL configuration.[/yellow]"
+            )
             raise typer.Exit(1)
     else:
         headers_dict = {}
@@ -64,54 +66,68 @@ def configure_logger(
                     key, value = header_pair.strip().split("=", 1)
                     headers_dict[key.strip()] = value.strip()
             except ValueError:
-                console.print("[red]Error: Headers must be in format 'key=value,key2=value2'[/red]")
+                console.print(
+                    "[red]Error: Headers must be in format 'key=value,key2=value2'[/red]"
+                )
                 raise typer.Exit(1)
-    
+
     if endpoint:
         console.print(f"[blue]Testing connection to {endpoint}...[/blue]")
-        
+
         try:
             with httpx.Client(timeout=10.0) as client:
                 response = client.get(
-                    endpoint.replace("/v1/logs", "/health") if "/v1/logs" in endpoint else f"{endpoint}/health",
-                    headers=headers_dict
+                    endpoint.replace("/v1/logs", "/health")
+                    if "/v1/logs" in endpoint
+                    else f"{endpoint}/health",
+                    headers=headers_dict,
                 )
-                
-                if response.status_code in [200, 404]:  # 404 is fine, means endpoint exists
+
+                if response.status_code in [
+                    200,
+                    404,
+                ]:  # 404 is fine, means endpoint exists
                     console.print("[green]✓ Connection successful[/green]")
                 else:
-                    console.print(f"[yellow]⚠ Got status {response.status_code}, but endpoint is reachable[/yellow]")
-                    
+                    console.print(
+                        f"[yellow]⚠ Got status {response.status_code}, but endpoint is reachable[/yellow]"
+                    )
+
         except httpx.RequestError as e:
             console.print(f"[red]✗ Connection failed: {e}[/red]")
             if not test:
-                console.print("[yellow]Configuration will be saved anyway. Check your endpoint URL and network connection.[/yellow]")
-    
+                console.print(
+                    "[yellow]Configuration will be saved anyway. Check your endpoint URL and network connection.[/yellow]"
+                )
+
     if not test:
         if not config_path:
             config_path = Path.cwd() / "mcp_agent.config.yaml"
-            
+
         config = _load_config(config_path) if config_path.exists() else {}
-        
+
         if "otel" not in config:
             config["otel"] = {}
-            
+
         config["otel"]["endpoint"] = endpoint
         config["otel"]["headers"] = headers_dict
-        
+
         try:
             config_path.parent.mkdir(parents=True, exist_ok=True)
             with open(config_path, "w") as f:
                 yaml.dump(config, f, default_flow_style=False, sort_keys=False)
-                
-            console.print(Panel(
-                f"[green]✓ OTEL configuration saved to {config_path}[/green]\n\n"
-                f"Endpoint: {endpoint}\n"
-                f"Headers: {len(headers_dict)} configured" + (f" ({', '.join(headers_dict.keys())})" if headers_dict else ""),
-                title="Configuration Saved",
-                border_style="green"
-            ))
-            
+
+            console.print(
+                Panel(
+                    f"[green]✓ OTEL configuration saved to {config_path}[/green]\n\n"
+                    f"Endpoint: {endpoint}\n"
+                    f"Headers: {len(headers_dict)} configured"
+                    + (f" ({', '.join(headers_dict.keys())})" if headers_dict else ""),
+                    title="Configuration Saved",
+                    border_style="green",
+                )
+            )
+
         except Exception as e:
             console.print(f"[red]Error saving configuration: {e}[/red]")
             raise typer.Exit(1)
@@ -134,4 +150,4 @@ def _load_config(config_path: Path) -> dict:
         with open(config_path, "r") as f:
             return yaml.safe_load(f) or {}
     except Exception as e:
-        raise CLIError(f"Failed to load config from {config_path}: {e}")
+        raise CLIError(f"Failed to load config from {config_path}: {e}")

src/mcp_agent/cli/cloud/commands/logger/tail/__init__.py

@@ -2,4 +2,4 @@
 
 from .main import tail_logs
 
-__all__ = ["tail_logs"]
+__all__ = ["tail_logs"]