Sure, try it. It's harder than you think. It's not just imports, it's the entire program.

> you're suggesting there's some kind of information flow from the anticipated body of the script back up to the imports

Yes, I am suggesting this. I don't think it is possible to write programs without either anticipating what you're going to write down below before you get there, or else being able to go back and edit what you already wrote.

Of course agent harnesses allow the latter, but raw models without a harness can still do an exceptionally, superhumanly, good job of straight-line programming with no editing.

> Infer imports from context, infer body from context + imports. All strictly causal.

Of course it's causal, that's kind of a reductive way to look at it.

Just infer the entire program from context and then type what you inferred.