Fix samples to use the latest func.OrchestrationContext annotation

Author: Hanzhang Zeng (Roger)

samples/external_events/DurableOrchestrationTrigger/__init__.py

@@ -1,5 +1,6 @@
 import logging
 import azure.durable_functions as df
+import azure.functions as func
 import json
 
 def generator_function(context):
@@ -22,12 +23,12 @@ def generator_function(context):
     tasks = []
     for event in json_rule["condition"]["wait_events"]:
         tasks.append(context.wait_for_external_event(event))
-    
+
     if json_rule["condition"]["logic"] == 'and':
         yield context.task_all(tasks)
-    elif json_rule["condition"]["logic"] == 'or': 
+    elif json_rule["condition"]["logic"] == 'or':
         yield context.task_any(tasks)
-    
+
     output = []
     for action in json_rule["satisfied"]:
         result = yield context.call_activity(action["activity_func_name"], json.dumps(action["args"]))
@@ -36,17 +37,17 @@ def generator_function(context):
     return output
 
 
-def main(context: str):
+def main(context: func.OrchestrationContext):
     """This function creates the orchestration and provides
     the durable framework with the core orchestration logic
-    
+
     Arguments:
-        context {str} -- Function context containing the orchestration API's 
+        context {str} -- Function context containing the orchestration API's
         and current context of the long running workflow.
-    
+
     Returns:
         OrchestratorState - State of current orchestration
     """
     orchestrate = df.Orchestrator.create(generator_function)
-    result = orchestrate(context)
+    result = orchestrate(context.body)
     return result

samples/external_events/DurableOrchestrationTrigger/function.json

@@ -4,9 +4,8 @@
     {
       "name": "context",
       "type": "orchestrationTrigger",
-      "direction": "in",
-      "dataType": "string"
+      "direction": "in"
     }
   ],
   "disabled": false
-}
+}

samples/fan_out_fan_in/FanOutFanIn/__init__.py

@@ -1,23 +1,21 @@
 import json
+import azure.functions as func
 import azure.durable_functions as df
 
-
 def generator_function(context):
     activity_count = yield context.call_activity("GetActivityCount", 5)
     activity_list = json.loads(activity_count)
 
     tasks = [context.call_activity("ParrotValue", i) for i in activity_list]
-        
+
     tasks_result = yield context.task_all(tasks)
     values = [int(t) for t in tasks_result]
     message = yield context.call_activity("ShowMeTheSum", values)
 
     return message
 
 
-def main(context: str):
+def main(context: func.OrchestrationContext):
     orchestrate = df.Orchestrator.create(generator_function)
-
-    result = orchestrate(context)
-
+    result = orchestrate(context.body)
     return result

samples/fan_out_fan_in/FanOutFanIn/function.json

@@ -4,9 +4,8 @@
     {
       "name": "context",
       "type": "orchestrationTrigger",
-      "direction": "in",
-      "dataType": "string"
+      "direction": "in"
     }
   ],
   "disabled": false
-}
+}

samples/fan_out_fan_in_tensorflow/FanOutFanIn/__init__.py

@@ -5,7 +5,7 @@
 def _get_classify_images_tasks(config, image_list, context):
     """Get list of tasks that breaks down the execution of the predications.
 
-    will create a list of tasks to perform that is split evenly across the 
+    will create a list of tasks to perform that is split evenly across the
     different instances
 
     Arguments:
@@ -38,7 +38,7 @@ def _get_classify_images_tasks(config, image_list, context):
 def generator_function(context):
     """Get the generator that will need to be orchestrated by durable functions.
 
-    This function will get a list of images to do a prediction of, fan out the 
+    This function will get a list of images to do a prediction of, fan out the
     prediction tasks then summarize the results
 
     Arguments:

samples/fan_out_fan_in_tensorflow/FanOutFanIn/function.json

@@ -4,9 +4,8 @@
     {
       "name": "context",
       "type": "orchestrationTrigger",
-      "direction": "in",
-      "dataType": "string"
+      "direction": "in"
     }
   ],
   "disabled": false
-}
+}

samples/function_chaining/DurableOrchestrationTrigger/__init__.py

@@ -1,47 +1,50 @@
 import logging
+import azure.functions as func
 import azure.durable_functions as df
 
 def generator_function(context):
     """This function provides the core function chaining orchestration logic
-    
+
     Parameters
     ----------
     context : DurableOrchestrationContext
-        This context has the past history 
+        This context has the past history
         and the durable orchestration API's to chain a set of functions
-    
+
     Returns
     -------
     final_result: str
         Returns the final result after the chain completes
-    
+
     Yields
     -------
     call_activity: str
         Yields at every step of the function chain orchestration logic
     """
 
-    # Chained functions - output of a function is passed as 
+    # Chained functions - output of a function is passed as
     # input to the next function in the chain
     r1 = yield context.call_activity("DurableActivity", "One")
     r2 = yield context.call_activity("DurableActivity", r1)
     final_result = yield context.call_activity("DurableActivity", r2)
 
     return final_result
 
-def main(context:str):
+def main(context: func.OrchestrationContext):
     """This function creates the orchestration and provides
     the durable framework with the core orchestration logic
-    
+
     Parameters
     ----------
     context : str
-        Function context containing the orchestration API's 
+        Function context containing the orchestration API's
         and current context of the long running workflow.
-    
+
     Returns:
         OrchestratorState - State of currently running orchestration
     """
     orchestrate = df.Orchestrator.create(generator_function)
-    result = orchestrate(context)
-    return result
+
+    result = orchestrate(context.body)
+
+    return result

samples/function_chaining/DurableOrchestrationTrigger/function.json

@@ -4,9 +4,8 @@
     {
       "name": "context",
       "type": "orchestrationTrigger",
-      "direction": "in",
-      "dataType": "string"
+      "direction": "in"
     }
   ],
   "disabled": false
-}
+}

samples/python_durable_bindings/DurableFanoutOrchestrationTrigger/__init__.py

@@ -1,5 +1,5 @@
 import logging
-
+import azure.functions as func
 import azure.durable_functions as df
 
 
@@ -15,11 +15,11 @@ def generator_function(context):
     return results
 
 
-def main(context: str):
-    logging.warning("Durable Orchestration Trigger: " + context)
+def main(context: func.OrchestrationContext):
+    logging.warning(f"Durable Orchestration Trigger: {context}")
     orchestrate = df.Orchestrator.create(generator_function)
     logging.warning("!!!type(orchestrate) " + str(type(orchestrate)))
-    result = orchestrate(context)
+    result = orchestrate(context.body)
     logging.warning("!!!serialized json : " + result)
     logging.warning("!!!type(result) " + str(type(result)))
     return result

samples/python_durable_bindings/DurableFanoutOrchestrationTrigger/function.json

@@ -4,9 +4,8 @@
       {
         "name": "context",
         "type": "orchestrationTrigger",
-        "direction": "in",
-        "dataType": "string"
+        "direction": "in"
       }
     ],
     "disabled": false
-  }
+  }