Updated stack trace sequence + Added internal file flag

Author: bhogayatakb

middleware/__init__.py

@@ -83,3 +83,50 @@ def _auto_register(sender, **extra):
 appcontext_pushed.connect(_auto_register)
 
 
+# Automatic exception handling for FastAPI
+# import sys
+# import traceback
+# from fastapi import FastAPI, Request
+# from starlette.middleware.base import BaseHTTPMiddleware
+# from starlette.exceptions import HTTPException as StarletteHTTPException
+
+# async def fastapi_exception_handler(request: Request, exc: Exception):
+#     print("111 FastAPI _capture_exception Unhandled exception detected.")
+#     """Handles unhandled exceptions globally in FastAPI."""
+#     exc_type, exc_value, exc_traceback = sys.exc_info()
+#     if exc_type and exc_value and exc_traceback:
+#         record_exception(exc_type, exc_value, exc_traceback)
+#     return StarletteHTTPException(status_code=500, detail="Internal Server Error")
+
+# class ExceptionMiddleware(BaseHTTPMiddleware):
+#     print("222 FastAPI _capture_exception Unhandled exception detected.")
+#     """Middleware to capture unhandled exceptions in FastAPI."""
+#     async def dispatch(self, request, call_next):
+#         print("3333 FastAPI _capture_exception Unhandled exception detected.")
+#         try:
+#             return await call_next(request)
+#         except Exception as e:
+#             exc_type, exc_value, exc_traceback = sys.exc_info()
+#             if exc_type and exc_value and exc_traceback:
+#                 record_exception(exc_type, exc_value, exc_traceback)
+#             raise e
+
+# def try_register_fastapi_handler(app: FastAPI):
+#     """Registers the exception handler automatically when FastAPI is detected."""
+#     app.add_exception_handler(Exception, fastapi_exception_handler)
+#     app.add_middleware(ExceptionMiddleware)
+#     print("✅ FastAPI error handler registered automatically.")
+
+# def auto_register_fastapi():
+#     """Automatically detect FastAPI and register the handler."""
+#     if "fastapi" in sys.modules:
+#         app = None
+#         for obj in sys.modules["fastapi"].__dict__.values():
+#             if isinstance(obj, type) and issubclass(obj, FastAPI):
+#                 app = obj()
+#                 break
+#         if app:
+#             try_register_fastapi_handler(app)
+
+# # Try auto-registering when the package is imported
+# auto_register_fastapi()

middleware/distro.py

@@ -134,15 +134,23 @@ def custom_record_exception(span: Span, exc: Exception):
 
     for (frame, _), (filename, lineno, func_name, _) in zip(traceback.walk_tb(exc_tb), tb_details):
         function_details = extract_function_code(frame, lineno) if frame else "Function source not found."
-        
-        stack_info.append({
+  
+        stack_entry = {
             "exception.file": filename,
             "exception.line": lineno,
             "exception.function_name": func_name,
             "exception.function_body": function_details["function_code"],
             "exception.start_line": function_details["function_start_line"],
             "exception.end_line": function_details["function_end_line"],
-        })
+        }
+
+        # Check if the file is from site-packages
+        if "site-packages" in filename:
+            stack_entry["exception.is_file_internal"] = "true"
+        else:
+            stack_entry["exception.is_file_internal"] = "false"
+
+        stack_info.insert(0, stack_entry)  # Prepend instead of append
 
     # Determine if the exception is escaping
     current_exc = sys.exc_info()[1]  # Get the currently active exception

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "middleware-io"
-version = "2.1.2rc5"
+version = "2.1.2rc6"
 requires-python = ">=3.8"
 description = "Middleware's APM tool enables Python developers to effortlessly monitor their applications, gathering distributed tracing, metrics, logs, and profiling data for valuable insights and performance optimization."
 authors = [{ name = "middleware-dev" }]