From b9b5e4ea0af725f7ea0d83a9fb3177df3b052d59 Mon Sep 17 00:00:00 2001 From: Peter Jung Date: Tue, 7 Jan 2025 10:43:16 +0000 Subject: [PATCH] fix --- scripts/purge_messages.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/scripts/purge_messages.py b/scripts/purge_messages.py index 3ac0eecb..5991bce5 100644 --- 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 )