Fix trailing whitespace issues to pass pre-commit checks

Author: codegen-sh[bot]

QUICK_START_LOGGING.md

@@ -38,7 +38,7 @@ codegen config telemetry enable
 # 2. Run the demo
 python example_enhanced_agent_command.py
 
-# 3. Run any CLI command 
+# 3. Run any CLI command
 codegen agents  # or any other command
 
 # 4. Check status
@@ -57,7 +57,7 @@ logger.info("Operation started", extra={
     "user_input": relevant_input
 })
 
-# At end of function  
+# At end of function
 logger.info("Operation completed", extra={
     "operation": "command.subcommand",
     "success": True
@@ -71,7 +71,7 @@ try:
     pass
 except SomeSpecificError as e:
     logger.error("Specific error occurred", extra={
-        "operation": "command.subcommand", 
+        "operation": "command.subcommand",
         "error_type": "specific_error",
         "error_details": str(e)
     }, exc_info=True)
@@ -89,7 +89,7 @@ logger.info("Making API request", extra={
 
 # After successful API call
 logger.info("API request successful", extra={
-    "operation": "api.request", 
+    "operation": "api.request",
     "endpoint": "agent/run",
     "response_id": response.get("id"),
     "status_code": response.status_code
@@ -101,7 +101,7 @@ logger.info("API request successful", extra={
 ### 🔥 High Priority (Add These First)
 - **Operation start/end**: When commands begin/complete
 - **API calls**: Requests to your backend
-- **Authentication events**: Login/logout/token issues  
+- **Authentication events**: Login/logout/token issues
 - **Errors**: Any exception or failure
 - **User actions**: Commands run, options selected
 
@@ -126,30 +126,30 @@ logger = get_logger(__name__)
 
 def create(prompt: str, org_id: int | None = None, ...):
     """Create a new agent run with the given prompt."""
-    
+
     # ADD: Log start
     logger.info("Agent creation started", extra={
         "operation": "agent.create",
         "org_id": org_id,
         "prompt_length": len(prompt)
     })
-    
+
     # Your existing code...
     try:
         response = requests.post(url, headers=headers, json=payload)
         agent_run_data = response.json()
-        
-        # ADD: Log success  
+
+        # ADD: Log success
         logger.info("Agent created successfully", extra={
             "operation": "agent.create",
             "agent_run_id": agent_run_data.get("id"),
             "status": agent_run_data.get("status")
         })
-        
+
     except requests.RequestException as e:
         # ADD: Log error
         logger.error("Agent creation failed", extra={
-            "operation": "agent.create", 
+            "operation": "agent.create",
             "error_type": "api_error",
             "error": str(e)
         })
@@ -165,28 +165,28 @@ logger = get_logger(__name__)
 
 def _run_claude_interactive(resolved_org_id: int, no_mcp: bool | None) -> None:
     session_id = generate_session_id()
-    
+
     # ADD: Log session start
     logger.info("Claude session started", extra={
         "operation": "claude.session_start",
         "session_id": session_id[:8],  # Short version for privacy
         "org_id": resolved_org_id
     })
-    
+
     # Your existing code...
-    
+
     try:
         process = subprocess.Popen([claude_path, "--session-id", session_id])
         returncode = process.wait()
-        
+
         # ADD: Log session end
         logger.info("Claude session completed", extra={
             "operation": "claude.session_complete",
             "session_id": session_id[:8],
             "exit_code": returncode,
             "status": "COMPLETE" if returncode == 0 else "ERROR"
         })
-        
+
     except Exception as e:
         # ADD: Log session error
         logger.error("Claude session failed", extra={
@@ -201,14 +201,14 @@ def _run_claude_interactive(resolved_org_id: int, no_mcp: bool | None) -> None:
 After making changes:
 
 1. **Run the command**: Execute your enhanced CLI command
-2. **Check telemetry status**: `codegen config telemetry status` 
+2. **Check telemetry status**: `codegen config telemetry status`
 3. **Look for logs in Grafana Cloud**: Search for your operation names
 4. **Test with telemetry disabled**: `codegen config telemetry disable` - should still work normally
 
 ## 🚀 Progressive Enhancement
 
 **Week 1**: Add basic operation logging to 2-3 commands
-**Week 2**: Add error logging to all commands  
+**Week 2**: Add error logging to all commands
 **Week 3**: Add performance metrics and detailed context
 **Week 4**: Create Grafana dashboards using the collected data
 

src/codegen/cli/commands/org/main.py

@@ -73,7 +73,7 @@ def _set_default_organization(org_id: int, cached_orgs: list[dict]) -> None:
 
     # Set the environment variable
     os.environ["CODEGEN_ORG_ID"] = str(org_id)
-    
+
     # Try to update .env file if it exists
     env_file_path = ".env"
     if os.path.exists(env_file_path):

src/codegen/cli/commands/org/tui.py

@@ -29,37 +29,37 @@ def __init__(self):
     def compose(self) -> ComposeResult:
         """Create child widgets for the app."""
         yield Header()
-        
+
         if not self.organizations:
             yield Container(
-                Static("⚠️  No organizations found. Please run 'codegen login' first.", classes="warning-message"), 
+                Static("⚠️  No organizations found. Please run 'codegen login' first.", classes="warning-message"),
                 id="no-orgs-warning"
             )
         else:
             with Vertical():
                 yield Static("🏢 Select Your Organization", classes="title")
                 yield Static("Use ↑↓ to navigate, Enter to select, Q/Esc to quit", classes="help")
-                
+
                 table = DataTable(id="orgs-table", cursor_type="row")
                 table.add_columns("Current", "ID", "Organization Name")
-                
+
                 # Get the actual current org ID (checks environment variables first)
                 actual_current_org_id = resolve_org_id()
-                
+
                 for org in self.organizations:
                     org_id = org["id"]
                     org_name = org["name"]
                     is_current = "●" if org_id == actual_current_org_id else " "
-                    
+
                     table.add_row(is_current, str(org_id), org_name, key=str(org_id))
-                
+
                 yield table
-                
+
                 yield Static(
-                    "\n💡 Selecting an organization will update your CODEGEN_ORG_ID environment variable.", 
+                    "\n💡 Selecting an organization will update your CODEGEN_ORG_ID environment variable.",
                     classes="help"
                 )
-        
+
         yield Footer()
 
     def on_mount(self) -> None:
@@ -89,7 +89,7 @@ def _handle_org_selection(self) -> None:
 
         try:
             table = self.query_one("#orgs-table", DataTable)
-            
+
             if table.cursor_row is not None and table.cursor_row < len(self.organizations):
                 # Get the selected organization directly from the cursor position
                 selected_org = self.organizations[table.cursor_row]
@@ -106,24 +106,24 @@ def _set_organization(self, org_id: int, org_name: str) -> None:
         """Set the selected organization as default."""
         # Set environment variable
         os.environ["CODEGEN_ORG_ID"] = str(org_id)
-        
+
         # Try to update .env file
         env_updated = self._update_env_file(org_id)
-        
+
         if env_updated:
             self.notify(f"✓ Set default organization: {org_name} (ID: {org_id})")
             self.notify("✓ Updated .env file with CODEGEN_ORG_ID")
         else:
             self.notify(f"✓ Set organization: {org_name} (ID: {org_id})")
             self.notify("ℹ  Add 'export CODEGEN_ORG_ID={org_id}' to your shell for persistence")
-        
+
         # Wait a moment for user to see the notifications, then close
         self.set_timer(2.0, self._close_screen)
 
     def _update_env_file(self, org_id: int) -> bool:
         """Update the .env file with the new organization ID."""
         env_file_path = ".env"
-        
+
         try:
             lines = []
             key_found = False
@@ -152,9 +152,9 @@ def _update_env_file(self, org_id: int) -> bool:
             # Write back to file
             with open(env_file_path, "w") as f:
                 f.writelines(lines)
-            
+
             return True
-            
+
         except Exception:
             return False
 
@@ -174,7 +174,7 @@ def action_quit(self) -> None:
 
 class OrgSelectorApp(App):
     """Standalone app wrapper for the organization selector."""
-    
+
     CSS_PATH = "../../tui/codegen_theme.tcss"  # Use custom Codegen theme
     TITLE = "Organization Selector - Codegen CLI"
     BINDINGS = [
@@ -191,37 +191,37 @@ def __init__(self):
     def compose(self) -> ComposeResult:
         """Create child widgets for the app."""
         yield Header()
-        
+
         if not self.organizations:
             yield Container(
-                Static("⚠️  No organizations found. Please run 'codegen login' first.", classes="warning-message"), 
+                Static("⚠️  No organizations found. Please run 'codegen login' first.", classes="warning-message"),
                 id="no-orgs-warning"
             )
         else:
             with Vertical():
                 yield Static("🏢 Select Your Organization", classes="title")
                 yield Static("Use ↑↓ to navigate, Enter to select, Q/Esc to quit", classes="help")
-                
+
                 table = DataTable(id="orgs-table", cursor_type="row")
                 table.add_columns("Current", "ID", "Organization Name")
-                
+
                 # Get the actual current org ID (checks environment variables first)
                 actual_current_org_id = resolve_org_id()
-                
+
                 for org in self.organizations:
                     org_id = org["id"]
                     org_name = org["name"]
                     is_current = "●" if org_id == actual_current_org_id else " "
-                    
+
                     table.add_row(is_current, str(org_id), org_name, key=str(org_id))
-                
+
                 yield table
-                
+
                 yield Static(
-                    "\n💡 Selecting an organization will update your CODEGEN_ORG_ID environment variable.", 
+                    "\n💡 Selecting an organization will update your CODEGEN_ORG_ID environment variable.",
                     classes="help"
                 )
-        
+
         yield Footer()
 
     def on_mount(self) -> None:
@@ -251,7 +251,7 @@ def _handle_org_selection(self) -> None:
 
         try:
             table = self.query_one("#orgs-table", DataTable)
-            
+
             if table.cursor_row is not None and table.cursor_row < len(self.organizations):
                 # Get the selected organization directly from the cursor position
                 selected_org = self.organizations[table.cursor_row]
@@ -268,24 +268,24 @@ def _set_organization(self, org_id: int, org_name: str) -> None:
         """Set the selected organization as default."""
         # Set environment variable
         os.environ["CODEGEN_ORG_ID"] = str(org_id)
-        
+
         # Try to update .env file
         env_updated = self._update_env_file(org_id)
-        
+
         if env_updated:
             self.notify(f"✓ Set default organization: {org_name} (ID: {org_id})")
             self.notify("✓ Updated .env file with CODEGEN_ORG_ID")
         else:
             self.notify(f"✓ Set organization: {org_name} (ID: {org_id})")
             self.notify("ℹ  Add 'export CODEGEN_ORG_ID={org_id}' to your shell for persistence")
-        
+
         # Wait a moment for user to see the notifications, then exit
         self.set_timer(2.0, self.exit)
 
     def _update_env_file(self, org_id: int) -> bool:
         """Update the .env file with the new organization ID."""
         env_file_path = ".env"
-        
+
         try:
             lines = []
             key_found = False
@@ -314,9 +314,9 @@ def _update_env_file(self, org_id: int) -> bool:
             # Write back to file
             with open(env_file_path, "w") as f:
                 f.writelines(lines)
-            
+
             return True
-            
+
         except Exception:
             return False