Set Enterprise AI User Budget Policy

export ONA_HOST=https://app.ona.com
export ONA_API_KEY=<your-token>

curl --request POST \
  --url "$ONA_HOST/api/gitpod.v1.BillingService/SetEnterpriseAIUserBudgetPolicy" \
  --header "Authorization: Bearer $ONA_API_KEY" \
  --header "Content-Type: application/json" \
  --data '{
  "mode": "ENTERPRISE_AI_USER_BUDGET_MODE_CREDITS",
  "organizationId": "<organization-id>"
}'

import gitpod.v1.billing_pb2 as billing_pb2
from ona_sdk import create_client_from_env

ona = create_client_from_env()
request = billing_pb2.SetEnterpriseAIUserBudgetPolicyRequest(
    organization_id="<organization-id>",
    mode=billing_pb2.ENTERPRISE_AI_USER_BUDGET_MODE_CREDITS,
)
response = ona.services.billing.set_enterprise_ai_user_budget_policy(request)
print(response)

import { create } from "@bufbuild/protobuf";
import { createClientFromEnv } from "@gitpod/sdk";
import { EnterpriseAIUserBudgetMode, SetEnterpriseAIUserBudgetPolicyRequestSchema } from "@gitpod/sdk/gitpod/v1/billing_pb";

async function main() {
  const ona = createClientFromEnv();
  const request = create(SetEnterpriseAIUserBudgetPolicyRequestSchema, {
    organizationId: "<organization-id>",
    mode: EnterpriseAIUserBudgetMode.CREDITS,
  });
  const response = await ona.services.billing.setEnterpriseAIUserBudgetPolicy(request);
  console.log(response);
}

main().catch(console.error);

package main

import (
	"context"
	"fmt"
	"log"

	"connectrpc.com/connect"
	"github.com/gitpod-io/gitpod-sdk-go/sdk"
	gitpodpb "github.com/gitpod-io/gitpod-sdk-go/v1"
)

func main() {
	ona, err := sdk.NewFromEnv()
	if err != nil {
		log.Fatal(err)
	}

	request := connect.NewRequest(&gitpodpb.SetEnterpriseAIUserBudgetPolicyRequest{
		OrganizationId: "<organization-id>",
		Mode: gitpodpb.EnterpriseAIUserBudgetMode_ENTERPRISE_AI_USER_BUDGET_MODE_CREDITS,
	})
	response, err := ona.Services.Billing.SetEnterpriseAIUserBudgetPolicy(context.Background(), request)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println(response.Msg)
}

{
  "mode": "ENTERPRISE_AI_USER_BUDGET_MODE_CREDITS",
  "organizationId": "<organization-id>"
}