diff --git a/llm/templates.py b/llm/templates.py index 4a18a44..3b819c3 100644 --- a/llm/templates.py +++ b/llm/templates.py @@ -1,4 +1,4 @@ -from pydantic import ConfigDict, BaseModel +from pydantic import BaseModel import string from typing import Optional, Any, Dict, List, Tuple @@ -9,7 +9,6 @@ class Template(BaseModel): system: Optional[str] = None model: Optional[str] = None defaults: Optional[Dict[str, Any]] = None - model_config = ConfigDict(extra="forbid") class MissingVariables(Exception): pass