github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-08 12:31:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

community: Fix redundancy in code. (#29022)

Browse Source 
In my previous PR (#28953), I added an unwanted condition for validating
the Azure ML Endpoint. In this PR, I have rectified the issue.
This commit is contained in:
Ashvin 2025-01-06 23:28:16 +05:30 committed by GitHub
parent c8d6f9d52b
commit 20a715a103
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libs/community/langchain_community/llms/azureml_endpoint.py
View File

@ -434,7 +434,7 @@ class AzureMLBaseEndpoint(BaseModel):
raise ValueError( raise ValueError(
"`endpoint_url` should contain the full invocation URL including " "`endpoint_url` should contain the full invocation URL including "
"`/score` for `endpoint_api_type='dedicated'` or `/completions` " "`/score` for `endpoint_api_type='dedicated'` or `/completions` "
"or `/chat/completions` or `/models/chat/completions` " "or `/models/chat/completions` "
"for `endpoint_api_type='serverless'`" "for `endpoint_api_type='serverless'`"
) )
return field_value return field_value
@ -456,18 +456,16 @@ class AzureMLBaseEndpoint(BaseModel):
"Endpoints of type `dedicated` should follow the format " "Endpoints of type `dedicated` should follow the format "
"`https://<your-endpoint>.<your_region>.inference.ml.azure.com/score`." "`https://<your-endpoint>.<your_region>.inference.ml.azure.com/score`."
" If your endpoint URL ends with `/completions` or" " If your endpoint URL ends with `/completions` or"
"`/chat/completions` or `/models/chat/completions`," "`/models/chat/completions`,"
"use `endpoint_api_type='serverless'` instead." "use `endpoint_api_type='serverless'` instead."
) )
if field_value == AzureMLEndpointApiType.serverless and not ( if field_value == AzureMLEndpointApiType.serverless and not (
endpoint_url.endswith("/completions") # type: ignore[union-attr] 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] or endpoint_url.endswith("/models/chat/completions") # type: ignore[union-attr]
): ):
raise ValueError( raise ValueError(
"Endpoints of type `serverless` should follow the format " "Endpoints of type `serverless` should follow the format "
"`https://<your-endpoint>.<your_region>.inference.ml.azure.com/completions`" "`https://<your-endpoint>.<your_region>.inference.ml.azure.com/completions`"
" or `https://<your-endpoint>.<your_region>.inference.ml.azure.com/chat/completions`"
" or `https://<your-endpoint>.<your_region>.inference.ml.azure.com/models/chat/completions`" " or `https://<your-endpoint>.<your_region>.inference.ml.azure.com/models/chat/completions`"
) )