docs: make StoryFlowAgent example fully async/await (#379)

Author: merdandt

examples/python/snippets/agents/custom-agent/storyflow_agent.py

@@ -219,7 +219,7 @@ async def _run_async_impl(
 # --- Setup Runner and Session ---
 session_service = InMemorySessionService()
 initial_state = {"topic": "a brave kitten exploring a haunted house"}
-session = session_service.create_session(
+session = await session_service.create_session(
     app_name=APP_NAME,
     user_id=USER_ID,
     session_id=SESSION_ID,
@@ -234,12 +234,12 @@ async def _run_async_impl(
 )
 
 # --- Function to Interact with the Agent ---
-def call_agent(user_input_topic: str):
+async def call_agent(user_input_topic: str):
     """
     Sends a new topic to the agent (overwriting the initial one if needed)
     and runs the workflow.
     """
-    current_session = session_service.get_session(app_name=APP_NAME, 
+    current_session = await session_service.get_session(app_name=APP_NAME, 
                                                   user_id=USER_ID, 
                                                   session_id=SESSION_ID)
     if not current_session:
@@ -261,7 +261,7 @@ def call_agent(user_input_topic: str):
     print("\n--- Agent Interaction Result ---")
     print("Agent Final Response: ", final_response)
 
-    final_session = session_service.get_session(app_name=APP_NAME, 
+    final_session = await session_service.get_session(app_name=APP_NAME, 
                                                 user_id=USER_ID, 
                                                 session_id=SESSION_ID)
     print("Final Session State:")
@@ -270,5 +270,5 @@ def call_agent(user_input_topic: str):
     print("-------------------------------\n")
 
 # --- Run the Agent ---
-call_agent("a lonely robot finding a friend in a junkyard")
+await call_agent("a lonely robot finding a friend in a junkyard")
 # --8<-- [end:story_flow_agent]