with open('eden_state_management.py', 'r') as f:
    content = f.read()

# Fix the import
content = content.replace(
    'from typing import Any, Dict, List, Optional',
    'from typing import Any, Callable, Dict, List, Optional'
)

with open('eden_state_management.py', 'w') as f:
    f.write(content)

print("Fixed imports in eden_state_management.py")