Update agent.py

agent.py

@@ -14,6 +14,7 @@ import yaml
 import os
 import re
 import json
+from typing import Optional
 
 # --------------------------
 # Core Tools from Previous Implementation