stateprobe 0.3.0__py3-none-any.whl

stateprobe/__init__.py

@@ -0,0 +1,53 @@
+"""StateProbe — Prompt State Debugger.
+
+Diagnose which behavioral vectors your prompt activates in an LLM,
+identify pollution sources, and get rewrite suggestions to reach a target state.
+
+Theoretical foundation:
+- Anthropic, "Persona Vectors: Monitoring and Controlling Character Traits in
+  Language Models" (arXiv:2507.21509, 2025)
+- DeepSeek-AI, "DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via
+  Reinforcement Learning" (arXiv:2501.12948, 2025)
+"""
+
+__version__ = "0.3.0"
+
+from stateprobe.models import (
+    Axis,
+    AxisReading,
+    PollutionSource,
+    Report,
+    TargetPreset,
+)
+from stateprobe.detector import diagnose
+from stateprobe.rewriter import suggest_rewrite
+from stateprobe.engines import (
+    Engine,
+    EngineError,
+    EngineUnavailable,
+    EvidenceContributor,
+    LLMJudgeContributor,
+    LLMJudgeEngine,
+    StaticEngine,
+    StaticRuleContributor,
+)
+
+__all__ = [
+    "Axis",
+    "AxisReading",
+    "PollutionSource",
+    "Report",
+    "TargetPreset",
+    "diagnose",
+    "suggest_rewrite",
+    "EvidenceContributor",
+    "EngineError",
+    "EngineUnavailable",
+    "StaticRuleContributor",
+    "LLMJudgeContributor",
+    # Deprecated aliases (removed in v0.3):
+    "Engine",
+    "StaticEngine",
+    "LLMJudgeEngine",
+    "__version__",
+]

stateprobe/__main__.py

@@ -0,0 +1,13 @@
+"""Entry point for ``python -m stateprobe``.
+
+Forwards to the standard CLI defined by ``[project.scripts] stateprobe`` so
+users who don't have the ``stateprobe`` script on PATH (e.g. ``pip install
+--user`` without PATH set, or running directly from a checkout) can still
+invoke the CLI in the conventional way.
+"""
+
+from stateprobe.cli import main
+
+
+if __name__ == "__main__":
+    main()