improve: mordernizing validation by migrating pydantic from 1.x to 2.x (#4592)

2024-06-14 01:05:37 +08:00 · 2024-06-14 01:05:37 +08:00 · f976740b57
commit f976740b57
parent e8afc416dd
87 changed files with 697 additions and 300 deletions
--- a/api/core/callback_handler/agent_tool_callback_handler.py
+++ b/api/core/callback_handler/agent_tool_callback_handler.py
@ -29,7 +29,7 @@ def print_text(
 class DifyAgentCallbackHandler(BaseModel):
    """Callback Handler that prints to std out."""
    color: Optional[str] = ''
-    current_loop = 1
+    current_loop: int = 1

    def __init__(self, color: Optional[str] = None) -> None:
        super().__init__()