import { Photon } from "@photon-ai/api";

const photon = new Photon({
  headers: { Authorization: `Bearer ${process.env.PHOTON_API_TOKEN}` },
});
const result = await photon.auth.updateOrganizationSsoPolicy({
  path: {
    orgId: "orgId",
  },
  body: {
    expectedVersion: "expectedVersion",
    ssoJitEnabled: true,
  },
});

import os

from photon_api import Photon
from photon_api.rpc_generated import UpdateOrganizationSsoPolicyInput

photon = Photon(headers={"Authorization": f"Bearer {os.environ['PHOTON_API_TOKEN']}"})
result = photon.auth.update_organization_sso_policy(
    UpdateOrganizationSsoPolicyInput.model_validate({
        "path": {
            "orgId": "orgId",
        },
        "body": {
            "expectedVersion": "expectedVersion",
            "ssoJitEnabled": True,
        },
    })
)

use photon_ai_api::{Credential, PhotonClientBuilder, SecretString};

let token = std::env::var("PHOTON_API_TOKEN")?;
let client = PhotonClientBuilder::new()
    .credential("accountServiceKey", Credential::Bearer(SecretString::from(token)))
    .build()?;
let result = client
    .update_organization_sso_policy(
        "orgId".to_owned(),
        &serde_json::from_value(serde_json::json!({
            "expectedVersion": "expectedVersion",
            "ssoJitEnabled": true,
        }))?,
    )
    .await
    .map_err(|error| format!("{error:?}"))?
    .into_inner();

{
  "connectionChangedAt": null,
  "connectionGeneration": "string",
  "connectionState": "not_configured",
  "connectionType": "string",
  "desiredSsoEnforced": true,
  "domains": [
    {
      "domain": "string",
      "state": "verified"
    }
  ],
  "organizationId": "string",
  "policySyncStatus": "synced",
  "providerAvailable": true,
  "ssoEnforced": true,
  "ssoJitEnabled": true,
  "version": "string"
}