removed changes from PowerShellServices and added helper function

Author: bachuv

src/AzureFunctions.PowerShell.Durable.SDK.psm1

@@ -114,6 +114,34 @@ function Start-DurableOrchestration {
     }
 
     $invocationId = GetInvocationIdFromModulePrivateData
+    $headers = Get-TraceHeaders -InvocationId $invocationId
+
+    $Uri =
+        if ($DurableClient.rpcBaseUrl) {
+            # Fast local RPC path
+            "$($DurableClient.rpcBaseUrl)orchestrators/$FunctionName$($InstanceId ? "/$InstanceId" : '')"
+        } else {
+            # Legacy app frontend path
+            $UriTemplate = $DurableClient.creationUrls.createNewInstancePostUri
+            $UriTemplate.Replace('{functionName}', $FunctionName).Replace('[/{instanceId}]', "/$InstanceId")
+        }
+
+    $Body = $InputObject | ConvertTo-Json -Compress -Depth 100
+              
+    $null = Invoke-RestMethod -Uri $Uri -Method 'POST' -ContentType 'application/json' -Body $Body -Headers $headers
+    
+    return $instanceId
+}
+
+function Get-TraceHeaders {
+    param(
+        [Parameter(Mandatory = $true)]
+        [string] $InvocationId
+    )
+
+    if ($null -eq $InvocationId) {
+        return @{} # Return an empty headers object
+    }
 
     $activityResponse = Get-CurrentActivityForInvocation -InvocationId $invocationId
     $activity = $activityResponse.activity
@@ -135,21 +163,7 @@ function Start-DurableOrchestration {
         "tracestate"  = $traceState
     }
 
-    $Uri =
-        if ($DurableClient.rpcBaseUrl) {
-            # Fast local RPC path
-            "$($DurableClient.rpcBaseUrl)orchestrators/$FunctionName$($InstanceId ? "/$InstanceId" : '')"
-        } else {
-            # Legacy app frontend path
-            $UriTemplate = $DurableClient.creationUrls.createNewInstancePostUri
-            $UriTemplate.Replace('{functionName}', $FunctionName).Replace('[/{instanceId}]', "/$InstanceId")
-        }
-
-    $Body = $InputObject | ConvertTo-Json -Compress -Depth 100
-              
-    $null = Invoke-RestMethod -Uri $Uri -Method 'POST' -ContentType 'application/json' -Body $Body -Headers $headers
-    
-    return $instanceId
+    return $headers
 }
 
 function Stop-DurableOrchestration {

src/DurableEngine/Utilities/IPowerShellServices.cs

@@ -13,8 +13,6 @@ public interface IPowerShellServices
     {
         void SetDurableClient(object durableClient);
 
-        void SetInvocationId(string invocationId);
-
         void SetOrchestrationContext(OrchestrationContext orchestrationContext);
 
         void ClearOrchestrationContext();

src/DurableEngine/Utilities/PowerShellServices.cs

@@ -45,15 +45,6 @@ public void SetDurableClient(object durableClient)
             _hasSetOrchestrationContext = true;
         }
 
-        public void SetInvocationId(string invocationId)
-        {
-            _pwsh.AddCommand(SetFunctionInvocationContextCommand)
-                .AddParameter("InvocationId", invocationId)
-                .InvokeAndClearCommands();
-
-            _hasSetOrchestrationContext = true;
-        }
-
         public void SetOrchestrationContext(OrchestrationContext orchestrationContext)
         {
             _pwsh.AddCommand(SetFunctionInvocationContextCommand)