import ironcore_alloy as alloy
import os
import asyncio


async def main():
    tenant_id = os.environ.get("TENANT_ID", "tenant-gcp")
    api_key = os.environ.get("API_KEY", "0WUaXesNgbTAuLwn")
    tsp_uri = "http://localhost:32804"
    approximation_factor = 7.2
    config = alloy.SaasShieldConfiguration(
        tsp_uri, api_key, False, approximation_factor
    )
    sdk = alloy.SaasShield(config)

    data = [1.2, -1.23, 3.24, 2.37]
    print("Plaintext vector: ", data)
    plaintext = alloy.PlaintextVector(
        plaintext_vector=data,
        secret_path="contacts",
        derivation_path="conversation-sentiment",
    )
    metadata = alloy.AlloyMetadata.new_simple(tenant_id)
    encrypted = await sdk.vector().encrypt(plaintext, metadata)
    print("Encrypted vector: ", encrypted.encrypted_vector)
    # Store off encrypted_vector and paired_icl_info

    decrypted = await sdk.vector().decrypt(encrypted, metadata)
    print("Decrypted vector: ", decrypted.plaintext_vector)
    print("Note that the encryption/decryption is lossy due to floating point math.")


if __name__ == "__main__":
    loop = asyncio.get_event_loop()
    loop.run_until_complete(main())