test_azure_embedding_max_retries_0

This commit is contained in:
Ishaan Jaff 2025-03-18 12:35:34 -07:00
parent b316911120
commit 80a5cfa01d
2 changed files with 0 additions and 22 deletions

View file

@ -12,18 +12,6 @@ from ..common_utils import AzureOpenAIError, BaseAzureLLM
openai_text_completion_config = OpenAITextCompletionConfig()
def select_azure_base_url_or_endpoint(azure_client_params: dict):
azure_endpoint = azure_client_params.get("azure_endpoint", None)
if azure_endpoint is not None:
# see : https://github.com/openai/openai-python/blob/3d61ed42aba652b547029095a7eb269ad4e1e957/src/openai/lib/azure.py#L192
if "/openai/deployments" in azure_endpoint:
# this is base_url, not an azure_endpoint
azure_client_params["base_url"] = azure_endpoint
azure_client_params.pop("azure_endpoint")
return azure_client_params
class AzureTextCompletion(BaseAzureLLM):
def __init__(self) -> None:
super().__init__()