Follow-up: Address non-blocking review comments from chat completions PR #426

[guygir]

This issue tracks follow-up work from review comments on the chat completions preprocessing PR.

These comments are non-blocking and can be addressed in separate PRs after the main chat completions feature merges.

See review comments from @nirrozenbaum on #426 for details.

[guygir]

/assign