From b477c00f2d332967a7d7f6e88826f2b83e090e43 Mon Sep 17 00:00:00 2001 From: teaxio Date: Mon, 10 Jun 2024 21:27:33 -0400 Subject: [PATCH] fix type error --- ae/core/autogen_wrapper.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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: