diff --git a/template_prompt.py b/template_prompt.py
index fff1f90..b208fc8 100644
--- a/template_prompt.py
+++ b/template_prompt.py
@@ -1,4 +1,4 @@
-from typing import List
+from typing import List, Dict
 
 from towhee.operator import PyOperator