add update and image tools to compute server (oracle#14)

Signed-off-by: Richard Gebhardt <[email protected]>

Author: gebhardtr

oci_compute.py

@@ -1,5 +1,6 @@
 import os
 from logging import Logger
+from typing import Annotated
 
 import oci
 from fastmcp import FastMCP
@@ -44,9 +45,9 @@ def launch_instance(
     compartment_id: str,
     display_name: str,
     availability_domain: str,
-    shape: str,
     image_id: str,
     subnet_id: str,
+    shape: Annotated[str, "Instance shape"] = "VM.Standard.A1.Flex",
 ):
     compute = get_compute_client()
     launch_details = oci.core.models.LaunchInstanceDetails(
@@ -81,13 +82,86 @@ def get_instance(instance_id: str):
 #    }
 
 
+@mcp.tool
+def get_image(image_id: str):
+    compute = get_compute_client()
+    return compute.get_image(image_id).data
+
+
+@mcp.tool
+def list_images(compartment_id: str, operating_system: str = None):
+    """List images, optionally filtered by operating system"""
+    compute = get_compute_client()
+    images = compute.list_images(compartment_id).data
+
+    if operating_system:
+        images = [img for img in images if img.operating_system == operating_system]
+
+    return [
+        {
+            "id": img.id,
+            "display_name": img.display_name,
+            "operating_system": img.operating_system,
+            "operating_system_version": img.operating_system_version,
+        }
+        for img in images
+    ]
+
+
 @mcp.tool
 def instance_action(instance_id: str, action: str):
     compute = get_compute_client()
     response = compute.instance_action(instance_id, action)
     return {"status": action, "opc_request_id": response.headers.get("opc-request-id")}
 
 
+@mcp.tool
+def update_instance_details(
+    instance_id: str,
+    ocpus: Annotated[int, "Number of CPUs allocated to the instance"],
+    memory_in_gbs: Annotated[
+        int, "Amount of memory in gigabytes (GB) allocated to the instance"
+    ],
+):
+    """Update instance details; this may restart the instance so warn the user"""
+    compute_client = get_compute_client()
+
+    shape_config_details = oci.core.models.UpdateInstanceShapeConfigDetails(
+        ocpus=ocpus, memory_in_gbs=memory_in_gbs
+    )
+
+    update_instance_details = oci.core.models.UpdateInstanceDetails(
+        shape_config=shape_config_details
+    )
+
+    try:
+        compute_client.update_instance(
+            instance_id=instance_id, update_instance_details=update_instance_details
+        )
+
+        get_response = compute_client.get_instance(instance_id)
+
+        final_response = oci.wait_until(
+            client=compute_client,
+            response=get_response,
+            evaluate_response=lambda x: x.data.shape_config.ocpus == ocpus
+            and x.data.shape_config.memory_in_gbs == memory_in_gbs,
+            max_interval_seconds=5,
+            max_wait_seconds=240,
+        )
+
+        return {
+            "instance_id": instance_id,
+            "ocpus": final_response.data.shape_config.ocpus,
+            "memory_in_gbs": final_response.data.shape_config.memory_in_gbs,
+        }
+    except oci.exceptions.ServiceError as e:
+        return {
+            "error_code": e.code,
+            "message": e.message,
+        }
+
+
 if __name__ == "__main__":
 
     # MCP spec: OpenAPI exposed at /openapi.json, native MCP at /mcp/v1