This lesson's interactive features are locked, please to keep using them

User Input

We've hard-coded the prompt that goes to the LLM, which is... not very useful. Let's update our code to accept the prompt as a command-line argument.

We don't want our users to have to edit the code to change the prompt!

Assignment

We'll use Python's built-in argparse module to handle command-line arguments. Start by importing that.
The way argparse works is that we create a parser object, define the arguments we want to accept, and then parse whatever arguments the user actually provided when they ran the script. See the example code below; you may want to customize the description, argument name, help message, etc. But the idea is that we're telling the argument parser to expect a single positional argument, i.e., the user-provided prompt.
```
parser = argparse.ArgumentParser(description="Chatbot")
parser.add_argument("user_prompt", type=str, help="User prompt")
args = parser.parse_args()
# Now we can access `args.user_prompt`
```
Update your code so that the user message's content is set to the user's prompt (args.user_prompt).

Try running your script again while passing an argument, e.g.,

uv run main.py "Why are episodes 7-9 so much worse than 1-6? Use one paragraph."

Also try running the script without an argument. Notice that argparse automatically halts the script and prints a helpful error message (with an exit code of 2).

The remainder of the script can remain unchanged for now.

If things seem to be working correctly, submit the CLI tests.