feat: Update rich html Ray on Vertex Client Context widget

PiperOrigin-RevId: 646173988

Author: matthew29tang

google/cloud/aiplatform/vertex_ray/client_builder.py

@@ -59,14 +59,14 @@ def __init__(
         self.vertex_sdk_version = str(VERTEX_SDK_VERSION)
         self.shell_uri = ray_head_uris.get("RAY_HEAD_NODE_INTERACTIVE_SHELL_URI")
 
-    def _repr_html_(self):
+    def _context_table_template(self):
         shell_uri_row = None
         if self.shell_uri is not None:
             shell_uri_row = VertexRayTemplate("context_shellurirow.html.j2").render(
                 shell_uri=self.shell_uri
             )
 
-        return VertexRayTemplate("context.html.j2").render(
+        return VertexRayTemplate("context_table.html.j2").render(
             python_version=self.python_version,
             ray_version=self.ray_version,
             vertex_sdk_version=self.vertex_sdk_version,

google/cloud/aiplatform/vertex_ray/templates/context.html.j2

@@ -0,0 +1,24 @@
+
+<table class="jp-RenderedHTMLCommon" style="border-collapse: collapse;color: var(--jp-ui-font-color1);font-size: var(--jp-ui-font-size1);">
+    <tr>
+        <td style="text-align: left"><b>Python version:</b></td>
+        <td style="text-align: left"><b>{{ python_version }}</b></td>
+    </tr>
+    <tr>
+        <td style="text-align: left"><b>Ray version:</b></td>
+        <td style="text-align: left"><b> {{ ray_version }}</b></td>
+    </tr>
+    <tr>
+        <td style="text-align: left"><b>Vertex SDK version:</b></td>
+        <td style="text-align: left"><b> {{ vertex_sdk_version }}</b></td>
+    </tr>
+    <tr>
+        <td style="text-align: left"><b>Dashboard:</b></td>
+        <td style="text-align: left"><b><a href="https://{{ dashboard_url }}" target="_blank">{{ dashboard_url }}</a></b></td>
+    </tr>
+    {{ shell_uri_row }}
+    <tr>
+        <td style="text-align: left"><b>Cluster Name:</b></td>
+        <td style="text-align: left"><b> {{ persistent_resource_id }}</b></td>
+    </tr>
+</table>