def initCosmosClient(): CosmosClient = {
    println(s"Getting CosmosClient")

    val managedIdentityCredential = new ManagedIdentityCredentialBuilder()
    .clientId(miClientId)
    .build()

      val clientAssertionCredential = new ClientAssertionCredentialBuilder()
    .tenantId(tenantId)
    .clientId(appId)
    .clientAssertion(() => {
      try {
        val token = managedIdentityCredential.getTokenSync(new TokenRequestContext().addScopes("api://AzureADTokenExchange/.default"))
        if (token != null){
        token.getToken
    } else {
      ""
    }
    } catch {
      case e: Exception => {
      println(s"Getting CosmosClient with Exception ${e.getMessage}")

      ""
    }
    }
    })
    .additionallyAllowedTenants("*")
    .build()

      new CosmosClientBuilder()
    .endpoint(configs.host)
    .credential(clientAssertionCredential)
    .buildClient()
  }