diff --git a/ae/core/autogen_wrapper.py b/ae/core/autogen_wrapper.py
index d3ac84a..1e6f552 100644
--- a/ae/core/autogen_wrapper.py
+++ b/ae/core/autogen_wrapper.py
@@ -4,6 +4,7 @@
 import traceback
 from string import Template
 from time import time_ns
+from typing import Any
 
 import autogen  # type: ignore
 import openai
@@ -109,7 +110,7 @@ def set_chat_logs_dir(self, chat_logs_dir: str):
         self.chat_logs_dir = chat_logs_dir
 
 
-    def __save_chat_log(self, chat_log: list[dict[str, str]]):
+    def __save_chat_log(self, chat_log: list[dict[str, Any]]):
         chat_logs_file = os.path.join(self.get_chat_logs_dir() or "", f"nested_chat_log_{str(time_ns())}.json")
         # Save the chat log to a file
         with open(chat_logs_file, "w") as file: