fix: Py3.13+ logging issue with operation_Name (#1811)

* Py3.13+ logging fix

* Added runtime check for backward compatibility

* Fixed flake8 error

* ``

Author: gavin-aguiar

runtimes/v1/azure_functions_runtime_v1/__init__.py

@@ -11,6 +11,7 @@
     stop_threadpool_executor,
     get_threadpool_executor,
 )
+from .utils.executor import invocation_id_cv
 
 __all__ = ('worker_init_request',
            'functions_metadata_request',
@@ -19,4 +20,5 @@
            'function_load_request',
            'start_threadpool_executor',
            'stop_threadpool_executor',
-           'get_threadpool_executor')
+           'get_threadpool_executor',
+           'invocation_id_cv')

runtimes/v1/azure_functions_runtime_v1/utils/executor.py

@@ -2,6 +2,7 @@
 # Licensed under the MIT License.
 
 import asyncio
+import contextvars
 import functools
 
 from typing import Any
@@ -21,14 +22,19 @@ def execute_sync(function, args) -> Any:
     return function(**args)
 
 
+invocation_id_cv = contextvars.ContextVar('invocation_id', default=None)
+
+
 def run_sync_func(invocation_id, context, func, params):
     # This helper exists because we need to access the current
     # invocation_id from ThreadPoolExecutor's threads.
     context.thread_local_storage.invocation_id = invocation_id
+    token = invocation_id_cv.set(invocation_id)
     try:
         if otel_manager.get_azure_monitor_available():
             configure_opentelemetry(context)
         result = functools.partial(execute_sync, func)
         return result(params)
     finally:
+        invocation_id_cv.reset(token)
         context.thread_local_storage.invocation_id = None

runtimes/v2/azure_functions_runtime/__init__.py

@@ -10,6 +10,7 @@
     stop_threadpool_executor,
     get_threadpool_executor,
 )
+from .utils.executor import invocation_id_cv
 
 __all__ = ('worker_init_request',
            'functions_metadata_request',
@@ -18,4 +19,5 @@
            'function_load_request',
            'start_threadpool_executor',
            'stop_threadpool_executor',
-           'get_threadpool_executor')
+           'get_threadpool_executor',
+           'invocation_id_cv')

runtimes/v2/azure_functions_runtime/utils/executor.py

@@ -1,6 +1,7 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # Licensed under the MIT License.
 import asyncio
+import contextvars
 import functools
 
 from typing import Any
@@ -20,15 +21,20 @@ def execute_sync(function, args) -> Any:
     return function(**args)
 
 
+invocation_id_cv = contextvars.ContextVar('invocation_id', default=None)
+
+
 def run_sync_func(invocation_id, context, func, params):
     # This helper exists because we need to access the current
     # invocation_id from ThreadPoolExecutor's threads.
     context.thread_local_storage.invocation_id = invocation_id
+    token = invocation_id_cv.set(invocation_id)
     try:
         if (otel_manager.get_azure_monitor_available()
                 or otel_manager.get_otel_libs_available()):
             configure_opentelemetry(context)
         result = functools.partial(execute_sync, func)
         return result(params)
     finally:
+        invocation_id_cv.reset(token)
         context.thread_local_storage.invocation_id = None

workers/proxy_worker/dispatcher.py

@@ -14,6 +14,8 @@
 from typing import Any, Optional
 
 import grpc
+from packaging.version import Version
+
 from proxy_worker import protos
 from proxy_worker.logging import (
     CONSOLE_LOG_PREFIX,
@@ -32,7 +34,6 @@
     PYTHON_ENABLE_DEBUG_LOGGING,
 )
 from proxy_worker.version import VERSION
-
 from .utils.dependency import DependencyManager
 
 # Library worker import reloaded in init and reload request
@@ -99,10 +100,13 @@ def get_global_current_invocation_id() -> Optional[str]:
 
 
 def get_current_invocation_id() -> Optional[Any]:
+    global _library_worker
     # Check global current invocation first (most up-to-date)
-    global_invocation_id = get_global_current_invocation_id()
-    if global_invocation_id is not None:
-        return global_invocation_id
+    if (_library_worker
+            and Version(_library_worker.version.VERSION) < Version("1.1.0b4")):
+        global_invocation_id = get_global_current_invocation_id()
+        if global_invocation_id is not None:
+            return global_invocation_id
 
     # Check asyncio task context
     try:
@@ -125,6 +129,17 @@ def get_current_invocation_id() -> Optional[Any]:
     if thread_invocation_id is not None:
         return thread_invocation_id
 
+    # Check contextvar from library worker
+    if _library_worker:
+        try:
+            cv = getattr(_library_worker, 'invocation_id_cv', None)
+            if cv:
+                val = cv.get()
+                if val is not None:
+                    return val
+        except (AttributeError, LookupError):
+            pass
+
     return getattr(_invocation_id_local, 'invocation_id', None)
 
 

workers/tests/unittest_proxy/test_dispatcher.py

@@ -460,8 +460,8 @@ def test_thread_invocation_registry(self):
         # Test clear non-existent (should not raise)
         clear_thread_invocation_id(99999)
 
-    def test_get_current_invocation_id_priority_global(self):
-        """Test that global invocation ID has highest priority"""
+    def test_get_current_invocation_id_ignores_global_by_default(self):
+        """Test that global invocation ID is ignored by default"""
         global_id = "global-123"
         thread_id = threading.get_ident()
         thread_id_value = "thread-456"
@@ -470,9 +470,9 @@ def test_get_current_invocation_id_priority_global(self):
         set_current_invocation_id(global_id)
         set_thread_invocation_id(thread_id, thread_id_value)
 
-        # Global should take priority
+        # Thread should take priority (global is ignored)
         result = get_current_invocation_id()
-        self.assertEqual(result, global_id)
+        self.assertEqual(result, thread_id_value)
 
     def test_get_current_invocation_id_fallback_to_thread(self):
         """Test fallback to thread registry when global is None"""