fix

gnosis · Jan 7, 2025 · b9b5e4e · b9b5e4e
1 parent 10c4cb8
commit b9b5e4e
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/scripts/purge_messages.py b/scripts/purge_messages.py
@@ -1,6 +1,5 @@
 import typer
-from prediction_market_agent_tooling.gtypes import PrivateKey
-from pydantic import SecretStr
+from prediction_market_agent_tooling.gtypes import private_key_type
 
 from prediction_market_agent.db.agent_communication import (
     fetch_count_unprocessed_transactions,
@@ -9,8 +8,8 @@
 from prediction_market_agent.utils import APIKeys
 
 
-def main(private_key: SecretStr) -> None:
-    keys = APIKeys(BET_FROM_PRIVATE_KEY=PrivateKey(private_key))
+def main(private_key: str) -> None:
+    keys = APIKeys(BET_FROM_PRIVATE_KEY=private_key_type(private_key))
     n_messages = fetch_count_unprocessed_transactions(
         consumer_address=keys.bet_from_address
     )