Changed typing to fit python 3.8

AntonOsika · Jun 23, 2023 · 102e44e · 102e44e
1 parent 9bde354
commit 102e44e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/gpt_engineer/ai.py b/gpt_engineer/ai.py
@@ -2,6 +2,8 @@
 
 import logging
 
+from typing import Dict, List
+
 import openai
 
 logger = logging.getLogger(__name__)
@@ -39,7 +41,7 @@ def fuser(self, msg):
     def fassistant(self, msg):
         return {"role": "assistant", "content": msg}
 
-    def next(self, messages: list[dict[str, str]], prompt=None):
+    def next(self, messages: List[Dict[str, str]], prompt=None):
         if prompt:
             messages += [{"role": "user", "content": prompt}]