close #43

acheong08 · Jun 1, 2023 · cff62da · cff62da
1 parent e778c84
commit cff62da
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/Bard.py b/src/Bard.py
@@ -17,6 +17,8 @@
 from rich.console import Console
 from rich.markdown import Markdown
 
+from typing import List, Dict
+
 
 def __create_session() -> PromptSession:
     return PromptSession(history=InMemoryHistory())
@@ -116,7 +118,7 @@ def save_conversation(self, file_path: str, conversation_name: str):
         with open(file_path, "w", encoding="utf-8") as f:
             json.dump(conversations, f, indent=4)
 
-    def load_conversations(self, file_path: str) -> list[dict]:
+    def load_conversations(self, file_path: str) -> List[Dict]:
         # Check if file exists
         if not os.path.isfile(file_path):
             return []