Azure Mgmt Weightsandbiases Dotnet

using Azure.Identity;using Azure.ResourceManager;using Azure.ResourceManager.WeightsAndBiases;
// Local dev: DefaultAzureCredential. Production: set AZURE_TOKEN_CREDENTIALS=prod or AZURE_TOKEN_CREDENTIALS=<specific_credential>var credential = new DefaultAzureCredential(    DefaultAzureCredential.DefaultEnvironmentVariableName);// Or use a specific credential directly in production:// See https://learn.microsoft.com/dotnet/api/overview/azure/identity-readme?view=azure-dotnet#credential-classes// var credential = new ManagedIdentityCredential();ArmClient client = new ArmClient(credential);

using Azure.ResourceManager.WeightsAndBiases;using Azure.ResourceManager.WeightsAndBiases.Models;
ResourceGroupResource resourceGroup = await client    .GetDefaultSubscriptionAsync()    .Result    .GetResourceGroupAsync("my-resource-group");
WeightsAndBiasesInstanceCollection instances = resourceGroup.GetWeightsAndBiasesInstances();
WeightsAndBiasesInstanceData data = new WeightsAndBiasesInstanceData(AzureLocation.EastUS){    Properties = new WeightsAndBiasesInstanceProperties    {        // Marketplace configuration        Marketplace = new WeightsAndBiasesMarketplaceDetails        {            SubscriptionId = "<marketplace-subscription-id>",            OfferDetails = new WeightsAndBiasesOfferDetails            {                PublisherId = "wandb",                OfferId = "wandb-pay-as-you-go",                PlanId = "wandb-payg",                PlanName = "Pay As You Go",                TermId = "monthly",                TermUnit = "P1M"            }        },        // Admin user        User = new WeightsAndBiasesUserDetails        {            FirstName = "Admin",            LastName = "User",            EmailAddress = "[email protected]",            Upn = "[email protected]"        },        // W&B-specific configuration        PartnerProperties = new WeightsAndBiasesPartnerProperties        {            Region = WeightsAndBiasesRegion.EastUS,            Subdomain = "my-company-wandb"        }    },    // Optional: Enable managed identity    Identity = new ManagedServiceIdentity(ManagedServiceIdentityType.SystemAssigned)};
ArmOperation<WeightsAndBiasesInstanceResource> operation = await instances    .CreateOrUpdateAsync(WaitUntil.Completed, "my-wandb-instance", data);
WeightsAndBiasesInstanceResource instance = operation.Value;
Console.WriteLine($"W&B Instance created: {instance.Data.Name}");Console.WriteLine($"Provisioning state: {instance.Data.Properties.ProvisioningState}");

WeightsAndBiasesInstanceResource instance = await resourceGroup    .GetWeightsAndBiasesInstanceAsync("my-wandb-instance");
Console.WriteLine($"Instance: {instance.Data.Name}");Console.WriteLine($"Location: {instance.Data.Location}");Console.WriteLine($"State: {instance.Data.Properties.ProvisioningState}");
if (instance.Data.Properties.PartnerProperties != null){    Console.WriteLine($"Region: {instance.Data.Properties.PartnerProperties.Region}");    Console.WriteLine($"Subdomain: {instance.Data.Properties.PartnerProperties.Subdomain}");}

// List in resource groupawait foreach (WeightsAndBiasesInstanceResource instance in     resourceGroup.GetWeightsAndBiasesInstances()){    Console.WriteLine($"Instance: {instance.Data.Name}");    Console.WriteLine($"  Location: {instance.Data.Location}");    Console.WriteLine($"  State: {instance.Data.Properties.ProvisioningState}");}
// List in subscriptionSubscriptionResource subscription = await client.GetDefaultSubscriptionAsync();await foreach (WeightsAndBiasesInstanceResource instance in     subscription.GetWeightsAndBiasesInstancesAsync()){    Console.WriteLine($"{instance.Data.Name} in {instance.Id.ResourceGroupName}");}

WeightsAndBiasesInstanceResource instance = await resourceGroup    .GetWeightsAndBiasesInstanceAsync("my-wandb-instance");
// Update with SSO configurationWeightsAndBiasesInstanceData updateData = instance.Data;
updateData.Properties.SingleSignOnPropertiesV2 = new WeightsAndBiasSingleSignOnPropertiesV2{    Type = WeightsAndBiasSingleSignOnType.Saml,    State = WeightsAndBiasSingleSignOnState.Enable,    EnterpriseAppId = "<entra-app-id>",    AadDomains = { "example.com", "contoso.com" }};
ArmOperation<WeightsAndBiasesInstanceResource> operation = await resourceGroup    .GetWeightsAndBiasesInstances()    .CreateOrUpdateAsync(WaitUntil.Completed, "my-wandb-instance", updateData);

WeightsAndBiasesInstanceResource instance = await resourceGroup    .GetWeightsAndBiasesInstanceAsync("my-wandb-instance");
// Update tagsWeightsAndBiasesInstancePatch patch = new WeightsAndBiasesInstancePatch{    Tags =    {        { "environment", "production" },        { "team", "ml-platform" },        { "costCenter", "CC-ML-001" }    }};
instance = await instance.UpdateAsync(patch);Console.WriteLine($"Updated instance: {instance.Data.Name}");

WeightsAndBiasesInstanceResource instance = await resourceGroup    .GetWeightsAndBiasesInstanceAsync("my-wandb-instance");
await instance.DeleteAsync(WaitUntil.Completed);Console.WriteLine("Instance deleted");

// Check if name is available before creating// (Implement via direct ARM call if SDK doesn't expose this)try{    await resourceGroup.GetWeightsAndBiasesInstanceAsync("desired-name");    Console.WriteLine("Name is already taken");}catch (RequestFailedException ex) when (ex.Status == 404){    Console.WriteLine("Name is available");}

using Azure;
try{    ArmOperation<WeightsAndBiasesInstanceResource> operation = await instances        .CreateOrUpdateAsync(WaitUntil.Completed, "my-wandb", data);}catch (RequestFailedException ex) when (ex.Status == 409){    Console.WriteLine("Instance already exists or name conflict");}catch (RequestFailedException ex) when (ex.Status == 400){    Console.WriteLine($"Invalid configuration: {ex.Message}");}catch (RequestFailedException ex){    Console.WriteLine($"Azure error: {ex.Status} - {ex.Message}");}