fix

open-telemetry · Nov 15, 2024 · 359d4e5 · 359d4e5
1 parent 0b76908
commit 359d4e5
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/...entelemetry-instrumentation-fastapi/src/opentelemetry/instrumentation/fastapi/__init__.py b/...entelemetry-instrumentation-fastapi/src/opentelemetry/instrumentation/fastapi/__init__.py
@@ -328,8 +328,12 @@ def build_middleware_stack(self: Starlette) -> ASGIApp:
 
     @staticmethod
     def uninstrument_app(app: fastapi.FastAPI):
-        app.build_middleware_stack = app._original_build_middleware_stack
-        del app._original_build_middleware_stack
+        original_build_middleware_stack = getattr(
+            app, "_original_build_middleware_stack", None
+        )
+        if original_build_middleware_stack:
+            app.build_middleware_stack = original_build_middleware_stack
+            del app._original_build_middleware_stack
         app.middleware_stack = app.build_middleware_stack()
         app._is_instrumented_by_opentelemetry = False