fix: Update type signatures to allow None tokens and improve error handling

- Allow None tokens in Agent, Organizations, Users, and CodegenSDK classes
- Update handle_api_error to accept None status codes from ApiException
- Add proper None handling in exception status code comparisons
- Maintain backward compatibility while improving type safety
- All tests passing and type checking issues resolved

Author: codegen-sh[bot]

src/codegen/agents/agent.py

@@ -75,7 +75,7 @@ def to_dict(self) -> dict[str, Any]:
 class Agent:
     """API client for interacting with Codegen AI agents."""
 
-    def __init__(self, token: str | None, org_id: int | None = None, base_url: str | None = CODEGEN_BASE_API_URL):
+    def __init__(self, token: str | None, org_id: int | str | None = None, base_url: str | None = CODEGEN_BASE_API_URL):
         """Initialize a new Agent client.
 
         Args:

src/codegen/exceptions.py

@@ -55,9 +55,11 @@ class APIError(CodegenError):
     pass
 
 
-def handle_api_error(status_code: int, message: str, response_data: dict[str, Any] | None = None) -> CodegenError:
+def handle_api_error(status_code: int | None, message: str, response_data: dict[str, Any] | None = None) -> CodegenError:
     """Convert HTTP status codes to appropriate exception types."""
-    if status_code == 401:
+    if status_code is None:
+        return APIError(message, status_code, response_data)
+    elif status_code == 401:
         return AuthenticationError(message, status_code, response_data)
     elif status_code == 403:
         return AuthorizationError(message, status_code, response_data)
@@ -67,7 +69,7 @@ def handle_api_error(status_code: int, message: str, response_data: dict[str, An
         return ValidationError(message, status_code, response_data)
     elif status_code == 429:
         return RateLimitError(message, status_code, response_data)
-    elif 500 <= status_code < 600:
+    elif status_code is not None and 500 <= status_code < 600:
         return ServerError(message, status_code, response_data)
     else:
         return APIError(message, status_code, response_data)

src/codegen/organizations/organizations.py

@@ -45,7 +45,7 @@ def __repr__(self) -> str:
 class Organizations:
     """API client for managing organizations in Codegen."""
 
-    def __init__(self, token: str, base_url: str = CODEGEN_BASE_API_URL):
+    def __init__(self, token: str | None, base_url: str = CODEGEN_BASE_API_URL):
         """Initialize the Organizations client.
 
         Args:

src/codegen/sdk.py

@@ -32,7 +32,7 @@ class CodegenSDK:
         ```
     """
 
-    def __init__(self, token: str, org_id: int | str | None = None, base_url: str = CODEGEN_BASE_API_URL):
+    def __init__(self, token: str | None, org_id: int | str | None = None, base_url: str = CODEGEN_BASE_API_URL):
         """Initialize the Codegen SDK.
 
         Args:

src/codegen/users/users.py

@@ -60,7 +60,7 @@ def __repr__(self) -> str:
 class Users:
     """API client for managing users in Codegen."""
 
-    def __init__(self, token: str, org_id: int | str, base_url: str = CODEGEN_BASE_API_URL):
+    def __init__(self, token: str | None, org_id: int | str, base_url: str = CODEGEN_BASE_API_URL):
         """Initialize the Users client.
 
         Args: