Skip to content

Commit

Permalink
fix type error
Browse files Browse the repository at this point in the history
  • Loading branch information
teaxio committed Jun 11, 2024
1 parent b23f0f4 commit b477c00
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion ae/core/autogen_wrapper.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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:
Expand Down

0 comments on commit b477c00

Please sign in to comment.