fix test

gnosis · Dec 16, 2024 · 6c762ab · 6c762ab
1 parent 3c0bbbc
commit 6c762ab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/agents/microchain/test_messages_functions.py b/tests/agents/microchain/test_messages_functions.py
@@ -43,7 +43,7 @@ def account2_private_key() -> SecretStr:
 )  # anvil account 1
 
 
-def mock_spice_query(query: str, api_key: str) -> pl.DataFrame:
+def mock_spice_query(query: str, api_key: str, cache: bool) -> pl.DataFrame:
     return pl.DataFrame(
         {
             "hash": [MOCK_HASH_1, MOCK_HASH_2],