feat: add logging for agent engine creation

PiperOrigin-RevId: 776644637

Author: shawn-yang-google

vertexai/_genai/agent_engines.py

@@ -920,6 +920,7 @@ def _RetrieveMemoriesResponse_from_vertex(
 
 
 class AgentEngines(_api_module.BaseModule):
+
     def _create(
         self, *, config: Optional[types.CreateAgentEngineConfigOrDict] = None
     ) -> types.AgentEngineOperation:
@@ -1988,12 +1989,20 @@ def create(
             env_vars=config.env_vars,
         )
         operation = self._create(config=api_config)
+        logger.info(
+            "View progress and logs at"
+            f" https://console.cloud.google.com/logs/query?project={self._api_client.project}."
+        )
         operation = self._await_operation(operation_name=operation.name)
         agent = types.AgentEngine(
             api_client=self,
             api_async_client=AsyncAgentEngines(api_client_=self._api_client),
             api_resource=operation.response,
         )
+        logger.info("Agent Engine created. To use it in another session:")
+        logger.info(
+            "agent_engine =" f" client.agent_engines.get('{agent.api_resource.name}')"
+        )
         if agent_engine is not None:
             # If the user did not provide an agent_engine (e.g. lightweight
             # provisioning), it will not have any API methods registered.
@@ -2082,10 +2091,9 @@ def _create_config(
                 )
             agent_engine_spec = {"package_spec": package_spec}
             if env_vars is not None:
-                (
-                    deployment_spec,
-                    deployment_update_masks,
-                ) = self._generate_deployment_spec_or_raise(env_vars=env_vars)
+                deployment_spec, deployment_update_masks = (
+                    self._generate_deployment_spec_or_raise(env_vars=env_vars)
+                )
                 update_masks.extend(deployment_update_masks)
                 agent_engine_spec["deployment_spec"] = deployment_spec
             class_methods = _agent_engines._generate_class_methods_spec_or_raise(
@@ -2515,6 +2523,7 @@ def retrieve_memories(
 
 
 class AsyncAgentEngines(_api_module.BaseModule):
+
     async def _create(
         self, *, config: Optional[types.CreateAgentEngineConfigOrDict] = None
     ) -> types.AgentEngineOperation: