Python SDK: Migrate from V1 to V2

1
def token_provider():
2
    global bearer_token
3
    if not is_expired(bearer_token):
4
        return bearer_token
5
    bearer_token, _ = generate_bearer_token('<YOUR_CREDENTIALS_FILE_PATH>')
6
    return bearer_token

1
# Option 1: API key (recommended)
2
credentials = {
3
    'api_key': '<YOUR_API_KEY>',
4
}
5
6
# Option 2: Environment variable — set SKYFLOW_CREDENTIALS in your environment
7
8
# Option 3: Credentials file
9
credentials = {
10
    'path': '<PATH_TO_CREDENTIALS_JSON>',
11
}
12
13
# Option 4: Stringified JSON
14
credentials = {
15
    'credentials_string': '<YOUR_CREDENTIALS_STRING>',
16
}
17
18
# Option 5: Bearer token
19
credentials = {
20
    'token': '<YOUR_BEARER_TOKEN>',
21
}

1
config = Configuration('<VAULT_ID>', '<VAULT_URL>', token_provider)
2
client = Client(config)

1
client = (
2
    Skyflow.builder()
3
    .add_vault_config({
4
        'vault_id': '<VAULT_ID>',
5
        'cluster_id': '<CLUSTER_ID>', # ID from your vault URL e.g., https://{clusterId}.vault.skyflowapis.com
6
        'env': Env.PROD,
7
        'credentials': credentials
8
    })
9
    .add_skyflow_credentials(credentials) # Used if no individual vault credentials are set
10
    .set_log_level(LogLevel.INFO)
11
    .build()
12
)

1
client.insert(
2
    {
3
        "records": [
4
            {
5
                "table": "cards",
6
                "fields": {
7
                    "cardNumber": "41111111111",
8
                    "cvv": "123",
9
                },
10
            }
11
        ]
12
    },
13
    InsertOptions(True),
14
)

1
from skyflow.vault.data import InsertRequest
2
3
insert_data = [
4
    {
5
        'card_number': '<VALUE1>',
6
        'cvv': '<VALUE2>',
7
    },
8
]
9
10
insert_request = InsertRequest(
11
    table='<SENSITIVE_DATA_TABLE>',
12
    values=insert_data,
13
    return_tokens=True,
14
    continue_on_error=True
15
)
16
17
response = skyflow_client.vault('<VAULT_ID>').insert(insert_request)

1
InsertResponse(
2
    inserted_fields=[
3
        {
4
            'skyflow_id': 'a8f3ed5d-55eb-4f32-bf7e-2dbf4b9d9097',
5
            'card_number': '5479-4229-4622-1393'
6
        }
7
    ],
8
    errors=[]
9
)

1
options = InsertOptions(
2
    tokens=True
3
)

1
insert_request = InsertRequest(
2
    table=table_name,
3
    values=insert_data,
4
    return_tokens=False,
5
    continue_on_error=False,
6
    upsert='<UPSERT_COLUMN>',
7
    token_mode=TokenMode.DISABLE,
8
    tokens='<TOKENS>'
9
)

1
{
2
    "http_status": "<http_status>",
3
    "grpc_code": "<grpc_code>",
4
    "http_code": "<http_code>",
5
    "message": "<message>",
6
    "request_id": "<request_id>",
7
    "details": ["<details>"]
8
}