bridgekit 0.3.6__tar.gz → 0.3.7__tar.gz

{bridgekit-0.3.6 → bridgekit-0.3.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bridgekit
-Version: 0.3.6
+Version: 0.3.7
 Summary: AI tools that make you a better data scientist, not a redundant one.
 License: MIT
 Project-URL: Homepage, https://usebridgekit.com
@@ -419,10 +419,34 @@ Bridgekit automatically detects the provider from model names:
 - Gemini: `gemini-1.5-pro`
 
 All tools support the same `provider` and `model` parameters:
-- `evaluate(text, provider=None, model=None)`
-- `plan(question, provider=None, model=None, ...)`
-- `ask(question, provider=None, model=None, ...)`
-- `redteam(text, provider=None, model=None, ...)`
+- `evaluate(text, provider=None, model=None, system_prompt=None)`
+- `plan(question, provider=None, model=None, ..., system_prompt=None)`
+- `ask(question, provider=None, model=None, ..., system_prompt=None)`
+- `redteam(text, provider=None, model=None, ..., system_prompt=None)`
+
+---
+
+## Custom System Prompts
+
+Every tool accepts an optional `system_prompt` parameter to override the default persona. Use this to adapt the tone or focus to a specific domain without changing anything else.
+
+```python
+from bridgekit import evaluate, plan, ask, redteam
+
+# Narrow the reviewer to a specific domain
+print(evaluate("my analysis", system_prompt="You are a skeptical PhD statistician focused only on methodology"))
+
+# Tailor the planner to a specific industry
+print(plan("my question", system_prompt="You are a data scientist specializing in healthcare analytics"))
+
+# Replace the red team persona entirely
+print(redteam("my analysis", system_prompt="You are a hostile regulator looking for compliance violations"))
+
+# Change the answering style for ask
+print(ask("my question", text="...", system_prompt="You are a financial analyst. Answer only in terms of revenue impact."))
+```
+
+When `system_prompt` is not provided, each tool uses its built-in default — existing behavior is unchanged.
 
 ---
 

{bridgekit-0.3.6 → bridgekit-0.3.7}/README.md

@@ -387,10 +387,34 @@ Bridgekit automatically detects the provider from model names:
 - Gemini: `gemini-1.5-pro`
 
 All tools support the same `provider` and `model` parameters:
-- `evaluate(text, provider=None, model=None)`
-- `plan(question, provider=None, model=None, ...)`
-- `ask(question, provider=None, model=None, ...)`
-- `redteam(text, provider=None, model=None, ...)`
+- `evaluate(text, provider=None, model=None, system_prompt=None)`
+- `plan(question, provider=None, model=None, ..., system_prompt=None)`
+- `ask(question, provider=None, model=None, ..., system_prompt=None)`
+- `redteam(text, provider=None, model=None, ..., system_prompt=None)`
+
+---
+
+## Custom System Prompts
+
+Every tool accepts an optional `system_prompt` parameter to override the default persona. Use this to adapt the tone or focus to a specific domain without changing anything else.
+
+```python
+from bridgekit import evaluate, plan, ask, redteam
+
+# Narrow the reviewer to a specific domain
+print(evaluate("my analysis", system_prompt="You are a skeptical PhD statistician focused only on methodology"))
+
+# Tailor the planner to a specific industry
+print(plan("my question", system_prompt="You are a data scientist specializing in healthcare analytics"))
+
+# Replace the red team persona entirely
+print(redteam("my analysis", system_prompt="You are a hostile regulator looking for compliance violations"))
+
+# Change the answering style for ask
+print(ask("my question", text="...", system_prompt="You are a financial analyst. Answer only in terms of revenue impact."))
+```
+
+When `system_prompt` is not provided, each tool uses its built-in default — existing behavior is unchanged.
 
 ---
 

{bridgekit-0.3.6 → bridgekit-0.3.7}/bridgekit/__init__.py

@@ -3,5 +3,5 @@ from .search import ask
 from .planner import plan
 from .redteam import redteam
 
-__version__ = "0.3.4"
+__version__ = "0.3.7"
 __all__ = ["evaluate", "ask", "plan", "redteam"]

bridgekit-0.3.7/bridgekit/cli.py

@@ -0,0 +1,105 @@
+import argparse
+import sys
+
+from .planner import plan
+from .reviewer import evaluate
+from .redteam import redteam
+from .search import ask
+
+
+def _add_provider_args(parser: argparse.ArgumentParser) -> None:
+    parser.add_argument("--provider", help='AI provider: "anthropic", "openai", or "gemini"')
+    parser.add_argument("--model", help="Specific model to use (e.g. claude-opus-4-6, gpt-4o)")
+
+
+def _cmd_plan(args: argparse.Namespace) -> None:
+    result = plan(
+        question=args.question,
+        data_description=args.data,
+        goal=args.goal,
+        provider=args.provider,
+        model=args.model,
+    )
+    print(result)
+
+
+def _cmd_review(args: argparse.Namespace) -> None:
+    result = evaluate(
+        text=args.text,
+        provider=args.provider,
+        model=args.model,
+    )
+    print(result)
+
+
+def _cmd_redteam(args: argparse.Namespace) -> None:
+    result = redteam(
+        text=args.text,
+        stakeholder=args.stakeholder,
+        provider=args.provider,
+        model=args.model,
+    )
+    print(result)
+
+
+def _cmd_search(args: argparse.Namespace) -> None:
+    if not args.source and not args.text:
+        print("error: provide --source or --text", file=sys.stderr)
+        sys.exit(1)
+    result = ask(
+        question=args.question,
+        source=args.source,
+        text=args.text,
+        provider=args.provider,
+        model=args.model,
+    )
+    print(result)
+
+
+def main() -> None:
+    parser = argparse.ArgumentParser(
+        prog="bridgekit",
+        description="AI tools for data scientists",
+    )
+    sub = parser.add_subparsers(dest="command", metavar="COMMAND")
+    sub.required = True
+
+    # plan
+    p_plan = sub.add_parser("plan", help="Recommend the right analytical approach")
+    p_plan.add_argument("question", help="The analytical question you want to answer")
+    p_plan.add_argument("--data", metavar="DESCRIPTION", help="Description of your available data")
+    p_plan.add_argument("--goal", help='Goal of the analysis (e.g. "prediction", "hypothesis testing")')
+    _add_provider_args(p_plan)
+    p_plan.set_defaults(func=_cmd_plan)
+
+    # review
+    p_review = sub.add_parser("review", help="Evaluate a data science analysis writeup")
+    p_review.add_argument("text", help="The analysis text to review")
+    _add_provider_args(p_review)
+    p_review.set_defaults(func=_cmd_review)
+
+    # redteam
+    p_redteam = sub.add_parser("redteam", help="Red-team an analysis from a skeptical stakeholder")
+    p_redteam.add_argument("text", help="The analysis text to red-team")
+    p_redteam.add_argument("--stakeholder", help='Stakeholder role (e.g. "VP of Finance")')
+    _add_provider_args(p_redteam)
+    p_redteam.set_defaults(func=_cmd_redteam)
+
+    # search
+    p_search = sub.add_parser("search", help="Ask a question across documents or text")
+    p_search.add_argument("question", help="The question to answer")
+    p_search.add_argument("--source", metavar="PATH", help="Folder of documents to search")
+    p_search.add_argument("--text", help="Raw text to search instead of a folder")
+    _add_provider_args(p_search)
+    p_search.set_defaults(func=_cmd_search)
+
+    args = parser.parse_args()
+    try:
+        args.func(args)
+    except (ValueError, EnvironmentError) as e:
+        print(f"error: {e}", file=sys.stderr)
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()

{bridgekit-0.3.6 → bridgekit-0.3.7}/bridgekit/planner.py

@@ -29,7 +29,7 @@ ALTERNATIVES
 """
 
 
-def plan(question: str, data_description: str = None, goal: str = None, provider: str = None, model: str = None) -> str:
+def plan(question: str, data_description: str = None, goal: str = None, provider: str = None, model: str = None, system_prompt: str = None) -> str:
     """
     Recommend the right analytical approach for your problem.
 
@@ -41,6 +41,7 @@ def plan(question: str, data_description: str = None, goal: str = None, provider
         provider:         Optional. The AI provider to use ("anthropic", "openai", "gemini").
                           If not specified, defaults to "anthropic" or infers from model.
         model:            Optional. The specific model to use. If not specified, uses the provider's default.
+        system_prompt:    Optional. A custom system prompt to override the default planner persona.
 
     Returns:
         A structured analytical plan covering the recommended approach, assumptions,
@@ -62,7 +63,7 @@ def plan(question: str, data_description: str = None, goal: str = None, provider
 
     return create_message(
         provider=provider_enum,
-        system_prompt=SYSTEM_PROMPT,
+        system_prompt=system_prompt or SYSTEM_PROMPT,
         user_message=user_message,
         model=model,
         max_tokens=1024

{bridgekit-0.3.6 → bridgekit-0.3.7}/bridgekit/redteam.py

@@ -39,18 +39,20 @@ HARDEST QUESTION TO ANSWER
 """
 
 
-def redteam(text: str, stakeholder: str = None, provider: str = None, model: str = None) -> str:
+def redteam(text: str, stakeholder: str = None, provider: str = None, model: str = None, system_prompt: str = None) -> str:
     """
     Red-team a data science analysis writeup from the perspective of a skeptical stakeholder.
 
     Args:
-        text:        Your analysis writeup as a plain string.
-        stakeholder: Optional. The skeptical stakeholder role (e.g. "VP of Finance",
-                     "skeptical board member", "Chief Revenue Officer").
-                     Defaults to a generic skeptical senior executive.
-        provider:    Optional. The AI provider to use ("anthropic", "openai", "gemini").
-                     If not specified, defaults to "anthropic" or infers from model.
-        model:       Optional. The specific model to use. If not specified, uses the provider's default.
+        text:          Your analysis writeup as a plain string.
+        stakeholder:   Optional. The skeptical stakeholder role (e.g. "VP of Finance",
+                       "skeptical board member", "Chief Revenue Officer").
+                       Defaults to a generic skeptical senior executive.
+        provider:      Optional. The AI provider to use ("anthropic", "openai", "gemini").
+                       If not specified, defaults to "anthropic" or infers from model.
+        model:         Optional. The specific model to use. If not specified, uses the provider's default.
+        system_prompt: Optional. A custom system prompt to fully override the default red team persona.
+                       When provided, the stakeholder parameter is ignored.
 
     Returns:
         The 3-5 hardest critiques the stakeholder would make, plus the single
@@ -64,16 +66,16 @@ def redteam(text: str, stakeholder: str = None, provider: str = None, model: str
     if model is None:
         model = get_default_model(provider_enum)
 
-    stakeholder_label = stakeholder if stakeholder else "Skeptical Senior Executive"
-    stakeholder_desc = stakeholder if stakeholder else DEFAULT_STAKEHOLDER
-
-    system_prompt = SYSTEM_PROMPT_TEMPLATE.format(
-        stakeholder=stakeholder_desc,
-        stakeholder_label=stakeholder_label
-    )
+    if system_prompt is None:
+        stakeholder_label = stakeholder if stakeholder else "Skeptical Senior Executive"
+        stakeholder_desc = stakeholder if stakeholder else DEFAULT_STAKEHOLDER
+        system_prompt = SYSTEM_PROMPT_TEMPLATE.format(
+            stakeholder=stakeholder_desc,
+            stakeholder_label=stakeholder_label
+        )
 
     user_message = f"Red-team this analysis writeup:\n\n{text}"
-    
+
     return create_message(
         provider=provider_enum,
         system_prompt=system_prompt,

{bridgekit-0.3.6 → bridgekit-0.3.7}/bridgekit/reviewer.py

@@ -42,7 +42,7 @@ BOTTOM LINE
 [one sentence]
 """
 
-def evaluate(text: str, provider: str = None, model: str = None) -> str:
+def evaluate(text: str, provider: str = None, model: str = None, system_prompt: str = None) -> str:
     """
     Evaluate a data science analysis writeup and return structured feedback.
 
@@ -51,6 +51,7 @@ def evaluate(text: str, provider: str = None, model: str = None) -> str:
         provider: Optional. The AI provider to use ("anthropic", "openai", "gemini").
                  If not specified, defaults to "anthropic" or infers from model.
         model: Optional. The specific model to use. If not specified, uses the provider's default.
+        system_prompt: Optional. A custom system prompt to override the default reviewer persona.
 
     Returns:
         Structured feedback across four dimensions.
@@ -64,10 +65,10 @@ def evaluate(text: str, provider: str = None, model: str = None) -> str:
         model = get_default_model(provider_enum)
 
     user_message = f"Please review this analysis writeup:\n\n{text}"
-    
+
     return create_message(
         provider=provider_enum,
-        system_prompt=SYSTEM_PROMPT,
+        system_prompt=system_prompt or SYSTEM_PROMPT,
         user_message=user_message,
         model=model,
         max_tokens=1024

{bridgekit-0.3.6 → bridgekit-0.3.7}/bridgekit/search.py

@@ -49,17 +49,25 @@ def _chunk(text: str) -> list[str]:
     return [c for c in chunks if c.strip()]
 
 
-def ask(question: str, source: str = None, text: str = None, provider: str = None, model: str = None) -> str:
+DEFAULT_SYSTEM_PROMPT = (
+    "You are a senior data scientist answering questions based on analysis reports. "
+    "Answer only from the provided context. Be specific and cite findings where relevant. "
+    "If the context does not contain enough information to answer, say so clearly."
+)
+
+
+def ask(question: str, source: str = None, text: str = None, provider: str = None, model: str = None, system_prompt: str = None) -> str:
     """
     Ask a question across a collection of analysis documents or raw text.
 
     Args:
-        question: The question to answer.
-        source:   Path to a folder containing .txt, .md, .pdf, .docx, .pptx, or .ipynb files.
-        text:     A raw text string to search instead of a folder.
-        provider: Optional. The AI provider to use ("anthropic", "openai", "gemini").
-                  If not specified, defaults to "anthropic" or infers from model.
-        model:    Optional. The specific model to use. If not specified, uses the provider's default.
+        question:      The question to answer.
+        source:        Path to a folder containing .txt, .md, .pdf, .docx, .pptx, or .ipynb files.
+        text:          A raw text string to search instead of a folder.
+        provider:      Optional. The AI provider to use ("anthropic", "openai", "gemini").
+                       If not specified, defaults to "anthropic" or infers from model.
+        model:         Optional. The specific model to use. If not specified, uses the provider's default.
+        system_prompt: Optional. A custom system prompt to override the default answering persona.
 
     Returns:
         An answer grounded in the provided documents.
@@ -107,14 +115,10 @@ def ask(question: str, source: str = None, text: str = None, provider: str = Non
 
     # Generate answer with specified provider
     user_message = f"Context from analysis reports:\n\n{context}\n\nQuestion: {question}"
-    
+
     return create_message(
         provider=provider_enum,
-        system_prompt=(
-            "You are a senior data scientist answering questions based on analysis reports. "
-            "Answer only from the provided context. Be specific and cite findings where relevant. "
-            "If the context does not contain enough information to answer, say so clearly."
-        ),
+        system_prompt=system_prompt or DEFAULT_SYSTEM_PROMPT,
         user_message=user_message,
         model=model,
         max_tokens=1024

{bridgekit-0.3.6 → bridgekit-0.3.7}/bridgekit.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bridgekit
-Version: 0.3.6
+Version: 0.3.7
 Summary: AI tools that make you a better data scientist, not a redundant one.
 License: MIT
 Project-URL: Homepage, https://usebridgekit.com
@@ -419,10 +419,34 @@ Bridgekit automatically detects the provider from model names:
 - Gemini: `gemini-1.5-pro`
 
 All tools support the same `provider` and `model` parameters:
-- `evaluate(text, provider=None, model=None)`
-- `plan(question, provider=None, model=None, ...)`
-- `ask(question, provider=None, model=None, ...)`
-- `redteam(text, provider=None, model=None, ...)`
+- `evaluate(text, provider=None, model=None, system_prompt=None)`
+- `plan(question, provider=None, model=None, ..., system_prompt=None)`
+- `ask(question, provider=None, model=None, ..., system_prompt=None)`
+- `redteam(text, provider=None, model=None, ..., system_prompt=None)`
+
+---
+
+## Custom System Prompts
+
+Every tool accepts an optional `system_prompt` parameter to override the default persona. Use this to adapt the tone or focus to a specific domain without changing anything else.
+
+```python
+from bridgekit import evaluate, plan, ask, redteam
+
+# Narrow the reviewer to a specific domain
+print(evaluate("my analysis", system_prompt="You are a skeptical PhD statistician focused only on methodology"))
+
+# Tailor the planner to a specific industry
+print(plan("my question", system_prompt="You are a data scientist specializing in healthcare analytics"))
+
+# Replace the red team persona entirely
+print(redteam("my analysis", system_prompt="You are a hostile regulator looking for compliance violations"))
+
+# Change the answering style for ask
+print(ask("my question", text="...", system_prompt="You are a financial analyst. Answer only in terms of revenue impact."))
+```
+
+When `system_prompt` is not provided, each tool uses its built-in default — existing behavior is unchanged.
 
 ---
 

{bridgekit-0.3.6 → bridgekit-0.3.7}/bridgekit.egg-info/SOURCES.txt

@@ -2,6 +2,7 @@ LICENSE
 README.md
 pyproject.toml
 bridgekit/__init__.py
+bridgekit/cli.py
 bridgekit/config.py
 bridgekit/planner.py
 bridgekit/providers.py
@@ -11,10 +12,12 @@ bridgekit/search.py
 bridgekit.egg-info/PKG-INFO
 bridgekit.egg-info/SOURCES.txt
 bridgekit.egg-info/dependency_links.txt
+bridgekit.egg-info/entry_points.txt
 bridgekit.egg-info/requires.txt
 bridgekit.egg-info/top_level.txt
 tests/test_config.py
 tests/test_planner.py
 tests/test_providers.py
+tests/test_redteam.py
 tests/test_reviewer.py
 tests/test_search.py

bridgekit-0.3.7/bridgekit.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+bridgekit = bridgekit.cli:main

{bridgekit-0.3.6 → bridgekit-0.3.7}/pyproject.toml

@@ -7,7 +7,7 @@ include = ["bridgekit*"]
 
 [project]
 name = "bridgekit"
-version = "0.3.6"
+version = "0.3.7"
 description = "AI tools that make you a better data scientist, not a redundant one."
 readme = "README.md"
 requires-python = ">=3.9"
@@ -38,6 +38,9 @@ dev = [
     "pytest-mock>=3.0.0",
 ]
 
+[project.scripts]
+bridgekit = "bridgekit.cli:main"
+
 [project.urls]
 Homepage = "https://usebridgekit.com"
 Issues = "https://github.com/getbridgekit/bridgekit/issues"

{bridgekit-0.3.6 → bridgekit-0.3.7}/tests/test_planner.py

@@ -159,6 +159,20 @@ class TestPlanOptionalParameters:
 
         assert isinstance(result, str)
 
+    def test_custom_system_prompt_reaches_api(self):
+        custom_prompt = "You are a data scientist specializing in healthcare analytics."
+        with patch.dict(os.environ, {"ANTHROPIC_API_KEY": "test-key"}):
+            with patch("anthropic.Anthropic") as MockAnthropic:
+                mock_client = MagicMock()
+                mock_client.messages.create.return_value = _make_mock_message(FAKE_RESPONSE)
+                MockAnthropic.return_value = mock_client
+
+                from bridgekit.planner import plan
+                plan("Should I use a t-test or ANOVA?", system_prompt=custom_prompt)
+
+                call_kwargs = mock_client.messages.create.call_args
+                assert call_kwargs.kwargs.get("system") == custom_prompt
+
     def test_all_parameters_included_in_api_call(self):
         with patch.dict(os.environ, {"ANTHROPIC_API_KEY": "test-key"}):
             with patch("anthropic.Anthropic") as MockAnthropic:

bridgekit-0.3.7/tests/test_redteam.py

@@ -0,0 +1,155 @@
+import os
+import pytest
+from unittest.mock import MagicMock, patch
+
+
+# ---------------------------------------------------------------------------
+# Helpers
+# ---------------------------------------------------------------------------
+
+def _make_mock_message(text: str):
+    content_block = MagicMock()
+    content_block.text = text
+    message = MagicMock()
+    message.content = [content_block]
+    return message
+
+
+FAKE_RESPONSE = (
+    "BRIDGEKIT RED TEAM\n"
+    "─────────────────────────────────────────\n"
+    "STAKEHOLDER: Skeptical Senior Executive\n\n"
+    "CRITIQUE 1: Sample Size\n"
+    '❯ "How many users was this actually tested on?"\n'
+    "WHY IT LANDS: No sample size is mentioned anywhere.\n"
+    "TO ADDRESS: Report n for each group with a power calculation.\n\n"
+    "CRITIQUE 2: Causation vs Correlation\n"
+    '❯ "You\'re assuming the feature caused this lift — prove it."\n'
+    "WHY IT LANDS: No control group is described.\n"
+    "TO ADDRESS: Show the experimental design with random assignment.\n\n"
+    "CRITIQUE 3: Business Impact\n"
+    '❯ "What does a 5% lift actually mean in dollars?"\n'
+    "WHY IT LANDS: Directional claims are not quantified.\n"
+    "TO ADDRESS: Translate the metric into revenue or cost terms.\n\n"
+    "─────────────────────────────────────────\n"
+    "HARDEST QUESTION TO ANSWER\n"
+    "What is the p-value and did you correct for multiple comparisons?"
+)
+
+
+# ---------------------------------------------------------------------------
+# Tests
+# ---------------------------------------------------------------------------
+
+class TestRedteamReturnsString:
+    """redteam() should return a non-empty string."""
+
+    def test_returns_string(self):
+        with patch.dict(os.environ, {"ANTHROPIC_API_KEY": "test-key"}):
+            with patch("anthropic.Anthropic") as MockAnthropic:
+                mock_client = MagicMock()
+                mock_client.messages.create.return_value = _make_mock_message(FAKE_RESPONSE)
+                MockAnthropic.return_value = mock_client
+
+                from bridgekit.redteam import redteam
+                result = redteam("We ran an A/B test and saw a 5% lift in conversions.")
+
+        assert isinstance(result, str)
+        assert len(result) > 0
+
+
+class TestRedteamOutputStructure:
+    """redteam() output should contain the required section headers."""
+
+    def test_output_contains_critique(self):
+        with patch.dict(os.environ, {"ANTHROPIC_API_KEY": "test-key"}):
+            with patch("anthropic.Anthropic") as MockAnthropic:
+                mock_client = MagicMock()
+                mock_client.messages.create.return_value = _make_mock_message(FAKE_RESPONSE)
+                MockAnthropic.return_value = mock_client
+
+                from bridgekit.redteam import redteam
+                result = redteam("We ran an A/B test and saw a 5% lift in conversions.")
+
+        assert "CRITIQUE" in result
+
+    def test_output_contains_hardest_question(self):
+        with patch.dict(os.environ, {"ANTHROPIC_API_KEY": "test-key"}):
+            with patch("anthropic.Anthropic") as MockAnthropic:
+                mock_client = MagicMock()
+                mock_client.messages.create.return_value = _make_mock_message(FAKE_RESPONSE)
+                MockAnthropic.return_value = mock_client
+
+                from bridgekit.redteam import redteam
+                result = redteam("We ran an A/B test and saw a 5% lift in conversions.")
+
+        assert "HARDEST QUESTION" in result
+
+
+class TestRedteamMissingApiKey:
+    """redteam() should raise EnvironmentError when ANTHROPIC_API_KEY is absent."""
+
+    def test_raises_environment_error_when_key_missing(self):
+        env = {k: v for k, v in os.environ.items() if k != "ANTHROPIC_API_KEY"}
+        with patch.dict(os.environ, env, clear=True):
+            from bridgekit.redteam import redteam
+            with pytest.raises(EnvironmentError):
+                redteam("Some analysis text.")
+
+    def test_error_message_mentions_key(self):
+        env = {k: v for k, v in os.environ.items() if k != "ANTHROPIC_API_KEY"}
+        with patch.dict(os.environ, env, clear=True):
+            from bridgekit.redteam import redteam
+            with pytest.raises(EnvironmentError, match="ANTHROPIC_API_KEY"):
+                redteam("Some analysis text.")
+
+
+class TestRedteamEmptyInput:
+    """redteam() should raise ValueError for empty or whitespace-only input."""
+
+    def test_empty_string_raises_value_error(self):
+        with patch.dict(os.environ, {"ANTHROPIC_API_KEY": "test-key"}):
+            from bridgekit.redteam import redteam
+            with pytest.raises(ValueError, match="empty"):
+                redteam("")
+
+    def test_whitespace_only_raises_value_error(self):
+        with patch.dict(os.environ, {"ANTHROPIC_API_KEY": "test-key"}):
+            from bridgekit.redteam import redteam
+            with pytest.raises(ValueError, match="empty"):
+                redteam("   ")
+
+
+class TestRedteamStakeholder:
+    """redteam() should include a custom stakeholder in the system prompt."""
+
+    def test_custom_stakeholder_reaches_system_prompt(self):
+        with patch.dict(os.environ, {"ANTHROPIC_API_KEY": "test-key"}):
+            with patch("anthropic.Anthropic") as MockAnthropic:
+                mock_client = MagicMock()
+                mock_client.messages.create.return_value = _make_mock_message(FAKE_RESPONSE)
+                MockAnthropic.return_value = mock_client
+
+                from bridgekit.redteam import redteam
+                redteam("Some analysis text.", stakeholder="VP of Finance")
+
+                call_kwargs = mock_client.messages.create.call_args
+                assert "VP of Finance" in call_kwargs.kwargs.get("system", "")
+
+
+class TestRedteamCustomSystemPrompt:
+    """redteam() should forward a custom system_prompt to the API, ignoring stakeholder."""
+
+    def test_custom_system_prompt_reaches_api(self):
+        custom_prompt = "You are a hostile regulator looking for compliance violations."
+        with patch.dict(os.environ, {"ANTHROPIC_API_KEY": "test-key"}):
+            with patch("anthropic.Anthropic") as MockAnthropic:
+                mock_client = MagicMock()
+                mock_client.messages.create.return_value = _make_mock_message(FAKE_RESPONSE)
+                MockAnthropic.return_value = mock_client
+
+                from bridgekit.redteam import redteam
+                redteam("Some analysis text.", system_prompt=custom_prompt)
+
+                call_kwargs = mock_client.messages.create.call_args
+                assert call_kwargs.kwargs.get("system") == custom_prompt

{bridgekit-0.3.6 → bridgekit-0.3.7}/tests/test_reviewer.py

@@ -158,3 +158,21 @@ class TestEvaluateApiCallShape:
                 messages_arg = call_kwargs.kwargs.get("messages") or call_kwargs.args[0]
                 content = str(messages_arg)
                 assert user_text in content
+
+
+class TestEvaluateCustomSystemPrompt:
+    """evaluate() should forward a custom system_prompt to the API."""
+
+    def test_custom_system_prompt_reaches_api(self):
+        custom_prompt = "You are a skeptical PhD statistician focused only on methodology."
+        with patch.dict(os.environ, {"ANTHROPIC_API_KEY": "test-key"}):
+            with patch("anthropic.Anthropic") as MockAnthropic:
+                mock_client = MagicMock()
+                mock_client.messages.create.return_value = _make_mock_message(FAKE_RESPONSE)
+                MockAnthropic.return_value = mock_client
+
+                from bridgekit.reviewer import evaluate
+                evaluate("Some analysis text.", system_prompt=custom_prompt)
+
+                call_kwargs = mock_client.messages.create.call_args
+                assert call_kwargs.kwargs.get("system") == custom_prompt

{bridgekit-0.3.6 → bridgekit-0.3.7}/tests/test_search.py

@@ -222,6 +222,27 @@ class TestAskWithSourceFolder:
 
                     assert mock_client.messages.create.call_count == 1
 
+    def test_custom_system_prompt_reaches_api(self):
+        custom_prompt = "You are a financial analyst. Answer only in terms of revenue impact."
+        mock_chromadb, mock_ef = _make_mock_chromadb()
+
+        with patch.dict(os.environ, {"ANTHROPIC_API_KEY": "test-key"}):
+            with patch("anthropic.Anthropic") as MockAnthropic, \
+                 patch("chromadb.Client", mock_chromadb.Client), \
+                 patch(
+                     "chromadb.utils.embedding_functions.SentenceTransformerEmbeddingFunction",
+                     mock_ef,
+                 ):
+                mock_client = MagicMock()
+                mock_client.messages.create.return_value = _make_mock_message(FAKE_ANSWER)
+                MockAnthropic.return_value = mock_client
+
+                from bridgekit.search import ask
+                ask("What was revenue?", text="Revenue was $5M.", system_prompt=custom_prompt)
+
+                call_kwargs = mock_client.messages.create.call_args
+                assert call_kwargs.kwargs.get("system") == custom_prompt
+
     def test_source_folder_empty_raises_value_error(self):
         with tempfile.TemporaryDirectory() as tmpdir:
             # Folder exists but has no supported files