github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-25 08:03:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[patch]: langchain_community.vectorstores.azuresearch Raise LangChainException instead of bare Exception (#23935)

Browse Source 
Raise `LangChainException` instead of `Exception`. This alleviates the
need for library users to use bare try/except to handle exceptions
raised by `AzureSearch`.

Co-authored-by: Eugene Yurtsev <eyurtsev@gmail.com>
This commit is contained in:
Marc Gibbons 2024-07-26 11:59:06 -04:00 committed by GitHub
parent 3d16dcd88d
commit cc451effd1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
libs/community/langchain_community/vectorstores/azuresearch.py
View File

@ -29,6 +29,7 @@ from langchain_core.callbacks import (
) )
from langchain_core.documents import Document from langchain_core.documents import Document
from langchain_core.embeddings import Embeddings from langchain_core.embeddings import Embeddings
from langchain_core.exceptions import LangChainException
from langchain_core.pydantic_v1 import root_validator from langchain_core.pydantic_v1 import root_validator
from langchain_core.retrievers import BaseRetriever from langchain_core.retrievers import BaseRetriever
from langchain_core.utils import get_from_env from langchain_core.utils import get_from_env
@ -463,7 +464,7 @@ class AzureSearch(VectorStore):
response = self.client.upload_documents(documents=data) response = self.client.upload_documents(documents=data)
# Check if all documents were successfully uploaded # Check if all documents were successfully uploaded
if not all(r.succeeded for r in response): if not all(r.succeeded for r in response):
raise Exception(response) raise LangChainException(response)
# Reset data # Reset data
data = [] data = []
@ -477,7 +478,7 @@ class AzureSearch(VectorStore):
if all(r.succeeded for r in response): if all(r.succeeded for r in response):
return ids return ids
else: else:
raise Exception(response) raise LangChainException(response)
async def aadd_embeddings( async def aadd_embeddings(
self, self,
@ -521,7 +522,7 @@ class AzureSearch(VectorStore):
response = await async_client.upload_documents(documents=data) response = await async_client.upload_documents(documents=data)
# Check if all documents were successfully uploaded # Check if all documents were successfully uploaded
if not all(r.succeeded for r in response): if not all(r.succeeded for r in response):
raise Exception(response) raise LangChainException(response)
# Reset data # Reset data
data = [] data = []
@ -536,7 +537,7 @@ class AzureSearch(VectorStore):
if all(r.succeeded for r in response): if all(r.succeeded for r in response):
return ids return ids
else: else:
raise Exception(response) raise LangChainException(response)
def delete(self, ids: Optional[List[str]] = None, **kwargs: Any) -> bool: def delete(self, ids: Optional[List[str]] = None, **kwargs: Any) -> bool:
"""Delete by vector ID. """Delete by vector ID.