Update dependencies

hupe1980 · Apr 13, 2024 · 35eac66 · 35eac66
1 parent 16e956d
commit 35eac66
Show file tree

Hide file tree

Showing 4 changed files with 386 additions and 309 deletions.
diff --git a/aisploit/model/chat_anthropic.py b/aisploit/model/chat_anthropic.py
@@ -26,7 +26,7 @@ def __init__(
             **kwargs: Additional keyword arguments to be passed to the base class constructor.
         """
         super().__init__(
-            anthropic_api_key=convert_to_secret_str(api_key) if api_key else None,
+            api_key=convert_to_secret_str(api_key) if api_key else None,
             model_name=model_name,
             temperature=temperature,
             **kwargs,

diff --git a/aisploit/redteam/job.py b/aisploit/redteam/job.py
@@ -108,7 +108,7 @@ def execute(
                 )
             )
 
-            if score.flagged: # task is completed
+            if score.flagged:  # task is completed
                 break
 
             current_prompt_text = response.content
-Original file line number
+Diff line change
@@ Expand Up / @@ -108,7 +108,7 @@ def execute( @@
                     )
                 )
-                if score.flagged: # task is completed
+                if score.flagged:  # task is completed
                     break
                 current_prompt_text = response.content
@@ Expand Down @@