From 20a715a103e9a048bf91414f4a68fca1493f04a3 Mon Sep 17 00:00:00 2001 From: Ashvin Date: Mon, 6 Jan 2025 23:28:16 +0530 Subject: [PATCH] community: Fix redundancy in code. (#29022) In my previous PR (#28953), I added an unwanted condition for validating the Azure ML Endpoint. In this PR, I have rectified the issue. --- libs/community/langchain_community/llms/azureml_endpoint.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/libs/community/langchain_community/llms/azureml_endpoint.py b/libs/community/langchain_community/llms/azureml_endpoint.py index bfe61fdbc5f..c09ea2edcf4 100644 --- a/libs/community/langchain_community/llms/azureml_endpoint.py +++ b/libs/community/langchain_community/llms/azureml_endpoint.py @@ -434,7 +434,7 @@ class AzureMLBaseEndpoint(BaseModel): raise ValueError( "`endpoint_url` should contain the full invocation URL including " "`/score` for `endpoint_api_type='dedicated'` or `/completions` " - "or `/chat/completions` or `/models/chat/completions` " + "or `/models/chat/completions` " "for `endpoint_api_type='serverless'`" ) return field_value @@ -456,18 +456,16 @@ class AzureMLBaseEndpoint(BaseModel): "Endpoints of type `dedicated` should follow the format " "`https://..inference.ml.azure.com/score`." " If your endpoint URL ends with `/completions` or" - "`/chat/completions` or `/models/chat/completions`," + "`/models/chat/completions`," "use `endpoint_api_type='serverless'` instead." ) if field_value == AzureMLEndpointApiType.serverless and not ( endpoint_url.endswith("/completions") # type: ignore[union-attr] - or endpoint_url.endswith("/chat/completions") # type: ignore[union-attr] or endpoint_url.endswith("/models/chat/completions") # type: ignore[union-attr] ): raise ValueError( "Endpoints of type `serverless` should follow the format " "`https://..inference.ml.azure.com/completions`" - " or `https://..inference.ml.azure.com/chat/completions`" " or `https://..inference.ml.azure.com/models/chat/completions`" )