wisent 0.5.14__py3-none-any.whl → 0.5.15__py3-none-any.whl

wisent/tests/examples/cli/synthetic/test_synthetic_pairs.py

@@ -0,0 +1,45 @@
+"""
+Tests for synthetic pairs generation example.
+
+Validates synthetic contrastive pair creation.
+"""
+
+import subprocess
+import pytest
+import tempfile
+import os
+import json
+
+
+def test_create_synthetic_pairs():
+    """Test creating synthetic contrastive pairs for a trait."""
+    with tempfile.TemporaryDirectory() as tmpdir:
+        output_file = os.path.join(tmpdir, "synthetic_pairs.json")
+        
+        result = subprocess.run(
+            [
+                "python", "-m", "wisent.core.main", "generate-pairs",
+                "--trait", "helpfulness",
+                "--output", output_file,
+                "--model", "meta-llama/Llama-3.2-1B-Instruct",
+                "--num-pairs", "5",
+                "--similarity-threshold", "0.8",
+                "--device", "cpu",
+                "--verbose"
+            ],
+            capture_output=True,
+            text=True,
+            timeout=300
+        )
+        
+        assert result.returncode == 0, f"Command failed: {result.stderr}"
+        assert os.path.exists(output_file), "Output file not created"
+        
+        # Verify JSON format
+        with open(output_file, 'r') as f:
+            data = json.load(f)
+            assert isinstance(data, (list, dict)), "Output should be JSON"
+
+
+if __name__ == "__main__":
+    pytest.main([__file__, "-v"])

{wisent-0.5.14.dist-info → wisent-0.5.15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wisent
-Version: 0.5.14
+Version: 0.5.15
 Summary: Monitor and guard against harmful content in language models
 Home-page: https://github.com/yourusername/wisent-activation-guardrails
 Author: Wisent Team
@@ -26,6 +26,8 @@ Requires-Dist: sentence-transformers>=2.0.0
 Requires-Dist: faiss-cpu>=1.7.0
 Provides-Extra: harness
 Requires-Dist: lm-eval==0.4.8; extra == "harness"
+Provides-Extra: cuda
+Requires-Dist: flash-attn>=2.5.0; extra == "cuda"
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

{wisent-0.5.14.dist-info → wisent-0.5.15.dist-info}/RECORD

@@ -1,4 +1,5 @@
-wisent/__init__.py,sha256=eZCvGSWzmhAuX33aknAmdOv-0F6FE4joo5igjfie5QI,23
+wisent/__init__.py,sha256=L6LoKMlJx-n68Agaz17S4PydD7S5Z4lwW3o79dy3l1c,23
+wisent/cli.py,sha256=XKzGIGstr38EowHYpr821c6YuV9Eaw3I1I3NvLztTO0,3960
 wisent/core/__init__.py,sha256=yNBOdCXpkwMuo1H1sNAYgPf5_-Hhf7y4H-RZJPCiOpI,463
 wisent/core/autonomous_agent.py,sha256=XszhMkONc8kClCMqQfM8ZRQPxiMKJCmZSF7hrY-hf8Q,52878
 wisent/core/benchmark_extractors.py,sha256=E51DOJM-cbtlDy9VrIeoBKSEQ6AZ1jeyDM7OAw-vn94,10473
@@ -9,20 +10,20 @@ wisent/core/detection_handling.py,sha256=iiuKpzAbJfx_KFn2SFABQHOeeWblDJMXjzGwGDe
 wisent/core/download_full_benchmarks.py,sha256=GylPUmd0JtrnqGaqyU5PVV8uQPvJyaRcCs_2QqHyO1I,58743
 wisent/core/hyperparameter_optimizer.py,sha256=GN-IY27kkFKuXwHbvalfEW5UVf4NfZ0G9rMWK2ThEAY,17319
 wisent/core/lm_eval_harness_ground_truth.py,sha256=Fb33sZ_nT0dhkzsEmWv3Ed40NCC-onWXGk379UpsEsA,64964
-wisent/core/main.py,sha256=3a4ARsrTdP3FdBs85a3vofXwt1696Rc6BIy_G6KicZQ,2202
+wisent/core/main.py,sha256=-EtcGAbzqrKeT0zzM-EyEMOYOfsBYmx2qvIGrdZCMFc,2419
 wisent/core/managed_cached_benchmarks.py,sha256=JbvpZ1fgSuQQhyQVKEvqrQZRHGqfnjo9NFhgITFoFsE,22854
 wisent/core/mixed_benchmark_sampler.py,sha256=tKQCHUXVuYeCyx4VZt8O1hGyB-TOY_SQ_SYi8cyApII,13585
 wisent/core/model_config_manager.py,sha256=rQAdSmk3GFlZXyHp3fSV1bORxiZWhmzIz1uo3H4JtkA,12009
-wisent/core/model_persistence.py,sha256=1IIJ_13pO0Q0FJYn25LiaDwEFDC7FAPJmxzL9TbaQmY,10624
+wisent/core/model_persistence.py,sha256=Pr1A3E0MxqwICKsFXmjKVq9KgG8Ip5UbtG-gujGFShc,9870
 wisent/core/multi_steering.py,sha256=YhVKmf08KacVEYZWLk6t2uNWSv-Pi_zBeLdDopo3QXk,13491
-wisent/core/parser.py,sha256=_YDeSuQMx0zNknz9rX3Ls1YPT1x5eohoY8rfjeoqxV8,69091
+wisent/core/parser.py,sha256=GFyHJklf7ReFMKk0ugSL2-bC8_Lyi6GSlNpeXlU-umY,69201
 wisent/core/representation.py,sha256=hBl_N9qbr5Gsa7GCQ0nMWRm82RqYEfhd9cyf0PPH5LY,195
 wisent/core/sample_size_optimizer.py,sha256=6wegGXZpdGpiR4R0YJ1D2JqLr6yinMndEx2gB5FL80s,23666
-wisent/core/sample_size_optimizer_v2.py,sha256=bVYJRZC4_Mrq-HFlYLyv-9tWvqEHJ3kCeIwlmYOwI6I,13286
+wisent/core/sample_size_optimizer_v2.py,sha256=wPkqNhiDkVAKl5iekMynqS9kILz4sY4ygHNZ3PPvbjs,13291
 wisent/core/save_results.py,sha256=PRwaA5qO6EOsvURvLBl3YhvanlC0D0G4iYqxYAQ7sw8,13737
 wisent/core/steering.py,sha256=I1mSLOQ6MMdUcJ_MmfmaCUH0IKxeCYlcsMCO6x6wLg4,22651
 wisent/core/steering_method.py,sha256=-hZqtvwRS7sGqQJUd36MoPm0rjbO1LrtPAYmcIk8BqQ,462
-wisent/core/steering_optimizer.py,sha256=TBSyO9I3z3uBY7kntD-ZJaJ1ltWIskL_eLvc5XIqM_Y,55872
+wisent/core/steering_optimizer.py,sha256=opgcNILzzKJJu4VuQIoMm72kiD5-a2aLsjsEUs-ZHZ4,55861
 wisent/core/task_interface.py,sha256=OlWdcxkprmZcOto-bXmg75kzUcWzH_kyW_e7w2FdPLM,4471
 wisent/core/task_selector.py,sha256=QVgozUuiM74BMUJ8Ucb_sn6HQk5v0wL_QUsqKb55vJE,6224
 wisent/core/time_estimator.py,sha256=DcgSzW-hr9BjmXJwBnGqE2dkFK0zgyz5WNF7934CJ9k,5778
@@ -30,7 +31,7 @@ wisent/core/timing_calibration.py,sha256=AVSHWetHa01omwGh4zp2WN2ByfCq8zpktfLdtqD
 wisent/core/user_model_config.py,sha256=8optjLqf9wTDtOf0c705d5_Rr2zE67jV9BNqoY-TRvA,6735
 wisent/core/activations/__init__.py,sha256=ZT3aU-eU5-RR_QDkmKZRweCltJTWVof8mgbL3ONAXKw,839
 wisent/core/activations/activations.py,sha256=SS-WEds9AoTnxcpVnAUYEydGHjfRRzLhKEHgTgwJYmE,3976
-wisent/core/activations/activations_collector.py,sha256=aKX3L7xAICXU1WRrn3THA5llR7TgaaQyfXDeHdaBtrc,16486
+wisent/core/activations/activations_collector.py,sha256=i354mWA7QNufxW2VN91uHjmO7-tbpJaQN40lsnfXY9w,16870
 wisent/core/activations/prompt_construction_strategy.py,sha256=KpaAdj75D3t6JR47mx70vr82uCijsuaMIolIEXuciO8,1403
 wisent/core/activations/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wisent/core/activations/core/atoms.py,sha256=FeeQj-YKb-1rLX0gnzFc5VM47ikWwfFh3datRduj6mQ,9101
@@ -47,23 +48,24 @@ wisent/core/agent/diagnose/create_classifier.py,sha256=UJKFgjGvIsRKfc5fPkfh227bA
 wisent/core/agent/diagnose/response_diagnostics.py,sha256=49P_J29nYaz4BW3vcs6SdI_kbVutKGSpq_neUPcWlro,10684
 wisent/core/agent/diagnose/select_classifiers.py,sha256=Uv5dxEmZRjjkZKmVmYQ1jC_XJVyM2MF5dv-4QRuTWKY,18320
 wisent/core/agent/diagnose/synthetic_classifier_option.py,sha256=SetaYBdgbAlbCGjpgbADq9MnCtbQogByJLhqi3ndiM4,32495
-wisent/core/agent/diagnose/test_synthetic_classifier.py,sha256=EWPPHtoND797XfoUjNZcmcuBp8w3PYm4YIBu1JXhzPE,2688
 wisent/core/agent/diagnose/tasks/__init__.py,sha256=rfImwPtKCAfz-ASOBQyF4DEhU6hgCuiZBcqh1AaUE80,704
 wisent/core/agent/diagnose/tasks/task_manager.py,sha256=OXodauLfbAMawL8rR1KL-CfOHtNbH05Ai4LP9gSo9yU,62746
 wisent/core/agent/diagnose/tasks/task_relevance.py,sha256=D4UBr0TqUNXkDZnNgA5wa4NYHSKtDaiugYeVg5zGQjs,3250
 wisent/core/agent/diagnose/tasks/task_selector.py,sha256=ll34stireeqW-B_T4daf_91kujzVFQ8sOilk-JrxpHA,5414
-wisent/core/cli/__init__.py,sha256=eNnjkY__dtWvohH7U4SwgAZrMfwZ4gAsR_7hgohfq5I,1078
-wisent/core/cli/create_steering_vector.py,sha256=RzKjuvzWYRdXO8BcTZeEhOFwuH7id0i0iIuWtrLUfVg,5586
-wisent/core/cli/evaluate_responses.py,sha256=9YAdA-gdIsAgkWscJcXGbn5pM3sHLyXmGCaYZi_HSKM,29098
+wisent/core/cli/__init__.py,sha256=hTo_urwejrc28w_4EqoiFyxYGTT0673r6gk36iU9eOg,1241
+wisent/core/cli/create_steering_vector.py,sha256=IYfwdxSiWpakkGE-VOFASbiHNwpZa7SMgQzb7Ggt1OY,7928
+wisent/core/cli/evaluate_responses.py,sha256=I-OUgBJYOXon5ijTt6Lm1iO5nb93pxNhvU_8Ci-9ZIE,29655
 wisent/core/cli/generate_pairs.py,sha256=6xhdkwnwKVvkSrSlQ0Zq-qNWBWPFVqhDqpg2YXiU9jA,5209
-wisent/core/cli/generate_pairs_from_task.py,sha256=WeEwLlmc4d0yy6QldYhPuzW0-6i3ok1BdVfM3K_QqDs,4574
+wisent/core/cli/generate_pairs_from_task.py,sha256=plbD5z4urG-kdJxIMlGyDexBNLJGetpWCdNxUZ5EtW4,5203
 wisent/core/cli/generate_responses.py,sha256=mV6wKGz1Ewyeph25tSaNzqjNuv1qtwwG2ktOwCkdtG0,3966
 wisent/core/cli/generate_vector_from_synthetic.py,sha256=TjmRD2k-ybwruolFC_na6bM79CPsYMweLDGP-q4XnNc,5567
 wisent/core/cli/generate_vector_from_task.py,sha256=ZzrddZvaskF1L83j6rl08AmTe-XTAo_5spX-3MuZHUo,5474
-wisent/core/cli/get_activations.py,sha256=ZTRjXzSxj3nzLlONJ9PMW4j4N1Thxu-AC4qKP4uU-Yo,7666
-wisent/core/cli/optimize_classification.py,sha256=7klpLkImMDYDZov35FSI-A_jLl3P9BhQcV6TmXFpZgI,14997
-wisent/core/cli/optimize_steering.py,sha256=6yBMNAt0AOBhzxjcc6G0Un_TyopNIMicN4BxSH5TQkw,16549
-wisent/core/cli/tasks.py,sha256=ujiui_CJou2PW3ro8-O5HFDihUyDc8LoEuntXRUyayY,7672
+wisent/core/cli/get_activations.py,sha256=LDtNksSxsf7zh_899r4ZoG-l3fQm1FXtBTCd6LQSfH4,7693
+wisent/core/cli/multi_steer.py,sha256=-ofVmsrSpZjuNq_UCQWT7TurpRkMCQdHPNFqcqnu9uU,4276
+wisent/core/cli/optimize_classification.py,sha256=VaboV8XLMoO4b7Ny9Za_EUBoMRdWWXi4k3sFuBadxPw,10095
+wisent/core/cli/optimize_sample_size.py,sha256=CGuqxvU_YpslZUhE1gJpcusFS_wiAh1mU9PdTd3H7r0,3005
+wisent/core/cli/optimize_steering.py,sha256=rJAhN1MEmOl8xlCkGOUNXQ-kZITkQ8mihkDZssGv0wg,31902
+wisent/core/cli/tasks.py,sha256=WWbPjwkoj6c4A0l6HjxLdRQ4ST80uXL7qLZRLvFeDG4,21164
 wisent/core/contrastive_pairs/__init__.py,sha256=AbaAf-t_nyVVy_vLjp8WAlMDmNun3KNp_GMWAK25r9g,429
 wisent/core/contrastive_pairs/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wisent/core/contrastive_pairs/core/atoms.py,sha256=_zghw6c8iisW_SqBIUCoAnzhc5q7t5EgZ4zzTPxeLwQ,1129
@@ -142,13 +144,15 @@ wisent/core/data_loaders/loaders/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRk
 wisent/core/data_loaders/loaders/custom.py,sha256=Xe1sOHH3_dRjBnQg9vwMM-XA8ROn65dUr9TeT-nuNtQ,4144
 wisent/core/data_loaders/loaders/lm_loader.py,sha256=Trdit9L--ZMqkUE_tT0S2OWrMEO4o9tjFkHx2r3LZVU,7858
 wisent/core/data_loaders/loaders/task_interface_loader.py,sha256=QGDfqMvdTFfogF9P533b-Gw52uHe0GnsS64AfnVVlek,11771
+wisent/core/errors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/core/errors/error_handler.py,sha256=6OE7pQHOrOm6ZtSYlrN3GjyeWgCzkTF1Znsl0nd2Yog,4435
 wisent/core/evaluators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-wisent/core/evaluators/rotator.py,sha256=cEoybk2MRsvCuM_lyKkvpYLFCTQgo-jIH23kdmN8quk,5681
+wisent/core/evaluators/rotator.py,sha256=gjJwHKncWWfJeK_603d-UQ_8oE8CjIcE6DbokMv6wFo,6542
 wisent/core/evaluators/benchmark_specific/__init__.py,sha256=Hjrh-p2bGp7Px2KpyPBxQLgYzlmFwBBk0MgofYkygHM,955
 wisent/core/evaluators/benchmark_specific/exact_match_evaluator.py,sha256=8v-yRdW8PD9eRwwcJIcIl9MZRMvqLECueYAHdL3DGrA,2645
 wisent/core/evaluators/benchmark_specific/f1_evaluator.py,sha256=E2nTdZGem1P4KwQJDH6l3p9CfcMqW_PfmAgpoU2ocKs,3153
 wisent/core/evaluators/benchmark_specific/generation_evaluator.py,sha256=o_gu2fG5A8cLAhPfhyGUjvmLxg8OlFpNS-GYUWegu3Q,7019
-wisent/core/evaluators/benchmark_specific/log_likelihoods_evaluator.py,sha256=Xuh5t7Ls5Cfc_yits0caZVJNOu0RLaxjRSG9-ub8rfo,15701
+wisent/core/evaluators/benchmark_specific/log_likelihoods_evaluator.py,sha256=khHjdfb517-tfNRWe-K6NkfiUc2wIlanSofesjs5ffI,7305
 wisent/core/evaluators/benchmark_specific/perplexity_evaluator.py,sha256=NUlpKaDwkQR8BB8gudC1qTdHqvSyBX3oGoKrtE5ZBVw,4743
 wisent/core/evaluators/benchmark_specific/personalization_evaluator.py,sha256=R9c0UQrdEDKK6ZRHeoDANy2lw1zJE6A8i6uL-nJezok,8887
 wisent/core/evaluators/benchmark_specific/coding/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -181,7 +185,7 @@ wisent/core/evaluators/oracles/interactive.py,sha256=f3v2_N17fKzGyeOxONRJbrbn8i5
 wisent/core/evaluators/oracles/nlp_evaluator.py,sha256=KxbnF-I2IFbBQpoYyjQKGbYh4NErsEuhTCRYX_Tob8o,18220
 wisent/core/evaluators/oracles/user_specified.py,sha256=V1dKrNj3Oq7UC_I7DT0WGnktP7R_DSW6UAwDdrA8SnE,2360
 wisent/core/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-wisent/core/models/wisent_model.py,sha256=rP1rLv9JZXXcngWCIVckuy7q-akqfi6xaotwYL9HYPI,32001
+wisent/core/models/wisent_model.py,sha256=oViMB8of-kGBEoKt4niwlVdZA2bfUhsT358Ju1ytW5Y,32410
 wisent/core/models/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wisent/core/models/core/atoms.py,sha256=_Bpz0Sfiq6_VswThIltUwNGj_ukl5MhAg8RrgMKwEBM,15756
 wisent/core/opti/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -215,16 +219,16 @@ wisent/core/parser_arguments/generate_vector_from_synthetic_parser.py,sha256=gC0
 wisent/core/parser_arguments/generate_vector_from_task_parser.py,sha256=pm_nfSN0UqL2QsFYvFg9R-14Ukjm4jEb1f6H40pQXCI,3518
 wisent/core/parser_arguments/generate_vector_parser.py,sha256=v9oi1ggqAV8GpGQhECy6htw4U_ZVmBgW9zWjZlnus6o,3618
 wisent/core/parser_arguments/get_activations_parser.py,sha256=ChOV21B3Jji-f8KnZW_8H97agJJJv-dYLv2a33OE0y4,2382
-wisent/core/parser_arguments/main_parser.py,sha256=PnYOt3M-c4mFtKKsiq-cogSMlP-2YQ_C0FF6tsFrqFY,8273
+wisent/core/parser_arguments/main_parser.py,sha256=lQkZ7IDFwGWZxPyiGuh2OTmfvXt8E_Ph-60jQ6Qssx4,8268
 wisent/core/parser_arguments/model_config_parser.py,sha256=MBbn5lUVi5n1iTOwKQ1IzSsFgOwDmHSuPJA9Y1Qp5Po,3150
 wisent/core/parser_arguments/monitor_parser.py,sha256=qo3vyyVmdZBkhGuhlHqbEzTiIKcdFIvG3IIuc-DIRQE,1234
-wisent/core/parser_arguments/multi_steer_parser.py,sha256=UMig_rdC0wexQjQtGByXHHMrWn6eSaWWFRGwrofSLo8,2035
+wisent/core/parser_arguments/multi_steer_parser.py,sha256=v1CCmkO_avVetYaxwS0ryUiTGyAluWVRalfFYxhzxNk,2145
+wisent/core/parser_arguments/nonsense_parser.py,sha256=ALxU27Hn7ewB8IVe7bMS78MqukU4riouy492nlM4BUM,1096
 wisent/core/parser_arguments/optimize_classification_parser.py,sha256=G3OtlPa5MrqAx10bD3PO7NeT0VehLk-WxY4z9g7ZHBY,2745
 wisent/core/parser_arguments/optimize_sample_size_parser.py,sha256=pgX3pyYowsrpWhM3aNtdQCClsQaLAFX_bb3oqHltU-U,2816
-wisent/core/parser_arguments/optimize_steering_parser.py,sha256=bZ1g5UvKn2WwD0gtdmJgOGc6wCPswgFycuaoJPM7L_w,6811
+wisent/core/parser_arguments/optimize_steering_parser.py,sha256=cAQ2XA3nBpidm0U9Dx4NIsdFojp-_3wmmtkBiv6DKEo,7451
 wisent/core/parser_arguments/synthetic_parser.py,sha256=NYxmdGmqo0680dkoKw82pleq4v0B1q77t1N7IBng3qI,3606
 wisent/core/parser_arguments/tasks_parser.py,sha256=oVLxDdUcQVDmJzSYYbULmSQNvSIdzSRWL_UxNAu_2F4,22719
-wisent/core/parser_arguments/test_nonsense_parser.py,sha256=ALxU27Hn7ewB8IVe7bMS78MqukU4riouy492nlM4BUM,1096
 wisent/core/parser_arguments/utils.py,sha256=E2s2VE14-iedjjcATuu60qnfYgxSKiYyGqeFeTwABUo,3654
 wisent/core/prompts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wisent/core/prompts/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -286,9 +290,35 @@ wisent/core/trainers/core/__init__.py,sha256=D0JX0-XCHdtLrCXhVDHNQafvyWCvJ4-o4UK
 wisent/core/trainers/core/atoms.py,sha256=ycWk0G-7EIAEOnQL-o5_V5B8KTQ7CQUilGF4ibjighM,1536
 wisent/core/utils/__init__.py,sha256=NavuBkpDSz6q1dN6m34-I-l8Aps1Sgcnx82FHxFufzY,457
 wisent/core/utils/device.py,sha256=5brw9tclTU77NNAokzLXYrKUnjjYBzFfA7wNJPM9ytM,1609
-wisent-0.5.14.dist-info/licenses/LICENSE,sha256=wy0iaw8b2tyqZAfKHib3lP3PJ9o88FDCg92oUHh3sDQ,1073
-wisent-0.5.14.dist-info/METADATA,sha256=AEpPbm_BRKb9Z04q4n8we84vbDFgrZsXgwhzd8WpLHc,2413
-wisent-0.5.14.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-wisent-0.5.14.dist-info/entry_points.txt,sha256=BM76j3xjtIcVZGk24iDf5w18s6SuqeOpaiAxfZhpnY8,49
-wisent-0.5.14.dist-info/top_level.txt,sha256=2Ts9Iyldnb3auIN2HBBaHPknRy7nSRDm2f6RGzYgr8A,7
-wisent-0.5.14.dist-info/RECORD,,
+wisent/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/tests/examples/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/tests/examples/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/tests/examples/cli/activations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/tests/examples/cli/activations/test_get_activations.py,sha256=sIEpXyN4O3pPiIiZmbDAnKnRG2IubGl1L3M3D0agP_U,4009
+wisent/tests/examples/cli/classifier/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/tests/examples/cli/classifier/test_classifier_examples.py,sha256=kLSw-fxVl0auFFRW3J_KEEucndtwWefFkyRTn6z4G1o,4855
+wisent/tests/examples/cli/contrastive_pairs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/tests/examples/cli/contrastive_pairs/test_generate_pairs.py,sha256=oQv44hEYEGtk0DimgDbXJGrvyw_bsdJ-MBKvZUdXYAo,2860
+wisent/tests/examples/cli/evaluation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/tests/examples/cli/evaluation/test_evaluation_examples.py,sha256=6YxRGtbPb08xFTjaBn1twz8_8OP7zMNM7ArxF8w4CyU,3551
+wisent/tests/examples/cli/generate/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/tests/examples/cli/generate/test_generate_with_classifier.py,sha256=FfUgHbeaeuexk61o0to9Cn6F47UjtVQyM1pUWRSm4mg,5096
+wisent/tests/examples/cli/generate/test_generate_with_steering.py,sha256=s-BmC-v8nkLFnCq42swo9gyFlT6Al3t_Dran3kswcoU,5183
+wisent/tests/examples/cli/generate/test_only_generate.py,sha256=WPuww-w9XtUWTLYMVgka2TFhkZWZ2EFZmLjQ5FZVjtg,3581
+wisent/tests/examples/cli/multi_steering/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/tests/examples/cli/multi_steering/test_multi_steer_from_trained_vectors.py,sha256=3MAD1aMGfP1s3Touczkl5mCbUmQuGuqnq8B7_y875dg,7537
+wisent/tests/examples/cli/multi_steering/test_multi_steer_with_different_parameters.py,sha256=vfUl6R7UsW5wADR5mleO4uGexKLrKocoUyhcIRVZCs4,6857
+wisent/tests/examples/cli/multi_steering/test_train_and_multi_steer.py,sha256=2nuvKhrIoCXxZfFoNWNs_V4Sifpuhiy2hAqmO8VJ2sg,6196
+wisent/tests/examples/cli/optimizer/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/tests/examples/cli/optimizer/test_optimize_sample_size.py,sha256=3Matgc0vcuJFIL1pu7Tz_EVmmyOP77YeXDoo9viiMQM,3428
+wisent/tests/examples/cli/optimizer/test_optimizer_examples.py,sha256=l6L4eZDjmGWyUdkTcJGOK533LM1vlV7u0QcjmvjP4_o,1795
+wisent/tests/examples/cli/steering/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/tests/examples/cli/steering/test_create_steering_vectors.py,sha256=MJnUtRlsJqeteS33WU-_3fPW33PdLxT3tLBtRikmA34,4585
+wisent/tests/examples/cli/synthetic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/tests/examples/cli/synthetic/test_synthetic_pairs.py,sha256=d-OhvfMLMu_69T_RRNREGeiC-0aHJTooJZ6J4yR9ZKc,1320
+wisent-0.5.15.dist-info/licenses/LICENSE,sha256=wy0iaw8b2tyqZAfKHib3lP3PJ9o88FDCg92oUHh3sDQ,1073
+wisent-0.5.15.dist-info/METADATA,sha256=HazdPEJUPt5DouANRksVe9_S2brv1qmXAegOEb1qVDI,2484
+wisent-0.5.15.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+wisent-0.5.15.dist-info/entry_points.txt,sha256=BM76j3xjtIcVZGk24iDf5w18s6SuqeOpaiAxfZhpnY8,49
+wisent-0.5.15.dist-info/top_level.txt,sha256=2Ts9Iyldnb3auIN2HBBaHPknRy7nSRDm2f6RGzYgr8A,7
+wisent-0.5.15.dist-info/RECORD,,

wisent/core/agent/diagnose/test_synthetic_classifier.py

@@ -1,71 +0,0 @@
-import sys
-import time
-import signal
-from pathlib import Path
-
-# Add the project root to the path
-project_root = Path(__file__).parent.parent.parent.parent
-sys.path.insert(0, str(project_root))
-
-from wisent.core.model import Model
-from wisent.core.agent.diagnose.synthetic_classifier_option import (
-    create_classifiers_for_prompt,
-    apply_classifiers_to_response
-)
-from wisent.core.agent.budget import set_time_budget
-
-class TimeoutError(Exception):
-    """Raised when test exceeds time budget."""
-    pass
-
-def timeout_handler(signum, frame):
-    raise TimeoutError("Test exceeded time budget!")
-
-def main():
-    # Set budget and timeout separately
-    budget_minutes = 1.0  # 1 minute - internal budget for classifier creation
-    timeout_seconds = 120  # 2 minutes - hard timeout for the test process
-    set_time_budget(budget_minutes)
-    
-    print(f"⏱️ Starting synthetic classifier test with {timeout_seconds}s timeout and {budget_minutes*60}s budget...")
-    
-    # Set up timeout signal
-    signal.signal(signal.SIGALRM, timeout_handler)
-    signal.alarm(timeout_seconds)
-    
-    start_time = time.time()
-    
-    try:
-        model = Model(name="meta-llama/Llama-3.1-8B-Instruct")
-        prompt = "What is the capital of France?"
-        
-        # Test the system
-        classifiers, trait_discovery = create_classifiers_for_prompt(model, prompt)
-        
-        # Clear the alarm since we completed successfully
-        signal.alarm(0)
-        
-        elapsed_time = time.time() - start_time
-        print(f"✅ SUCCESS: Created {len(classifiers)} classifiers for {len(trait_discovery.traits_discovered)} traits")
-        print(f"⏱️ Total time: {elapsed_time:.1f}s (timeout: {timeout_seconds}s, budget: {budget_minutes*60}s)")
-        
-        if elapsed_time > timeout_seconds:
-            print(f"⚠️ WARNING: Test completed but exceeded timeout by {elapsed_time - timeout_seconds:.1f}s")
-        else:
-            print(f"🎉 Test completed within timeout with {timeout_seconds - elapsed_time:.1f}s to spare!")
-            
-    except TimeoutError as e:
-        elapsed_time = time.time() - start_time
-        print(f"❌ ERROR: {e}")
-        print(f"❌ Test failed after {elapsed_time:.1f}s (timeout: {timeout_seconds}s, budget: {budget_minutes*60}s)")
-        print("❌ This indicates a performance issue that needs investigation.")
-        sys.exit(1)
-    except Exception as e:
-        signal.alarm(0)  # Clear timeout
-        elapsed_time = time.time() - start_time
-        print(f"❌ ERROR: Test failed with exception: {e}")
-        print(f"❌ Time elapsed: {elapsed_time:.1f}s")
-        sys.exit(1)
-
-if __name__ == "__main__":
-    main()