fix: Apply ruff formatting to resolve pre-commit failures

- Remove trailing whitespace from docstrings
- Modernize type hints (Union -> |, Optional -> | None)
- Fix import ordering and formatting
- Ensure consistent code style across all files

Author: codegen-sh[bot]

examples/sdk_usage.py

@@ -1,5 +1,4 @@
-"""
-Example usage of the enhanced Codegen SDK.
+"""Example usage of the enhanced Codegen SDK.
 
 This example demonstrates how to use the new SDK components:
 - CodegenSDK: Unified entry point for all API functionality
@@ -9,29 +8,31 @@
 """
 
 import os
+
 from codegen.agents import Agent, CodegenSDK, Organizations, Users
 
+
 # Example 1: Using the unified SDK
 def example_unified_sdk():
     """Example using the unified CodegenSDK."""
     print("=== Unified SDK Example ===")
-    
+
     # Initialize the SDK with your API token
     sdk = CodegenSDK(token=os.getenv("CODEGEN_API_TOKEN"))
-    
+
     # Use agents
     print("Running an agent task...")
     task = sdk.agents.run("Analyze the codebase and suggest improvements")
     print(f"Task created: {task.id}")
     print(f"Status: {task.status}")
     print(f"Web URL: {task.web_url}")
-    
+
     # List organizations
     print("\nListing organizations...")
     orgs = sdk.organizations.list()
     for org in orgs:
         print(f"- {org.name} (ID: {org.id})")
-    
+
     # Get users in the first organization
     if orgs:
         print(f"\nListing users in organization '{orgs[0].name}'...")
@@ -44,17 +45,17 @@ def example_unified_sdk():
 def example_individual_components():
     """Example using individual SDK components."""
     print("\n=== Individual Components Example ===")
-    
+
     token = os.getenv("CODEGEN_API_TOKEN")
-    
+
     # Enhanced Agent with better error handling
     print("Using enhanced Agent...")
     agent = Agent(token=token)
-    
+
     try:
         task = agent.run("Create a simple Python function")
         print(f"Agent task: {task.id}")
-        
+
         # Check task status with new helper methods
         if task.is_running():
             print("Task is currently running...")
@@ -64,55 +65,55 @@ def example_individual_components():
                 print("Task was successful!")
             elif task.is_failed():
                 print("Task failed.")
-        
+
         # Get task by ID
         retrieved_task = agent.get_task(task.id)
         print(f"Retrieved task: {retrieved_task.id}")
-        
+
     except Exception as e:
         print(f"Error: {e}")
-    
+
     # Organizations management
     print("\nUsing Organizations...")
     orgs_client = Organizations(token=token)
-    
+
     try:
         # List with pagination
         page_result = orgs_client.get_page(page=1, page_size=10)
         print(f"Organizations page: {page_result['page']}/{page_result['pages']}")
         print(f"Total organizations: {page_result['total']}")
-        
+
         # Iterate through all organizations
         print("All organizations:")
         for org in orgs_client.list_all():
             print(f"- {org.name} (ID: {org.id})")
             print(f"  Settings: {org.settings}")
-            
+
     except Exception as e:
         print(f"Error: {e}")
-    
+
     # Users management
     print("\nUsing Users...")
     if orgs:
         users_client = Users(token=token, org_id=orgs[0].id)
-        
+
         try:
             # List users with pagination
             page_result = users_client.get_page(page=1, page_size=5)
             print(f"Users page: {page_result['page']}/{page_result['pages']}")
             print(f"Total users: {page_result['total']}")
-            
+
             # Find specific users
             user = users_client.find_by_github_username("example-user")
             if user:
                 print(f"Found user: {user.github_username}")
-            
+
             # Get user by ID
-            if page_result['items']:
-                first_user = page_result['items'][0]
+            if page_result["items"]:
+                first_user = page_result["items"][0]
                 retrieved_user = users_client.get(first_user.id)
                 print(f"Retrieved user: {retrieved_user.github_username}")
-                
+
         except Exception as e:
             print(f"Error: {e}")
 
@@ -121,9 +122,9 @@ def example_individual_components():
 def example_error_handling():
     """Example demonstrating error handling."""
     print("\n=== Error Handling Example ===")
-    
-    from codegen.exceptions import AuthenticationError, NotFoundError, CodegenError
-    
+
+    from codegen.exceptions import AuthenticationError, CodegenError, NotFoundError
+
     # Invalid token example
     try:
         sdk = CodegenSDK(token="invalid-token")
@@ -133,7 +134,7 @@ def example_error_handling():
         print(f"Status code: {e.status_code}")
     except CodegenError as e:
         print(f"API error: {e.message}")
-    
+
     # Not found example
     try:
         sdk = CodegenSDK(token=os.getenv("CODEGEN_API_TOKEN"))
@@ -149,7 +150,7 @@ def example_error_handling():
     if not os.getenv("CODEGEN_API_TOKEN"):
         print("Please set the CODEGEN_API_TOKEN environment variable")
         exit(1)
-    
+
     example_unified_sdk()
     example_individual_components()
     example_error_handling()

src/codegen/agents/__init__.py

@@ -1,10 +1,10 @@
 """Codegen Agent API module."""
 
 from codegen.agents.agent import Agent
+from codegen.organizations import Organizations
 
 # Import SDK components for backward compatibility and convenience
 from codegen.sdk import CodegenSDK
-from codegen.organizations import Organizations
 from codegen.users import Users
 
 __all__ = ["Agent", "CodegenSDK", "Organizations", "Users"]

src/codegen/agents/agent.py

@@ -1,5 +1,5 @@
 import os
-from typing import Any, Dict, Optional, Union
+from typing import Any
 
 from codegen_api_client.api.agents_api import AgentsApi
 from codegen_api_client.api_client import ApiClient
@@ -26,7 +26,7 @@ def __init__(self, task_data: AgentRunResponse, api_client: ApiClient, org_id: i
 
     def refresh(self) -> None:
         """Refresh the job status from the API.
-        
+
         Raises:
             CodegenError: If the API request fails
         """
@@ -48,40 +48,34 @@ def refresh(self) -> None:
             self.status = job_dict.get("status")
             self.result = job_dict.get("result")
         except ApiException as e:
-            error = handle_api_error(e.status, str(e), getattr(e, 'body', None))
+            error = handle_api_error(e.status, str(e), getattr(e, "body", None))
             raise error from e
-    
+
     def is_completed(self) -> bool:
         """Check if the agent task is completed."""
-        return self.status in ['completed', 'failed', 'cancelled']
-    
+        return self.status in ["completed", "failed", "cancelled"]
+
     def is_running(self) -> bool:
         """Check if the agent task is currently running."""
-        return self.status in ['running', 'pending']
-    
+        return self.status in ["running", "pending"]
+
     def is_successful(self) -> bool:
         """Check if the agent task completed successfully."""
-        return self.status == 'completed'
-    
+        return self.status == "completed"
+
     def is_failed(self) -> bool:
         """Check if the agent task failed."""
-        return self.status == 'failed'
-    
-    def to_dict(self) -> Dict[str, Any]:
+        return self.status == "failed"
+
+    def to_dict(self) -> dict[str, Any]:
         """Convert agent task to dictionary."""
-        return {
-            'id': self.id,
-            'org_id': self.org_id,
-            'status': self.status,
-            'result': self.result,
-            'web_url': self.web_url
-        }
+        return {"id": self.id, "org_id": self.org_id, "status": self.status, "result": self.result, "web_url": self.web_url}
 
 
 class Agent:
     """API client for interacting with Codegen AI agents."""
 
-    def __init__(self, token: str, org_id: Optional[Union[int, str]] = None, base_url: str = CODEGEN_BASE_API_URL):
+    def __init__(self, token: str, org_id: int | str | None = None, base_url: str = CODEGEN_BASE_API_URL):
         """Initialize a new Agent client.
 
         Args:
@@ -98,7 +92,7 @@ def __init__(self, token: str, org_id: Optional[Union[int, str]] = None, base_ur
         self.agents_api = AgentsApi(self.api_client)
 
         # Current job
-        self.current_job: Optional[AgentTask] = None
+        self.current_job: AgentTask | None = None
 
     def run(self, prompt: str) -> AgentTask:
         """Run an agent with the given prompt.
@@ -108,7 +102,7 @@ def run(self, prompt: str) -> AgentTask:
 
         Returns:
             AgentTask: A task object representing the agent run
-            
+
         Raises:
             CodegenError: If the API request fails
         """
@@ -122,43 +116,39 @@ def run(self, prompt: str) -> AgentTask:
             self.current_job = job
             return job
         except ApiException as e:
-            error = handle_api_error(e.status, str(e), getattr(e, 'body', None))
+            error = handle_api_error(e.status, str(e), getattr(e, "body", None))
             raise error from e
 
-    def get_status(self) -> Optional[Dict[str, Any]]:
+    def get_status(self) -> dict[str, Any] | None:
         """Get the status of the current job.
 
         Returns:
             dict: A dictionary containing job status information,
                   or None if no job has been run.
-                  
+
         Raises:
             CodegenError: If the API request fails
         """
         if self.current_job:
             self.current_job.refresh()
             return {"id": self.current_job.id, "status": self.current_job.status, "result": self.current_job.result, "web_url": self.current_job.web_url}
         return None
-    
-    def get_task(self, task_id: Union[int, str]) -> AgentTask:
+
+    def get_task(self, task_id: int | str) -> AgentTask:
         """Get a specific agent task by ID.
-        
+
         Args:
             task_id: Agent task ID to retrieve
-            
+
         Returns:
             AgentTask: The requested agent task
-            
+
         Raises:
             CodegenError: If the API request fails or task is not found
         """
         try:
-            response = self.agents_api.get_agent_run_v1_organizations_org_id_agent_run_agent_run_id_get(
-                org_id=int(self.org_id),
-                agent_run_id=int(task_id),
-                authorization=f"Bearer {self.token}"
-            )
+            response = self.agents_api.get_agent_run_v1_organizations_org_id_agent_run_agent_run_id_get(org_id=int(self.org_id), agent_run_id=int(task_id), authorization=f"Bearer {self.token}")
             return AgentTask(response, self.api_client, self.org_id)
         except ApiException as e:
-            error = handle_api_error(e.status, str(e), getattr(e, 'body', None))
+            error = handle_api_error(e.status, str(e), getattr(e, "body", None))
             raise error from e

src/codegen/cli/auth/decorators.py

@@ -39,7 +39,7 @@ def wrapper(*args, **kwargs):
 
     # Remove the session parameter from the wrapper's signature so Typer doesn't see it
     sig = inspect.signature(f)
-    new_params = [param for name, param in sig.parameters.items() if name != 'session']
+    new_params = [param for name, param in sig.parameters.items() if name != "session"]
     new_sig = sig.replace(parameters=new_params)
     wrapper.__signature__ = new_sig  # type: ignore[attr-defined]
 

src/codegen/cli/cli.py

@@ -1,9 +1,10 @@
 import typer
 from rich.traceback import install
 
+from codegen import __version__
+
 # Import config command (still a Typer app)
 from codegen.cli.commands.config.main import config_command
-from codegen import __version__
 
 install(show_locals=True)
 
@@ -14,12 +15,9 @@ def version_callback(value: bool):
         print(__version__)
         raise typer.Exit()
 
+
 # Create the main Typer app
-main = typer.Typer(
-    name="codegen",
-    help="Codegen CLI - Transform your code with AI.",
-    rich_markup_mode="rich"
-)
+main = typer.Typer(name="codegen", help="Codegen CLI - Transform your code with AI.", rich_markup_mode="rich")
 
 # Import the actual command functions
 from codegen.cli.commands.init.main import init
@@ -42,9 +40,7 @@ def version_callback(value: bool):
 
 
 @main.callback()
-def main_callback(
-    version: bool = typer.Option(False, "--version", callback=version_callback, is_eager=True, help="Show version and exit")
-):
+def main_callback(version: bool = typer.Option(False, "--version", callback=version_callback, is_eager=True, help="Show version and exit")):
     """Codegen CLI - Transform your code with AI."""
     pass
 

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

@@ -90,10 +90,7 @@ def get_config(key: str = typer.Argument(..., help="Configuration key to get")):
 
 
 @config_command.command(name="set")
-def set_config(
-    key: str = typer.Argument(..., help="Configuration key to set"),
-    value: str = typer.Argument(..., help="Configuration value to set")
-):
+def set_config(key: str = typer.Argument(..., help="Configuration key to set"), value: str = typer.Argument(..., help="Configuration value to set")):
     """Set a configuration value and write to .env"""
     config = _get_user_config()
     if not config.has_key(key):