final fixes

Author: sohankshirsagar

drift/instrumentation/django/middleware.py

@@ -397,7 +397,7 @@ def dict_to_schema_merges(merges_dict):
             status = SpanStatus(code=StatusCode.OK, message="")
 
         # Django-specific: use route template for span name to avoid cardinality explosion
-        method = request.method
+        method = request.method or ""
         route_template = getattr(request, "_drift_route_template", None)
         if route_template:
             # Use route template (e.g., "users/<int:id>/")

drift/instrumentation/requests/instrumentation.py

@@ -44,7 +44,6 @@ def __init__(self, message: str, method: str, url: str):
     SpanStatus,
     StatusCode,
     TuskDriftMode,
-    calling_library_context,
 )
 from ..base import InstrumentationBase
 from ..http import HttpSpanData, HttpTransformEngine
@@ -267,16 +266,13 @@ def _handle_record(
                 error = None
                 response = None
 
-                # Set calling_library_context to prevent socket instrumentation warnings
-                calling_lib_token = calling_library_context.set("RequestsInstrumentation")
                 try:
                     response = original_request(session_self, method, url, **kwargs)
                     return response
                 except Exception as e:
                     error = e
                     raise
                 finally:
-                    calling_library_context.reset(calling_lib_token)
                     # Finalize span with request/response data
                     self._finalize_span(
                         span_info.span,

drift/instrumentation/socket/instrumentation.py

@@ -136,9 +136,8 @@ def _handle_socket_call(self, method_name: str, socket_self: Any) -> None:
 
         # Detect unpatched dependency:
         # - Must be in a SERVER span (inbound request context)
-        # - Must NOT be from an instrumented library (calling_library is set)
-        # Examples: ProtobufCommunicator, HttpxInstrumentation, RequestsInstrumentation
-        if span_kind == SpanKind.SERVER and calling_library is None:
+        # - Must NOT be from ProtobufCommunicator
+        if span_kind == SpanKind.SERVER and calling_library != "ProtobufCommunicator":
             self._log_unpatched_dependency(method_name, socket_self)
 
     def _log_unpatched_dependency(self, method_name: str, socket_self: Any) -> None: