LSP highlights wrong argument when using a call as argument #2500

Describe the Bug

In the following code, typing foo('a', will highlight the current argument as b. However, typing foo(max('a', 'b'), will highlight the argument c. It seems like pyrefly simply counts the , instead of actual arguments.

Working example when using simple arguments:

Highlighting the wrong argument c:

Sandbox Link

No response

(Only applicable for extension issues) IDE Information

No response