def _get_model_client()

in orchestration/strategies/base_agent_strategy.py [0:0]

• 17 lines of code
• 1 McCabe index (conditional complexity)

    def _get_model_client(self, response_format=None):
        """
        Set up the configuration for the Azure OpenAI language model client.

        Initializes the `AzureOpenAIChatCompletionClient` with the required settings for
        interaction with Azure OpenAI services.
        """
        token_provider = get_bearer_token_provider(
            ChainedTokenCredential(
                ManagedIdentityCredential(),
                AzureCliCredential()
            ), "https://cognitiveservices.azure.com/.default"
        )
        return AzureOpenAIChatCompletionClient(
            azure_deployment=self.chat_deployment,
            model=self.model,
            azure_endpoint=f"https://{self.aoai_resource}.openai.azure.com",
            azure_ad_token_provider=token_provider,
            api_version=self.api_version,
            temperature=self.temperature,
            max_tokens=self.max_tokens,
            response_format=response_format
        )