I am trying to use the language model API in my vscode extension to refactor code. However, the LLM sometimes provides incomplete responses. I could really use some help in figuring out why this is happening. The relevant code to the extension that I am using for the LLM is below. This is a snippet from a command I have registered to refactor code.