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