add type checking to prompt template

Signed-off-by: thiswillbeyourgithub <[email protected]>

Author: thiswillbeyourgithub

wdoc/utils/prompts.py

@@ -13,6 +13,7 @@
 from pydantic import BaseModel, Field, model_validator
 from loguru import logger
 
+from wdoc.utils.typechecker import optional_typecheck
 from wdoc.utils.misc import get_tkn_length
 
 # PROMPT FOR SUMMARY TASKS
@@ -311,6 +312,7 @@ def remove_thoughts(cls, values) -> List[str]:
 )
 
 
+@optional_typecheck
 class Prompts_class:
     evaluate: ChatPromptTemplate
     answer: ChatPromptTemplate