try/except

googleapis · Sep 9, 2024 · 563ad77 · 563ad77
1 parent c59cb49
commit 563ad77
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/src/langchain_google_alloydb_pg/vectorstore.py b/src/langchain_google_alloydb_pg/vectorstore.py
@@ -376,9 +376,9 @@ async def aadd_images(
         if hasattr(self.embedding_service, "embed_image"):
             try:
                 embeddings = self.embedding_service.embed_image(images)
-            except TypeError as e:
-                raise TypeError(
-                    f"Embedding model used does not support a list of image URIs as input. {str(e)}"
+            except Exception as e:
+                raise Exception(
+                    f"Make sure your selected embedding model supports list of image URIs as input. {str(e)}"
                 )
         else:
             raise ValueError(
@@ -700,9 +700,9 @@ def _embedding_helper(
             if hasattr(self.embedding_service, "embed_image"):
                 try:
                     embedding = self.embedding_service.embed_image([image_uri])[0]
-                except TypeError as e:
-                    raise TypeError(
-                        f"Embedding model used does not support a list of image URIs as input. {str(e)}"
+                except Exception as e:
+                    raise Exception(
+                        f"Make sure your selected embedding model supports list of image URIs as input. {str(e)}"
                     )
             else:
                 raise ValueError(
@@ -831,7 +831,7 @@ async def _asimilarity_search_with_relevance_scores(  # type: ignore[override]
             k: Number of Documents to return. Defaults to 4.
             **kwargs: kwargs to be passed to similarity search. Should include:
                 score_threshold: Optional, a floating point value between 0 to 1 to
-                    filter the resulting set of retrieved docs.
+                filter the resulting set of retrieved docs.
 
         Returns:
             List of Tuples of (doc, similarity_score)
@@ -860,7 +860,7 @@ def _similarity_search_with_relevance_scores(  # type: ignore[override]
             k: Number of Documents to return. Defaults to 4.
             **kwargs: kwargs to be passed to similarity search. Should include:
                 score_threshold: Optional, a floating point value between 0 to 1 to
-                    filter the resulting set of retrieved docs
+                filter the resulting set of retrieved docs
 
         Returns:
             List of Tuples of (doc, similarity_score)
@@ -887,7 +887,7 @@ def similarity_search_with_relevance_scores(  # type: ignore[override]
             k: Number of Documents to return. Defaults to 4.
             **kwargs: kwargs to be passed to similarity search. Should include:
                 score_threshold: Optional, a floating point value between 0 to 1 to
-                    filter the resulting set of retrieved docs.
+                filter the resulting set of retrieved docs.
 
         Returns:
             List of Tuples of (doc, similarity_score).
@@ -935,7 +935,7 @@ async def asimilarity_search_with_relevance_scores(  # type: ignore[override]
             k: Number of Documents to return. Defaults to 4.
             **kwargs: kwargs to be passed to similarity search. Should include:
                 score_threshold: Optional, a floating point value between 0 to 1 to
-                    filter the resulting set of retrieved docs
+                filter the resulting set of retrieved docs
 
         Returns:
             List of Tuples of (doc, similarity_score)