diff --git a/examples/basic/lifecycle_example.py b/examples/basic/lifecycle_example.py index 02ce449f4..c31ec5e12 100644 --- a/examples/basic/lifecycle_example.py +++ b/examples/basic/lifecycle_example.py @@ -88,11 +88,16 @@ class FinalResult(BaseModel): async def main() -> None: user_input = input("Enter a max number: ") - await Runner.run( - start_agent, - hooks=hooks, - input=f"Generate a random number between 0 and {user_input}.", - ) + try: + max_number = int(user_input) + await Runner.run( + start_agent, + hooks=hooks, + input=f"Generate a random number between 0 and {max_number}.", + ) + except ValueError: + print("Please enter a valid integer.") + return print("Done!")