bridgekit 0.3.5__tar.gz → 0.3.7__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bridgekit
-Version: 0.3.5
+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
@@ -21,6 +21,10 @@ Requires-Dist: pypdf>=3.0.0
 Requires-Dist: python-docx>=1.0.0
 Requires-Dist: python-pptx>=0.6.0
 Requires-Dist: nbformat>=5.0.0
+Provides-Extra: openai
+Requires-Dist: openai>=1.0.0; extra == "openai"
+Provides-Extra: gemini
+Requires-Dist: google-generativeai>=0.3.0; extra == "gemini"
 Provides-Extra: dev
 Requires-Dist: pytest>=7.0.0; extra == "dev"
 Requires-Dist: pytest-mock>=3.0.0; extra == "dev"
@@ -63,12 +67,26 @@ pip install bridgekit
 !pip install bridgekit
 ```
 
-Requires an Anthropic API key:
+Requires an API key for your chosen provider:
 
+**Anthropic (default):**
 ```bash
+pip install bridgekit
 export ANTHROPIC_API_KEY=your_key_here
 ```
 
+**OpenAI:**
+```bash
+pip install bridgekit[openai]
+export OPENAI_API_KEY=your_key_here
+```
+
+**Google Gemini:**
+```bash
+pip install bridgekit[gemini]
+export GOOGLE_API_KEY=your_key_here
+```
+
 ---
 
 ## Getting Started
@@ -369,6 +387,69 @@ willing to commit to — and what's your confidence interval on that estimate?"
 
 ---
 
+## Multi-Provider Support
+
+Bridgekit now supports multiple AI providers so you're not locked into one API. You can use Anthropic, OpenAI, or Google Gemini models with any tool.
+
+**Using different providers:**
+
+```python
+from bridgekit import evaluate, plan, ask, redteam
+
+# Use OpenAI (default model: gpt-4o)
+print(evaluate("Your analysis here", provider="openai"))
+
+# Use Google Gemini (default model: gemini-1.5-pro)
+print(plan("Your question here", provider="gemini"))
+
+# Use specific model
+print(redteam("Your analysis here", model="gpt-4-turbo"))
+print(ask("Your question here", source="reports/", model="claude-3-opus-20240229"))
+```
+
+**Provider auto-detection:**
+Bridgekit automatically detects the provider from model names:
+- Models starting with "claude" → Anthropic
+- Models starting with "gpt" → OpenAI  
+- Models starting with "gemini" → Google Gemini
+
+**Default models by provider:**
+- Anthropic: `claude-3-5-sonnet-20241022`
+- OpenAI: `gpt-4o`
+- Gemini: `gemini-1.5-pro`
+
+All tools support the same `provider` and `model` parameters:
+- `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.
+
+---
+
 ## Why not just use Claude?
 
 You could. But you'd need to know what to ask, how to frame it, and what a good answer looks like. Bridgekit has that baked in — it knows you're a data scientist presenting findings, so it asks the right questions automatically. No prompt engineering required. Just paste your work and run it.
@@ -389,7 +470,7 @@ Bridgekit is a suite, not a one-off. Four tools are live — more are coming:
 
 - **Stakeholder translator** — turn your technical findings into a narrative a non-technical audience will actually follow
 - **Assumption checker** — state your analytical assumptions, get the ones you missed
-- **Multi-model support** — use any LLM provider (OpenAI, Gemini, open source models via OpenRouter) instead of being tied to Anthropic
+- **More specialized tools** — focused on specific data science workflows and challenges
 
 Each tool is small, focused, and built for the way data scientists actually work.
 

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

@@ -35,12 +35,26 @@ pip install bridgekit
 !pip install bridgekit
 ```
 
-Requires an Anthropic API key:
+Requires an API key for your chosen provider:
 
+**Anthropic (default):**
 ```bash
+pip install bridgekit
 export ANTHROPIC_API_KEY=your_key_here
 ```
 
+**OpenAI:**
+```bash
+pip install bridgekit[openai]
+export OPENAI_API_KEY=your_key_here
+```
+
+**Google Gemini:**
+```bash
+pip install bridgekit[gemini]
+export GOOGLE_API_KEY=your_key_here
+```
+
 ---
 
 ## Getting Started
@@ -341,6 +355,69 @@ willing to commit to — and what's your confidence interval on that estimate?"
 
 ---
 
+## Multi-Provider Support
+
+Bridgekit now supports multiple AI providers so you're not locked into one API. You can use Anthropic, OpenAI, or Google Gemini models with any tool.
+
+**Using different providers:**
+
+```python
+from bridgekit import evaluate, plan, ask, redteam
+
+# Use OpenAI (default model: gpt-4o)
+print(evaluate("Your analysis here", provider="openai"))
+
+# Use Google Gemini (default model: gemini-1.5-pro)
+print(plan("Your question here", provider="gemini"))
+
+# Use specific model
+print(redteam("Your analysis here", model="gpt-4-turbo"))
+print(ask("Your question here", source="reports/", model="claude-3-opus-20240229"))
+```
+
+**Provider auto-detection:**
+Bridgekit automatically detects the provider from model names:
+- Models starting with "claude" → Anthropic
+- Models starting with "gpt" → OpenAI  
+- Models starting with "gemini" → Google Gemini
+
+**Default models by provider:**
+- Anthropic: `claude-3-5-sonnet-20241022`
+- OpenAI: `gpt-4o`
+- Gemini: `gemini-1.5-pro`
+
+All tools support the same `provider` and `model` parameters:
+- `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.
+
+---
+
 ## Why not just use Claude?
 
 You could. But you'd need to know what to ask, how to frame it, and what a good answer looks like. Bridgekit has that baked in — it knows you're a data scientist presenting findings, so it asks the right questions automatically. No prompt engineering required. Just paste your work and run it.
@@ -361,7 +438,7 @@ Bridgekit is a suite, not a one-off. Four tools are live — more are coming:
 
 - **Stakeholder translator** — turn your technical findings into a narrative a non-technical audience will actually follow
 - **Assumption checker** — state your analytical assumptions, get the ones you missed
-- **Multi-model support** — use any LLM provider (OpenAI, Gemini, open source models via OpenRouter) instead of being tied to Anthropic
+- **More specialized tools** — focused on specific data science workflows and challenges
 
 Each tool is small, focused, and built for the way data scientists actually work.
 

{bridgekit-0.3.5 → 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.7/bridgekit/config.py

@@ -0,0 +1,83 @@
+import os
+from enum import Enum
+from typing import Optional
+
+
+class Provider(Enum):
+    ANTHROPIC = "anthropic"
+    OPENAI = "openai"
+    GEMINI = "gemini"
+
+
+# Default models for each provider
+DEFAULT_MODELS = {
+    Provider.ANTHROPIC: "claude-opus-4-6",
+    Provider.OPENAI: "gpt-4o",
+    Provider.GEMINI: "gemini-1.5-pro"
+}
+
+# Legacy support
+DEFAULT_MODEL = DEFAULT_MODELS[Provider.ANTHROPIC]
+
+
+def require_api_key(provider: Provider = Provider.ANTHROPIC) -> str:
+    """Return the API key for the specified provider from the environment, or raise a clear error.
+
+    Each Bridgekit tool calls this before constructing a client so
+    users get the same friendly message instead of whatever the SDK surfaces
+    when the key is missing.
+    """
+    if provider == Provider.ANTHROPIC:
+        api_key = os.environ.get("ANTHROPIC_API_KEY")
+        if not api_key:
+            raise EnvironmentError(
+                "ANTHROPIC_API_KEY not found. Set it with: export ANTHROPIC_API_KEY=your_key_here"
+            )
+        return api_key
+    elif provider == Provider.OPENAI:
+        api_key = os.environ.get("OPENAI_API_KEY")
+        if not api_key:
+            raise EnvironmentError(
+                "OPENAI_API_KEY not found. Set it with: export OPENAI_API_KEY=your_key_here"
+            )
+        return api_key
+    elif provider == Provider.GEMINI:
+        api_key = os.environ.get("GOOGLE_API_KEY")
+        if not api_key:
+            raise EnvironmentError(
+                "GOOGLE_API_KEY not found. Set it with: export GOOGLE_API_KEY=your_key_here"
+            )
+        return api_key
+    else:
+        raise ValueError(f"Unsupported provider: {provider}")
+
+
+def require_anthropic_api_key() -> str:
+    """Legacy function for backward compatibility."""
+    return require_api_key(Provider.ANTHROPIC)
+
+
+def parse_provider(provider: Optional[str] = None, model: Optional[str] = None) -> Provider:
+    """Parse provider from provider string or model name."""
+    if provider:
+        try:
+            return Provider(provider.lower())
+        except ValueError:
+            raise ValueError(f"Unsupported provider: {provider}. Supported providers: {[p.value for p in Provider]}")
+    
+    if model:
+        # Infer provider from model name
+        if model.startswith("claude"):
+            return Provider.ANTHROPIC
+        elif model.startswith("gpt"):
+            return Provider.OPENAI
+        elif model.startswith("gemini"):
+            return Provider.GEMINI
+    
+    # Default to Anthropic for backward compatibility
+    return Provider.ANTHROPIC
+
+
+def get_default_model(provider: Provider) -> str:
+    """Get the default model for a provider."""
+    return DEFAULT_MODELS.get(provider, DEFAULT_MODELS[Provider.ANTHROPIC])

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

@@ -1,5 +1,5 @@
-import anthropic
-from .config import DEFAULT_MODEL, require_anthropic_api_key
+from .config import DEFAULT_MODEL, parse_provider, get_default_model
+from .providers import create_message
 
 SYSTEM_PROMPT = """You are a senior statistician and data scientist advising a colleague on the right analytical approach for their problem.
 
@@ -29,7 +29,7 @@ ALTERNATIVES
 """
 
 
-def plan(question: str, data_description: str = None, goal: 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.
 
@@ -38,6 +38,10 @@ def plan(question: str, data_description: str = None, goal: str = None) -> str:
         data_description: Optional. A plain text description of your available data.
         goal:             Optional. The goal of your analysis (e.g. "causal inference",
                           "prediction", "segmentation", "hypothesis testing", "exploration").
+        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,
@@ -46,7 +50,10 @@ def plan(question: str, data_description: str = None, goal: str = None) -> str:
     if not question or not question.strip():
         raise ValueError("Question cannot be empty.")
 
-    api_key = require_anthropic_api_key()
+    # Parse provider and determine model
+    provider_enum = parse_provider(provider, model)
+    if model is None:
+        model = get_default_model(provider_enum)
 
     user_message = f"Question: {question}"
     if data_description:
@@ -54,17 +61,10 @@ def plan(question: str, data_description: str = None, goal: str = None) -> str:
     if goal:
         user_message += f"\n\nGoal: {goal}"
 
-    client = anthropic.Anthropic(api_key=api_key)
-    message = client.messages.create(
-        model=DEFAULT_MODEL,
-        max_tokens=1024,
-        system=SYSTEM_PROMPT,
-        messages=[
-            {
-                "role": "user",
-                "content": user_message
-            }
-        ]
+    return create_message(
+        provider=provider_enum,
+        system_prompt=system_prompt or SYSTEM_PROMPT,
+        user_message=user_message,
+        model=model,
+        max_tokens=1024
     )
-
-    return message.content[0].text

bridgekit-0.3.7/bridgekit/providers.py

@@ -0,0 +1,112 @@
+"""Provider client factory for Bridgekit multi-provider support."""
+
+from typing import Union, Dict, Any, List
+from .config import Provider, require_api_key, get_default_model
+
+
+class BaseProviderClient:
+    """Base class for provider clients."""
+    
+    def __init__(self, api_key: str):
+        self.api_key = api_key
+    
+    def create_message(self, system_prompt: str, user_message: str, model: str, max_tokens: int = 1024) -> str:
+        """Create a message with the given parameters."""
+        raise NotImplementedError("Subclasses must implement create_message")
+
+
+class AnthropicClient(BaseProviderClient):
+    """Anthropic provider client."""
+    
+    def __init__(self, api_key: str):
+        super().__init__(api_key)
+        import anthropic
+        self.client = anthropic.Anthropic(api_key=api_key)
+    
+    def create_message(self, system_prompt: str, user_message: str, model: str, max_tokens: int = 1024) -> str:
+        """Create a message using Anthropic's API."""
+        message = self.client.messages.create(
+            model=model,
+            max_tokens=max_tokens,
+            system=system_prompt,
+            messages=[{
+                "role": "user",
+                "content": user_message
+            }]
+        )
+        return message.content[0].text
+
+
+class OpenAIClient(BaseProviderClient):
+    """OpenAI provider client."""
+    
+    def __init__(self, api_key: str):
+        super().__init__(api_key)
+        import openai
+        self.client = openai.OpenAI(api_key=api_key)
+    
+    def create_message(self, system_prompt: str, user_message: str, model: str, max_tokens: int = 1024) -> str:
+        """Create a message using OpenAI's API."""
+        messages = [
+            {"role": "system", "content": system_prompt},
+            {"role": "user", "content": user_message}
+        ]
+        
+        response = self.client.chat.completions.create(
+            model=model,
+            messages=messages,
+            max_tokens=max_tokens
+        )
+        
+        return response.choices[0].message.content
+
+
+class GeminiClient(BaseProviderClient):
+    """Google Gemini provider client."""
+    
+    def __init__(self, api_key: str):
+        super().__init__(api_key)
+        import google.generativeai as genai
+        genai.configure(api_key=api_key)
+    
+    def create_message(self, system_prompt: str, user_message: str, model: str, max_tokens: int = 1024) -> str:
+        """Create a message using Google's Gemini API."""
+        import google.generativeai as genai
+        
+        # Configure the model with system instruction
+        model_instance = genai.GenerativeModel(
+            model_name=model,
+            system_instruction=system_prompt,
+            generation_config=genai.types.GenerationConfig(
+                max_output_tokens=max_tokens,
+            )
+        )
+        
+        response = model_instance.generate_content(user_message)
+        return response.text
+
+
+def create_client(provider: Provider, model: str = None) -> BaseProviderClient:
+    """Create a client for the specified provider."""
+    if model is None:
+        model = get_default_model(provider)
+    
+    api_key = require_api_key(provider)
+    
+    if provider == Provider.ANTHROPIC:
+        return AnthropicClient(api_key)
+    elif provider == Provider.OPENAI:
+        return OpenAIClient(api_key)
+    elif provider == Provider.GEMINI:
+        return GeminiClient(api_key)
+    else:
+        raise ValueError(f"Unsupported provider: {provider}")
+
+
+def create_message(provider: Provider, system_prompt: str, user_message: str, model: str = None, max_tokens: int = 1024) -> str:
+    """Create a message using the specified provider."""
+    if model is None:
+        model = get_default_model(provider)
+    
+    client = create_client(provider, model)
+    return client.create_message(system_prompt, user_message, model, max_tokens)

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

@@ -1,5 +1,5 @@
-import anthropic
-from .config import DEFAULT_MODEL, require_anthropic_api_key
+from .config import DEFAULT_MODEL, parse_provider, get_default_model
+from .providers import create_message
 
 DEFAULT_STAKEHOLDER = "a skeptical senior executive with no tolerance for weak methodology, unsupported claims, or vague business impact"
 
@@ -39,15 +39,20 @@ HARDEST QUESTION TO ANSWER
 """
 
 
-def redteam(text: str, stakeholder: 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.
+        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
@@ -56,27 +61,25 @@ def redteam(text: str, stakeholder: str = None) -> str:
     if not text or not text.strip():
         raise ValueError("Text cannot be empty.")
 
-    api_key = require_anthropic_api_key()
-
-    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
+    # Parse provider and determine model
+    provider_enum = parse_provider(provider, model)
+    if model is None:
+        model = get_default_model(provider_enum)
+
+    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,
+        user_message=user_message,
+        model=model,
+        max_tokens=1024
     )
-
-    client = anthropic.Anthropic(api_key=api_key)
-    message = client.messages.create(
-        model=DEFAULT_MODEL,
-        max_tokens=1024,
-        system=system_prompt,
-        messages=[
-            {
-                "role": "user",
-                "content": f"Red-team this analysis writeup:\n\n{text}"
-            }
-        ]
-    )
-
-    return message.content[0].text