made fix local

tarn/compat.py

@@ -27,7 +27,7 @@
     Self = Any
 # we will try to support both versions 1 and 2 while they are more or less popular
 try:
-    from pydantic import BaseModel as _BaseModel, field_validator as _field_validator, model_validator
+    from pydantic import BaseModel, field_validator as _field_validator, model_validator
 
     def field_validator(*args, always=None, **kwargs):
         # we just ignore `always`
@@ -39,11 +39,6 @@ def model_validate(cls, data):
     def model_dump(obj, **kwargs):
         return obj.model_dump(**kwargs)
 
-    class BaseModel(_BaseModel):
-        model_config = {
-            'validate_default': True,
-        }
-
     class NoExtra(BaseModel):
         model_config = {
             'extra': 'forbid',

tarn/location/disk_dict/config.py

@@ -22,6 +22,7 @@ class HashConfig(NoExtra):
 
     @model_validator(mode='before')
     def normalize_kwargs(cls, values):
+        print('kek', values)
         kwargs = {}
         for field_name in list(values):
             if field_name not in ('name', 'kwargs'):
@@ -52,7 +53,7 @@ def build(self):
 class ToolConfig(NoExtra):
     name: str
     args: Tuple = ()
-    kwargs: Optional[Dict[str, Any]] = None
+    kwargs: Optional[Dict[str, Any]] = Field(None, validate_default=True)
 
     @field_validator('kwargs', always=True)
     def normalize_kwargs(cls, v):