Skip to content

Commit 4b229d1

Browse files
authored
Add input validation and type conversion for user input (#1476)
1 parent cec3a9a commit 4b229d1

File tree

1 file changed

+10
-5
lines changed

1 file changed

+10
-5
lines changed

examples/basic/lifecycle_example.py

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -88,11 +88,16 @@ class FinalResult(BaseModel):
8888

8989
async def main() -> None:
9090
user_input = input("Enter a max number: ")
91-
await Runner.run(
92-
start_agent,
93-
hooks=hooks,
94-
input=f"Generate a random number between 0 and {user_input}.",
95-
)
91+
try:
92+
max_number = int(user_input)
93+
await Runner.run(
94+
start_agent,
95+
hooks=hooks,
96+
input=f"Generate a random number between 0 and {max_number}.",
97+
)
98+
except ValueError:
99+
print("Please enter a valid integer.")
100+
return
96101

97102
print("Done!")
98103

0 commit comments

Comments
 (0)