| import json |
|
|
| from persona_capsule.export import build_capsule_export |
| from persona_capsule.profile import ExemplarPair, StyleDimensions, StyleProfile |
| from persona_capsule.repository import CapsuleRecord |
| from persona_capsule.steering import MODEL_ID, MODEL_REVISION |
|
|
|
|
| def _record() -> CapsuleRecord: |
| return CapsuleRecord( |
| capsule_id="capsule-export", |
| owner_id="hf:owner", |
| name="Clear Signal", |
| status="profile_approved", |
| style_profile=StyleProfile( |
| summary="Warm and direct.", |
| descriptors=("warm", "direct"), |
| lexical_tendencies=("concrete verbs",), |
| sentence_rhythm="Concise.", |
| dimensions=StyleDimensions(50, 70, 60, 65, 40, 85, 30), |
| evidence=("Private evidence.",), |
| uncertainty=0.2, |
| ), |
| exemplar_pairs=(ExemplarPair("Good. Test it now.", "Test it now.", 0),), |
| source_fingerprint="source-v1", |
| artifact_refs=("/Users/owner/private/card.png",), |
| voice_provider="openbmb-voxcpm2-modal", |
| voice_id="private-provider-voice-id", |
| voice_status="ready", |
| voice_retention="retained", |
| voice_model_id="eleven_multilingual_v2", |
| ) |
|
|
|
|
| def test_default_export_excludes_private_data_and_activation_tensors() -> None: |
| bundle = build_capsule_export(_record()) |
| persona = json.loads(bundle.persona_bytes) |
| manifest = json.loads(bundle.manifest_bytes) |
| serialized = bundle.persona_bytes + bundle.manifest_bytes |
|
|
| assert bundle.persona_filename.endswith(".persona") |
| assert "private_exemplar_pairs" not in persona |
| assert "Private evidence." not in serialized.decode() |
| assert persona["permissions"]["activation_tensor_included"] is False |
| assert manifest["steering_recipe"]["model_id"] == MODEL_ID |
| assert manifest["steering_recipe"]["model_revision"] == MODEL_REVISION |
| assert manifest["security"]["serialized_activation_tensor"] is False |
| assert manifest["visual_recipe"]["model_id"] == "" |
| assert manifest["voice_recipe"]["provider"] == "openbmb-voxcpm2-modal" |
| assert manifest["voice_recipe"]["provider_voice_id_included"] is False |
| assert "private-provider-voice-id" not in serialized.decode() |
| assert manifest["persona_sha256"] |
| assert "/Users/" not in serialized.decode() |
| assert "private/card.png" not in serialized.decode() |
| assert "HF_TOKEN" not in serialized.decode() |
|
|
|
|
| def test_private_exemplars_require_explicit_export_selection() -> None: |
| bundle = build_capsule_export(_record(), include_private_exemplars=True) |
| persona = json.loads(bundle.persona_bytes) |
|
|
| assert persona["permissions"]["contains_private_exemplars"] is True |
| assert persona["private_exemplar_pairs"][0]["positive"] == "Good. Test it now." |
|
|
