Propagate tags from worker

src/WebJobs.Script.Grpc/Channel/GrpcWorkerChannel.cs

@@ -1141,6 +1141,7 @@ internal async Task InvokeResponse(InvocationResponse invokeResponse)
                     }
 
                     LogSharedMemoryUsage(invokeResponse);
+                    AddWorkerTraceAttributes(invokeResponse, context);
 
                     if (invokeResponse.Result.IsInvocationSuccess())
                     {
@@ -1738,6 +1739,29 @@ private void AddAdditionalTraceContext(InvocationRequest invocationRequest, Scri
             }
         }
 
+        private void AddWorkerTraceAttributes(InvocationResponse invocationResponse, ScriptInvocationContext context)
+        {
+            var attributes = invocationResponse.TraceContext?.Attributes;
+            if (attributes is null)
+            {
+                return;
+            }
+
+            if (context.AsyncExecutionContext is null)
+            {
+                return;
+            }
+
+            System.Threading.ExecutionContext.Run(context.AsyncExecutionContext, static state =>
+            {
+                var attrs = (IDictionary<string, string>)state;
+                foreach (var kvp in attrs)
+                {
+                    Activity.Current?.AddTag(kvp.Key, kvp.Value);
+                }
+            }, attributes);
+        }
+
         private sealed class ExecutingInvocation : IDisposable
         {
             public ExecutingInvocation(ScriptInvocationContext context, IInvocationMessageDispatcher dispatcher)

src/WebJobs.Script.Grpc/azure-functions-language-worker-protobuf/src/proto/FunctionRpc.proto

@@ -442,6 +442,8 @@ message InvocationResponse {
 
   // Status of the invocation (success/failure/canceled)
   StatusResult result = 3;
+
+  RpcTraceContext trace_context = 5;
 }
 
 message WorkerWarmupRequest {