Python Basics

Control Flow

Control Flow

Control flow statements let your AI agents make decisions and handle different scenarios.

If/Elif/Else

if condition:
    # do something
elif another_condition:
    # do something else
else:
    # fallback

For Loops

for item in collection:
    process(item)

While Loops

while condition:
    # keep going

Comprehensions

Pythonic one-liners:

# List comprehension
squares = [x**2 for x in range(10)]

# Dict comprehension  
word_lengths = {w: len(w) for w in words}

Try It Yourself

Run this code example to practice what you've learned.

example.py
1# If/elif/else - handling different response types
2response_type = "function_call"
3
4if response_type == "text":
5 print("Processing text response...")
6elif response_type == "function_call":
7 print("Executing function call...")
8elif response_type == "error":
9 print("Handling error...")
10else:
11 print("Unknown response type")
12
13# For loop - processing messages
14messages = [
15 {"role": "user", "content": "Hello"},
16 {"role": "assistant", "content": "Hi there!"},
17 {"role": "user", "content": "How are you?"}
18]
19
20for msg in messages:
21 print(f"{msg['role'].upper()}: {msg['content']}")
22
23# While loop - retry logic
24max_retries = 3
25attempt = 0
26success = False
27
28while attempt < max_retries and not success:
29 attempt += 1
30 print(f"Attempt {attempt}...")
31 # Simulate success on third try
32 if attempt == 3:
33 success = True
34 print("Success!")
35
36# List comprehension - filter user messages
37user_messages = [m["content"] for m in messages if m["role"] == "user"]
38print(f"\nUser messages: {user_messages}")