arm-template-helper-advanced.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "00f0b41b-144f-4b16-93b8-1514ea6dd227",
   "metadata": {},
   "outputs": [],
   "source": [
    "import logging\n",
    "import sys\n",
    "from llama_index.core import VectorStoreIndex, SimpleDirectoryReader, Settings, StorageContext, load_index_from_storage\n",
    "from llama_index.core.embeddings import resolve_embed_model\n",
    "from llama_index.llms.ollama import Ollama\n",
    "\n",
    "logging.basicConfig(stream=sys.stdout, level=logging.ERROR)\n",
    "logging.getLogger().addHandler(logging.StreamHandler(stream=sys.stdout))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "id": "9a3b0abf-b197-4885-a5cb-b8d9e9dcb880",
   "metadata": {},
   "outputs": [],
   "source": [
    "#create index for all files\n",
    "#total of 18 files from azure arm template repo: https://github.com/Azure/azure-quickstart-templates\n",
    "documents = SimpleDirectoryReader(\"data\").load_data()\n",
    "\n",
    "# bge embedding model\n",
    "# no custom parameters has been passed to embedding model\n",
    "Settings.embed_model = resolve_embed_model(\"local:BAAI/bge-small-en-v1.5\")\n",
    "\n",
    "index = VectorStoreIndex.from_documents(\n",
    "    documents,\n",
    ")\n",
    "index.storage_context.persist(persist_dir=\"./indexes\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "id": "07c35f09-3362-4f4c-9a92-489e0b858272",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "INFO:llama_index.core.indices.loading:Loading all indices.\n",
      "Loading all indices.\n",
      "Loading all indices.\n",
      "Loading all indices.\n",
      "INFO:httpx:HTTP Request: POST http://localhost:11434/api/chat \"HTTP/1.1 200 OK\"\n",
      "HTTP Request: POST http://localhost:11434/api/chat \"HTTP/1.1 200 OK\"\n",
      "HTTP Request: POST http://localhost:11434/api/chat \"HTTP/1.1 200 OK\"\n",
      "HTTP Request: POST http://localhost:11434/api/chat \"HTTP/1.1 200 OK\"\n",
      " I cannot directly provide you with a JSON ARM template using the given context information as it would require combining elements from multiple files, potentially introducing changes that are not in the original context. However, I can guide you on how to create a new template by merging the existing ones and making necessary modifications.\n",
      "\n",
      "First, you need to merge both templates into one. You can do this by creating a new file and manually copying the contents of both files into it:\n",
      "\n",
      "```json\n",
      "{\n",
      "  // Include the \"Create a Storage Account\" module metadata\n",
      "  \"_schema\": \"https://aka.ms/azure-quickstart-templates-metadata-schema#\",\n",
      "  \"type\": \"Module\",\n",
      "  ... (rest of the metadata from azure-storage-metadata.json)\n",
      "\n",
      "  // Include the actual deployment template\n",
      "  \"resources\": [\n",
      "    ... (content of the /Users/vishwajeetkumar/work/arm-template-generator/data/azuredeploy-vm.json file)\n",
      "  ]\n",
      "}\n",
      "```\n",
      "\n",
      "Now, to modify the template to include a private endpoint for the storage account:\n",
      "\n",
      "1. Add a new resource in the \"resources\" array for creating a Private Endpoint:\n",
      "\n",
      "```json\n",
      "{\n",
      "  \"apiVersion\": \"2018-05-01\",\n",
      "  \"name\": \"[parameters('storageAccountName')]PrivateEndpoint\",\n",
      "  \"type\": \"Microsoft.Network/privateEndpoints\",\n",
      "  \"properties\": {\n",
      "    \"location\": \"[parameters('location')]\",\n",
      "    \"targetResourceId\": \"[resourceId('Microsoft.Storage/storageAccounts', parameters('storageAccountName'))]\",\n",
      "    \"subnetId\": \"[resourceId('Microsoft.Network/virtualNetworks/', variables('virtualNetworkName'), '/subnets/', variables('subnetName'))]\"\n",
      "  }\n",
      "}\n",
      "```\n",
      "\n",
      "2. Add the following lines inside the \"variables\" section to enable private endpoints for the storage account:\n",
      "\n",
      "```json\n",
      "\"privateEndpointsEnabled\": true,\n",
      "```\n",
      "\n",
      "3. Update the storage account creation with the new property:\n",
      "\n",
      "```json\n",
      "{\n",
      "  \"type\": \"Microsoft.Storage/storageAccounts\",\n",
      "  \"name\": \"[parameters('storageAccountName')]\",\n",
      "  \"apiVersion\": \"2021-01-01\",\n",
      "  \"location\": \"[parameters('location')]\",\n",
      "  \"properties\": {\n",
      "    \"skuName\": \"[parameters('skuName')]\",\n",
      "    \"kind\": \"[if(variables('privateEndpointsEnabled'), 'StorageV3', 'Standard_LRS')]\",\n",
      "    ... (rest of the properties)\n",
      "  }\n",
      "}\n",
      "```\n",
      "\n",
      "This should give you a starting point for creating an ARM template that deploys both a virtual machine and a storage account with private endpoints. However, keep in mind that this might not be the most efficient or complete solution, as there might be additional configurations required depending on your specific use case.\n"
     ]
    }
   ],
   "source": [
    "# ollama\n",
    "Settings.llm = Ollama(model=\"mistral\", request_timeout=300.0)\n",
    "# rebuild storage context\n",
    "storage_context = StorageContext.from_defaults(persist_dir=\"./indexes\")\n",
    "# load index\n",
    "index = load_index_from_storage(storage_context)\n",
    "\n",
    "query_engine = index.as_query_engine()\n",
    "response = query_engine.query(\"Give me a single JSON arm template file to deploy virtual machine and storage account. Storage account should use private endpoints.\")\n",
    "print(response)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "id": "175d3b5a-042c-4350-a55b-a5188439d47f",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "INFO:httpx:HTTP Request: POST http://localhost:11434/api/chat \"HTTP/1.1 200 OK\"\n",
      "HTTP Request: POST http://localhost:11434/api/chat \"HTTP/1.1 200 OK\"\n",
      "HTTP Request: POST http://localhost:11434/api/chat \"HTTP/1.1 200 OK\"\n",
      "HTTP Request: POST http://localhost:11434/api/chat \"HTTP/1.1 200 OK\"\n",
      " I cannot provide you with an exact Arm (Azure Resource Manager) template based on the given context alone, as the provided context consists of separate parts for an Azure deployment template and deployment parameters. An Arm template includes both schema definition and actual resource deployments. Here's an example of what your single JSON Arm template might look like:\n",
      "\n",
      "```json\n",
      "{\n",
      "  \"$schema\": \"https://schema.management.azure.com/schemas/2019-04-01/deploymentTemplate.json#\",\n",
      "  \"contentVersion\": \"1.0.0.0\",\n",
      "  \"parameters\": {\n",
      "    \"location\": {\n",
      "      \"type\": \"string\",\n",
      "      \"defaultValue\": \"[resourceGroup().location]\",\n",
      "      \"metadata\": {\n",
      "        \"description\": \"Specifies the location for all the resources.\"\n",
      "      }\n",
      "    },\n",
      "    \"virtualNetworkName\": {\n",
      "      \"defaultValue\": \"UbuntuVnet\",\n",
      "      \"type\": \"string\",\n",
      "      \"metadata\": {\n",
      "        \"description\": \"Specifies the name of the virtual network hosting the virtual machine and storage account.\"\n",
      "      }\n",
      "    },\n",
      "    \"virtualNetworkAddressPrefix\": {\n",
      "      \"defaultValue\": \"10.0.0.0/16\",\n",
      "      \"type\": \"string\",\n",
      "      \"metadata\": {\n",
      "        \"description\": \"Specifies the address prefix of the virtual network hosting the virtual machine and storage account.\"\n",
      "      }\n",
      "    },\n",
      "    \"subnetName\": {\n",
      "      \"defaultValue\": \"DefaultSubnet\",\n",
      "      \"type\": \"string\",\n",
      "      \"metadata\": {\n",
      "        \"description\": \"Specifies the name of the subnet used for both virtual machine and storage account.\"\n",
      "      }\n",
      "    },\n",
      "    \"subnetAddressPrefix\": {\n",
      "      \"defaultValue\": \"10.0.0.0/24\",\n",
      "      \"type\": \"string\",\n",
      "      \"metadata\": {\n",
      "        \"description\": \"Specifies the address prefix of the subnet used for both virtual machine and storage account.\"\n",
      "      }\n",
      "    },\n",
      "    \"storageAccountName\": {\n",
      "      \"defaultValue\": \"[concat('adls', uniquestring(resourceGroup().id))]\",\n",
      "      \"type\": \"string\",\n",
      "      \"metadata\": {\n",
      "        \"description\": \"Specifies the globally unique DNS Name for the ADLS Gen 2 storage account.\"\n",
      "      }\n",
      "    },\n",
      "    \"blobStorageAccountName\": {\n",
      "      \"defaultValue\": \"[concat('boot', uniquestring(resourceGroup().id))]\",\n",
      "      \"type\": \"string\",\n",
      "      \"metadata\": {\n",
      "        \"description\": \"Specifies the globally unique name for the storage account used to store the boot diagnostics logs of the virtual machine.\"\n",
      "      }\n",
      "    },\n",
      "    \"vmName\": {\n",
      "      \"type\": \"string\",\n",
      "      \"defaultValue\": \"TestVm\",\n",
      "      \"metadata\": {\n",
      "        \"description\": \"Specifies the name of the virtual machine.\"\n",
      "      }\n",
      "    },\n",
      "    \"vmSize\": {\n",
      "      \"type\": \"string\",\n",
      "      \"defaultValue\": \"Standard_D4s_v3\",\n",
      "      \"metadata\": {\n",
      "        \"description\": \"Specifies the size of the virtual machine.\"\n",
      "      }\n",
      "    },\n",
      "    \"imagePublisher\": {\n",
      "      \"type\": \"string\",\n",
      "      \"defaultValue\": \"Canonical\",\n",
      "      \"metadata\": {\n",
      "        \"description\": \"Specifies the image publisher of the disk image used to create the virtual machine.\"\n",
      "      }\n",
      "    },\n",
      "    \"imageOffer\": {\n",
      "      \"type\": \"string\",\n",
      "      \"defaultValue\": \"UbuntuServer\",\n",
      "      \"metadata\": {\n",
      "        \"description\": \"Specifies the offer of the platform image or marketplace image used to create the virtual machine.\"\n",
      "      }\n",
      "    },\n",
      "    \"imageSku\": {\n",
      "      \"type\": \"string\",\n",
      "      \"defaultValue\": \"18.04-LTS\",\n",
      "      \"metadata\": {\n",
      "        \"description\": \"Specifies the Ubuntu version for the VM. This will pick a fully patched image of this given Ubuntu version.\"\n",
      "      }\n",
      "    },\n",
      "    \"authenticationType\": {\n",
      "      \"type\": \"string\",\n",
      "      \"defaultValue\": \"password\",\n",
      "      \"allowedValues\": [\n",
      "        \"sshPublicKey\",\n",
      "        \"password\"\n",
      "      ],\n",
      "      \"metadata\": {\n",
      "        \"description\": \"Specifies the type of authentication when accessing the Virtual Machine. SSH key is recommended.\"\n",
      "      }\n",
      "    },\n",
      "    \"adminUsername\": {\n",
      "      \"type\": \"string\",\n",
      "      \"defaultValue\": \"[parameters('adminUsername')]\",\n",
      "      \"metadata\": {\n",
      "        \"description\": \"Specifies the name of the administrator account of the virtual machine and storage account.\"\n",
      "      }\n",
      "    },\n",
      "    \"adminPasswordOrKey\": {\n",
      "      \"type\": \"securestring\",\n",
      "      \"defaultValue\": \"[parameters('adminPasswordOrKey')]\",\n",
      "      \"metadata\": {\n",
      "        \"description\": \"Specifies the SSH Key or password for the virtual machine and storage account. SSH key is recommended.\"\n",
      "      }\n",
      "    },\n",
      "    \"diskStorageAccounType\": {\n",
      "      \"type\": \"string\",\n",
      "      \"defaultValue\": \"[parameters('diskStorageAccounType')]\",\n",
      "      \"allowedValues\": [\n",
      "        \"Standard_LRS\",\n",
      "        \"Standard_GRS\",\n",
      "        \"Standard_RAGRS\",\n",
      "        \"Standard_ZRS\",\n",
      "        \"Premium_LRS\",\n",
      "        \"Premium_ZRS\",\n",
      "        \"Standard_GZRS\",\n",
      "        \"Standard_RAGZRS\"\n",
      "      ],\n",
      "      \"metadata\": {\n",
      "        \"description\": \"Specifies the storage account type for both OS and data disk.\"\n",
      "      }\n",
      "    },\n",
      "  },\n",
      "  \"variables\": {},\n",
      "  \"resources\": [\n",
      "    {\n",
      "      \"type\": \"Microsoft.Network/virtualNetworks\",\n",
      "      \"apiVersion\": \"2019-04-01\",\n",
      "      \"name\": \"[parameters('virtualNetworkName')]\",\n",
      "      \"location\": \"[parameters('location')]\",\n",
      "      \"properties\": {\n",
      "        \"addressSpace\": {\n",
      "          \"addressPrefixes\": [\n",
      "            {\n",
      "              \"addressPrefix\": \"[parameters('virtualNetworkAddressPrefix')]\"\n",
      "            }\n",
      "          ]\n",
      "        },\n",
      "        \"subnets\": [\n",
      "          {\n",
      "            \"name\": \"[parameters('subnetName')]\",\n",
      "            \"properties\": {\n",
      "              \"addressPrefix\": \"[parameters('subnetAddressPrefix')]\"\n",
      "            }\n",
      "          }\n",
      "        ]\n",
      "      }\n",
      "    },\n",
      "    {\n",
      "      \"type\": \"Microsoft.Storage/storageAccounts\",\n",
      "      \"apiVersion\": \"2019-04-01\",\n",
      "      \"name\": \"[parameters('blobStorageAccountName')]\",\n",
      "      \"location\": \"[parameters('location')]\",\n",
      "      \"kind\": \"StorageV2\",\n",
      "      \"properties\": {\n",
      "        \"sku\": {\n",
      "          \"name\": \"[parameters('diskStorageAccounType')]\"\n",
      "        },\n",
      "        \"networkAcls\": {\n",
      "          \"bypass\": [\n",
      "            {\n",
      "              \"type\": \"Microsoft.Network/virtualNetworks/subnets\",\n",
      "              \"ids\": [\n",
      "                \"[resourceId('Microsoft.Network/virtualNetworks/', parameters('virtualNetworkName'))]/subnets/[parameters('subnetName')]\"\n",
      "              ]\n",
      "            }\n",
      "          ]\n",
      "        },\n",
      "        \"accessPolicy\": [],\n",
      "        \"locationKind\": \"Global\",\n",
      "        \"minimalBLOBServiceVersion\": null,\n",
      "        \"supportsHttpsTrafficOnly\": false\n",
      "      }\n",
      "    }\n",
      "  ],\n",
      "  \"outputs\": {\n",
      "    \"storageAccountOutput\": {\n",
      "      \"type\": \"object\",\n",
      "      \"value\": {\n",
      "        \"name\": \"[parameters('blobStorageAccountName')]\",\n",
      "        \"key\": \"[parameters('adminUsername')]\"\n",
      "      },\n",
      "      \"metadata\": {\n",
      "        \"description\": \"The name and admin username of the created storage account.\"\n",
      "      }\n",
      "    },\n",
      "    \"virtualNetworkOutput\": {\n",
      "      \"type\": \"object\",\n",
      "      \"value\": {\n",
      "        \"id\": \"[resourceId('Microsoft.Network/virtualNetworks/', parameters('virtualNetworkName'))]\"\n",
      "      },\n",
      "      \"metadata\": {\n",
      "        \"description\": \"The ID of the created virtual network.\"\n",
      "      }\n",
      "    }\n",
      "  }\n",
      "}\n",
      "```\n",
      "\n",
      "This is just an example and should be adjusted based on your specific requirements. You might need to change the parameters, variables, resources, or outputs as needed.\n"
     ]
    }
   ],
   "source": [
    "response = query_engine.query(\"Give me a single JSON arm template to deploy storage account and a virtual network. Take default values where required\")\n",
    "print(response)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "id": "bc656d35-41cc-464a-b340-2e2995266040",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "INFO:httpx:HTTP Request: POST http://localhost:11434/api/chat \"HTTP/1.1 200 OK\"\n",
      "HTTP Request: POST http://localhost:11434/api/chat \"HTTP/1.1 200 OK\"\n",
      "HTTP Request: POST http://localhost:11434/api/chat \"HTTP/1.1 200 OK\"\n",
      "HTTP Request: POST http://localhost:11434/api/chat \"HTTP/1.1 200 OK\"\n",
      " {\n",
      "  \"$schema\": \"https://schema.management.azure.com/schemas/2019-04-01/deploymentTemplate.json#\",\n",
      "  \"contentVersion\": \"1.0.0.0\",\n",
      "  \"parameters\": {\n",
      "    \"vmName\": {\n",
      "      \"type\": \"string\",\n",
      "      \"defaultValue\": \"simpleLinuxVM\"\n",
      "    },\n",
      "    \"adminUsername\": {\n",
      "      \"type\": \"string\"\n",
      "    },\n",
      "    \"authenticationType\": {\n",
      "      \"type\": \"string\",\n",
      "      \"defaultValue\": \"password\"\n",
      "    },\n",
      "    \"adminPasswordOrKey\": {\n",
      "      \"type\": \"securestring\"\n",
      "    },\n",
      "    \"dnsLabelPrefix\": {\n",
      "      \"type\": \"string\",\n",
      "      \"defaultValue\": \"[toLower(format('{0}-{1}', parameters('vmName'), uniqueString(resourceGroup().id)))]\"\n",
      "    },\n",
      "    \"ubuntuOSVersion\": {\n",
      "      \"type\": \"string\",\n",
      "      \"defaultValue\": \"Ubuntu-2004\"\n",
      "    },\n",
      "    \"location\": {\n",
      "      \"type\": \"string\",\n",
      "      \"defaultValue\": \"[resourceGroup().location]\"\n",
      "    },\n",
      "    \"vmSize\": {\n",
      "      \"type\": \"string\",\n",
      "      \"defaultValue\": \"Standard_D2s_v3\"\n",
      "    },\n",
      "    \"virtualNetworkName\": {\n",
      "      \"type\": \"string\",\n",
      "      \"defaultValue\": \"vNet\"\n",
      "    },\n",
      "    \"subnetName\": {\n",
      "      \"type\": \"string\",\n",
      "      \"defaultValue\": \"Subnet\"\n",
      "    },\n",
      "    \"networkSecurityGroupName\": {\n",
      "      \"type\": \"string\",\n",
      "      \"defaultValue\": \"SecGroupNet\"\n",
      "    },\n",
      "    \"securityType\": {\n",
      "      \"type\": \"string\",\n",
      "      \"defaultValue\": \"TrustedLaunch\"\n",
      "    }\n",
      "  },\n",
      "  \"variables\": {\n",
      "    \"imageReference\": {\n",
      "      \"Ubuntu-2004\": {\n",
      "        \"publisher\": \"Canonical\",\n",
      "        \"offer\": \"0001-com-ubuntu-server-focal\",\n",
      "        \"sku\": \"20_04-lts-gen2\",\n",
      "        \"version\": \"latest\"\n",
      "      },\n",
      "      \"Ubuntu-2204\": {\n",
      "        \"publisher\": \"Canonical\",\n",
      "        \"offer\": \"0001-com-ubuntu-server-jammy\",\n",
      "        \"sku\": \"22_04-lts-gen2\",\n",
      "        \"version\": \"latest\"\n",
      "      }\n",
      "    },\n",
      "    \"publicIPAddressName\": \"[format('{0}PublicIP', parameters('vmName'))]\",\n",
      "    \"networkInterfaceName\": \"[format('{0}NetInt', parameters('vmName'))]\",\n",
      "    \"osDiskType\": \"Standard_LRS\",\n",
      "    \"subnetAddressPrefix\": \"10.1.0.0/24\",\n",
      "    \"addressPrefix\": \"10.1.0.0/16\"\n",
      "  },\n",
      "  \"resources\": [\n",
      "    {\n",
      "      \"type\": \"Microsoft.Network/publicIPAddresses\",\n",
      "      \"apiVersion\": \"2019-06-01\",\n",
      "      \"name\": \"[variables.publicIPAddressName]\",\n",
      "      \"location\": \"[parameters('location')]\",\n",
      "      \"properties\": {\n",
      "        \"allocationMethod\": \"Dynamic\",\n",
      "        \"dnsLabels\": {\n",
      "          \"[parameters('dnsLabelPrefix')]\": \"\"\n",
      "        }\n",
      "      },\n",
      "      \"dependsOn\": []\n",
      "    },\n",
      "    {\n",
      "      \"type\": \"Microsoft.Network/networkInterfaces\",\n",
      "      \"apiVersion\": \"2019-06-01\",\n",
      "      \"name\": \"[variables.networkInterfaceName]\",\n",
      "      \"location\": \"[parameters('location')]\",\n",
      "      \"dependsOn\": [\n",
      "        \"[variables.publicIPAddressName]\"\n",
      "      ],\n",
      "      \"properties\": {\n",
      "        \"ipConfigurations\": [\n",
      "          {\n",
      "            \"name\": \"ipconfig1\",\n",
      "            \"properties\": {\n",
      "              \"privateIPAddressAllocationMethod\": \"Dynamic\",\n",
      "              \"subnet\": {\n",
      "                \"id\": \"[resourceId('Microsoft.Network/virtualNetworks/{0}/subnets/{1}', parameters('virtualNetworkName'), parameters('subnetName'))]\"\n",
      "              },\n",
      "              \"publicIPAddress\": {\n",
      "                \"id\": \"[resourceId('Microsoft.Network/publicIPAddresses/{0}', variables.publicIPAddressName)]\"\n",
      "              },\n",
      "              \"privateIPAddress\": \"\",\n",
      "              \"primary\": true,\n",
      "              \"publicIPAddressVersion\": \"IPv4\",\n",
      "              \"securityGroup\": {\n",
      "                \"id\": \"[resourceId('Microsoft.Network/networkSecurityGroups/{0}/{1}', parameters('networkSecurityGroupName'), variables.networkSecurityGroupName)]\"\n",
      "              },\n",
      "              \"privateIPAddressZoneNumber\": null,\n",
      "              \"enableIPForwarding\": false\n",
      "            }\n",
      "          }\n",
      "        ],\n",
      "        \"location\": \"[parameters('location')]\",\n",
      "        \"properties\": {\n",
      "          \"macAddress\": \"\",\n",
      "          \"primary\": true,\n",
      "          \"ipConfigurations\": [\n",
      "            {\n",
      "              \"id\": \"[listFirst(variables.networkInterfaceName, 'IpConfiguration[0]').id]\"\n",
      "            }\n",
      "          ],\n",
      "          \"securityGroups\": [\n",
      "            {\n",
      "              \"id\": \"[listFirst(variables.networkSecurityGroupName, 'SecurityGroup').id]\"\n",
      "            }\n",
      "          ],\n",
      "          \"primary\": true\n",
      "        },\n",
      "        \"sku\": {\n",
      "          \"name\": \"VirtualNetwork\",\n",
      "          \"tier\": \"Basic\"\n",
      "        },\n",
      "        \"location\": \"[parameters('location')]\",\n",
      "        \"properties\": {}\n",
      "      }\n",
      "    },\n",
      "    {\n",
      "      \"type\": \"Microsoft.Compute/virtualMachines\",\n",
      "      \"apiVersion\": \"2019-06-01\",\n",
      "      \"name\": \"[parameters('vmName')]\",\n",
      "      \"location\": \"[parameters('location')]\",\n",
      "      \"dependsOn\": [\n",
      "        \"[variables.networkInterfaceName]\"\n",
      "      ],\n",
      "      \"properties\": {\n",
      "        \"hardwareProfile\": {\n",
      "          \"vmSize\": \"[parameters('vmSize')]\"\n",
      "        },\n",
      "        \"osProfile\": {\n",
      "          \"computerName\": \"[parameters('vmName')]\",\n",
      "          \"adminUsername\": \"[parameters('adminUsername')]\",\n",
      "          \"adminPassword\": \"[parameters('authenticationType')=='password' ? parameters('adminPasswordOrKey').value : (parameters('authenticationType')=='sshPublicKey' ? variables.sshPublicKey : 'null')]\",\n",
      "          \"linuxConfiguration\": {\n",
      "            \"disablePasswordAuthentication\": true,\n",
      "            \"ssh\": {\n",
      "              \"publicKeys\": [\n",
      "                {\n",
      "                  \"path\": \"[format('/home/{0}/.ssh/authorized_keys', parameters('adminUsername'))]\"\n",
      "                }\n",
      "              ]\n",
      "            }\n",
      "          },\n",
      "          \"location\": \"[parameters('location')]\",\n",
      "          \"osDisk\": {\n",
      "            \"name\": \"[format('{0}-disk', parameters('vmName'))]\",\n",
      "            \"caching\": \"ReadWrite\",\n",
      "            \"createOption\": \"FromImage\",\n",
      "            \"managedDisk\": {\n",
      "              \"id\": \"[resourceId('Microsoft.Storage/disks/{0}', variables.osDiskType)]\"\n",
      "            },\n",
      "            \"vhdOptions\": {},\n",
      "            \"imageReference\": {\n",
      "              \"publisher\": \"[variables.imageReference[parameters('ubuntuOSVersion')].publisher]\",\n",
      "              \"offer\": \"[variables.imageReference[parameters('ubuntuOSVersion')].offer]\",\n",
      "              \"sku\": \"[variables.imageReference[parameters('ubuntuOSVersion')].sku]\",\n",
      "              \"version\": \"[variables.imageReference[parameters('ubuntuOSVersion')].version]\"\n",
      "            },\n",
      "            \"osType\": \"Linux\",\n",
      "            \"name\": \"[format('{0}-disk', parameters('vmName'))]\"\n",
      "          }\n",
      "        },\n",
      "        \"plan\": {\n",
      "          \"publisher\": \"MicrosoftWindowsServer\",\n",
      "          \"product\": \"WindowsServer\",\n",
      "          \"sku\": \"[parameters('ubuntuOSVersion')=='Ubuntu-2004' ? 'RHEL20DSv3' : 'RHEL22DSv3']\",\n",
      "          \"version\": \"latest\"\n",
      "        },\n",
      "        \"availabilityProfile\": {\n",
      "          \"id\": \"[resourceId('Microsoft.Compute/virtualMachines/{0}/availabilitySets/{1}', parameters('vmName'), variables.availabilitySetName)]\"\n",
      "        },\n",
      "        \"diagnosticsProfile\": {\n",
      "          \"bootDiagnostics\": {\n",
      "            \"enabled\": true,\n",
      "            \"storageUri\": \"\"\n",
      "          }\n",
      "        },\n",
      "        \"networkProfile\": {\n",
      "          \"networkInterfaces\": [\n",
      "            {\n",
      "              \"id\": \"[resourceId('Microsoft.Network/networkInterfaces/{0}/{1}', variables.networkInterfaceName, '[0]')]\"\n",
      "            }\n",
      "          ],\n",
      "          \"primary\": true\n",
      "        },\n",
      "        \"securityProfile\": {},\n",
      "        \"extensionProfile\": {}\n",
      "      },\n",
      "      \"sku\": {\n",
      "        \"name\": \"[parameters('vmSize')]\",\n",
      "        \"tier\": \"Standard\"\n",
      "      },\n",
      "      \"location\": \"[parameters('location')]\"\n",
      "    }\n",
      "  ]\n",
      "}\n"
     ]
    }
   ],
   "source": [
    "response = query_engine.query(\"Give me a JSON arm template to deploy virtual machine. Take default values where required\")\n",
    "print(response)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "68d7f479-f2cb-4ae1-80d8-d87e3aa126a7",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": ".venv",
   "language": "python",
   "name": ".venv"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.12.2"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}