address feedback

Author: vdusek

docs/02_concepts/code/01_class_manual.py

@@ -6,18 +6,20 @@
 async def main() -> None:
     await Actor.init()
 
-    # Get input
-    actor_input = await Actor.get_input()
-    Actor.log.info('Actor input: %s', actor_input)
+    try:
+        # Get input
+        actor_input = await Actor.get_input()
+        Actor.log.info('Actor input: %s', actor_input)
 
-    # Your Actor logic here
-    data = {'message': 'Hello from Actor!', 'input': actor_input}
-    await Actor.push_data(data)
+        # Your Actor logic here
+        data = {'message': 'Hello from Actor!', 'input': actor_input}
+        await Actor.push_data(data)
 
-    # Set status message
-    await Actor.set_status_message('Actor completed successfully')
+        # Set status message
+        await Actor.set_status_message('Actor completed successfully')
 
-    await Actor.exit()
+    finally:
+        await Actor.exit()
 
 
 if __name__ == '__main__':

docs/02_concepts/code/01_instance_context.py

@@ -20,7 +20,7 @@ async def main() -> None:
         await actor.push_data(data)
 
         # Set status message
-        await Actor.set_status_message('Actor completed successfully')
+        await actor.set_status_message('Actor completed successfully')
 
 
 if __name__ == '__main__':

docs/02_concepts/code/01_instance_manual.py

@@ -12,18 +12,20 @@ async def main() -> None:
 
     await actor.init()
 
-    # Get input
-    actor_input = await actor.get_input()
-    actor.log.info('Actor input: %s', actor_input)
+    try:
+        # Get input
+        actor_input = await actor.get_input()
+        actor.log.info('Actor input: %s', actor_input)
 
-    # Your Actor logic here
-    data = {'message': 'Hello from Actor!', 'input': actor_input}
-    await actor.push_data(data)
+        # Your Actor logic here
+        data = {'message': 'Hello from Actor!', 'input': actor_input}
+        await actor.push_data(data)
 
-    # Set status message
-    await actor.set_status_message('Actor completed successfully')
+        # Set status message
+        await actor.set_status_message('Actor completed successfully')
 
-    await actor.exit()
+    finally:
+        await actor.exit()
 
 
 if __name__ == '__main__':