Change tool_context_manager context location

libs/vertexai/langchain_google_vertexai/_base.py

@@ -27,6 +27,7 @@
 from langchain_google_vertexai._enums import HarmBlockThreshold, HarmCategory
 from langchain_google_vertexai._utils import (
     get_client_info,
+    get_user_agent,
     is_codey_model,
     is_gemini_model,
 )
@@ -142,6 +143,12 @@ def _default_params(self) -> Dict[str, Any]:
                 )
         return updated_params
 
+    @property
+    def _user_agent(self) -> str:
+        """Gets the User Agent."""
+        _, user_agent = get_user_agent(f"{type(self).__name__}_{self.model_name}")
+        return user_agent
+
     @classmethod
     def _init_vertexai(cls, values: Dict) -> None:
         vertexai.init(

libs/vertexai/langchain_google_vertexai/_utils.py

@@ -66,7 +66,7 @@ def get_user_agent(module: Optional[str] = None) -> Tuple[str, str]:
         module (Optional[str]):
             Optional. The module for a custom user agent header.
     Returns:
-        google.api_core.gapic_v1.client_info.ClientInfo
+        Tuple[str, str]
     """
     try:
         langchain_version = metadata.version("langchain")
@@ -95,7 +95,7 @@ def get_client_info(module: Optional[str] = None) -> "ClientInfo":
 
 
 def load_image_from_gcs(path: str, project: Optional[str] = None) -> Image:
-    """Loads im Image from GCS."""
+    """Loads an Image from GCS."""
     gcs_client = storage.Client(project=project)
     pieces = path.split("/")
     blobs = list(gcs_client.list_blobs(pieces[2], prefix="/".join(pieces[3:])))