diff --git a/src/agents/specialized/code_generator.py b/src/agents/specialized/code_generator.py
index 345f51cb6..7e51d2c97 100644
--- a/src/agents/specialized/code_generator.py
+++ b/src/agents/specialized/code_generator.py
@@ -43,7 +43,8 @@ async def {function_name}({parameters}):
                     except ValidationError as e:
                         raise HTTPException(status_code=400, detail=str(e))
                     except Exception as e:
-                        raise HTTPException(status_code=500, detail=str(e))
+                        logger.error(f"Unexpected error: {{e}}", exc_info=True)
+                        raise HTTPException(status_code=500, detail="Internal server error")
             """
             ),
             "rest_api": textwrap.dedent(
@@ -52,10 +53,13 @@ async def {function_name}({parameters}):
                 # Generated API endpoint
 
                 from fastapi import FastAPI, HTTPException
-                from pydantic import BaseModel
+                import logging
+                from pydantic import BaseModel, ValidationError
                 from datetime import datetime
                 from typing import Optional, List
 
+                logger = logging.getLogger(__name__)
+
                 {models}
 
                 {endpoints}