Fix the issue where the parameters passed to VertexAI ignored langcha…

…in-ai#5889 (langchain-ai#5891) Fixes langchain-ai#5889 and fixes the name of the argument in init_vertexai @hwchase17 @agola11 Co-authored-by: Lior Durahly <lior.durahly@superwise.ai>
AlmyAI · Jun 9, 2023 · 0eb1bc1 · 0eb1bc1
1 parent 63fcf41
commit 0eb1bc1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/langchain/llms/vertexai.py b/langchain/llms/vertexai.py
@@ -68,7 +68,7 @@ def _llm_type(self) -> str:
     @classmethod
     def _try_init_vertexai(cls, values: Dict) -> None:
         allowed_params = ["project", "location", "credentials"]
-        params = {k: v for k, v in values.items() if v in allowed_params}
+        params = {k: v for k, v in values.items() if k in allowed_params}
         init_vertexai(**params)
         return None
 

diff --git a/langchain/utilities/vertexai.py b/langchain/utilities/vertexai.py
@@ -18,7 +18,7 @@ def raise_vertex_import_error() -> None:
 
 
 def init_vertexai(
-    project_id: Optional[str] = None,
+    project: Optional[str] = None,
     location: Optional[str] = None,
     credentials: Optional["Credentials"] = None,
 ) -> None:
@@ -40,7 +40,7 @@ def init_vertexai(
         raise_vertex_import_error()
 
     vertexai.init(
-        project=project_id,
+        project=project,
         location=location,
         credentials=credentials,
     )