Add try/except for metadata.version

- TODO: Undo this change
langchain-ai · Feb 5, 2024 · a6a738a · a6a738a
1 parent efb15d7
commit a6a738a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/libs/community/langchain_community/utilities/vertexai.py b/libs/community/langchain_community/utilities/vertexai.py
@@ -53,7 +53,12 @@ def raise_vertex_import_error(minimum_expected_version: str = "1.40.0") -> None:
 
 
 def _get_user_agent(module: Optional[str] = None) -> Tuple[str, str]:
-    langchain_version = metadata.version("langchain")
+    try:
+        langchain_version = metadata.version("langchain-google-vertexai")
+        langchain_version = metadata.version("langchain-core")
+        langchain_version = metadata.version("langchain")
+    except metadata.PackageNotFoundError as ex:
+        print(ex)
     client_library_version = (
         f"{langchain_version}-{module}" if module else langchain_version
     )