removing vscode -based auth for azure

[pelikhan]

• Refactored the Azure token handling, including:

  • Introduced the LanguageModelAuthenticationToken interface for language model authentications.
  • The azure token is now prefixed with Bearer .
  • The method setLanguageModelConnectionToken is added to RuntimeHost, and is implemented in NodeHost to store the token based on the provider.
  • The method isLanguageModelAuthenticationTokenExpired is introduced to check token expiration.
  • On NodeHost, the previously used AuthenticationToken is replaced with LanguageModelAuthenticationToken, and storage of the token is changed to a Record based on provider.
  • Refactored the requesting of Azure Token in getLanguageModelConfiguration based on the new changes.

• Updated the code to set the language model connection token in runScript if provided.

• The server now receives an additional modelToken with script start request and passes it to the client script run.

• The client now sends an additional modelToken with the script start request.

• Updated the VSCode extension to fetch Azure token on-demand and prefix it with Bearer .

• The Language Model Configuration now extends LanguageModelAuthenticationToken.

• The vscode engine is updated from "1.92.0" to "1.93.0".

• Fixed a typo error in throwing an error message in lmaccess.ts and state.ts.

💡 Major updates revolve around handling azure tokens used in Language Model authentications, providing a more structured approach to manage different providers and their tokens. User-facing APIs are unaffected by these changes.

generated by pr-describe

[github-actions]

The MODEL_PROVIDER_AZURE constant is used but not imported in this file.

generated by pr-review-commit missing_import

[github-actions]

The type of _languageModelAuthenticationTokens is not defined. It should be Record<string, LanguageModelAuthenticationToken>.

generated by pr-review-commit missing_type

[github-actions]

The LANGUAGE_MODEL_TOKEN_EXPIRATION_OFFSET constant is used but not imported in this file.

generated by pr-review-commit missing_import

[github-actions]

The changes in the GIT_DIFF:

1. Refactor the AuthenticationToken interface to LanguageModelAuthenticationToken and moved it to the host.ts file.
2. Change the returned token in createAzureToken function to prefix it with Bearer .
3. In NodeHost:
   • A method setLanguageModelConnectionToken is added to set the Language Model token.
   • The _azureToken is updated to _languageModelAuthenticationTokens, a Record of Language Model tokens.
   • In getLanguageModelConfiguration, the Azure token is fetched from _languageModelAuthenticationTokens instead of _azureToken.
4. The runScript function now accepts a modelToken option, and it sets the Language Model token into the runtimeHost.
5. The startServer function now accepts modelToken from data and passes it on to the runScript function.
6. A constant LANGUAGE_MODEL_TOKEN_EXPIRATION_OFFSET is added in constants.ts.
7. A function isLanguageModelAuthenticationTokenExpired is added in host.ts.
8. LanguageModelAuthenticationToken is added as a property in LanguageModelConfiguration interface.
9. The access token in the AzureManager is now prefixed with Bearer , and the account is selected before getting the Session.
10. A modelToken is added in the data in PromptScriptStart interface.
11. modelToken is accepted in the startScript function in WebSocketClient and sent in the request.
12. setLanguageModelConnectionToken is added in TestHost.
13. In state.ts, modelToken is set according to some conditions before calling startScript.
14. In VSCodeHost, the token is prefixed with Bearer if the provider is MODEL_PROVIDER_CLIENT and the modelId provider is MODEL_PROVIDER_AZURE.

Concerns:

1. In AzureManager, the conditions for the QuickPickItem prompt when multiple accounts are found may not cover all edge cases. The user response res could be undefined if the user cancels the prompt, and the account variable would be undefined.

LGTM overall 🚀, but the concern above should be addressed to prevent potential issues.

generated by pr-review