From 61c34915a9017158a875698542e9fc94c8af5934 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@langflow.org>
Date: Sun, 23 Jun 2024 22:34:28 -0300
Subject: [PATCH] refactor: update dict to model_dump

---
 src/backend/base/langflow/api/v1/endpoints.py |  1 -
 tests/test_user.py                            | 10 +++++-----
 2 files changed, 5 insertions(+), 6 deletions(-)

diff --git a/src/backend/base/langflow/api/v1/endpoints.py b/src/backend/base/langflow/api/v1/endpoints.py
index 965c29292..deb4032de 100644
--- a/src/backend/base/langflow/api/v1/endpoints.py
+++ b/src/backend/base/langflow/api/v1/endpoints.py
@@ -180,7 +180,6 @@ async def simplified_run_flow(
             stream=stream,
             api_key_user=api_key_user,
         )
-
     except ValueError as exc:
         if "badly formed hexadecimal UUID string" in str(exc):
             # This means the Flow ID is not a valid UUID which means it can't find the flow
diff --git a/tests/test_user.py b/tests/test_user.py
index 759ce6aaf..cd1cf9165 100644
--- a/tests/test_user.py
+++ b/tests/test_user.py
@@ -90,7 +90,7 @@ def test_data_consistency_after_update(client, active_user, logged_in_headers, s
     user_id = active_user.id
     update_data = UserUpdate(is_active=False)
 
-    response = client.patch(f"/api/v1/users/{user_id}", json=update_data.dict(), headers=super_user_headers)
+    response = client.patch(f"/api/v1/users/{user_id}", json=update_data.model_dump(), headers=super_user_headers)
     assert response.status_code == 200, response.json()
 
     # Fetch the updated user from the database
@@ -164,13 +164,13 @@ def test_patch_user(client, active_user, logged_in_headers):
         username="newname",
     )
 
-    response = client.patch(f"/api/v1/users/{user_id}", json=update_data.dict(), headers=logged_in_headers)
+    response = client.patch(f"/api/v1/users/{user_id}", json=update_data.model_dump(), headers=logged_in_headers)
     assert response.status_code == 200, response.json()
     update_data = UserUpdate(
         profile_image="new_image",
     )
 
-    response = client.patch(f"/api/v1/users/{user_id}", json=update_data.dict(), headers=logged_in_headers)
+    response = client.patch(f"/api/v1/users/{user_id}", json=update_data.model_dump(), headers=logged_in_headers)
     assert response.status_code == 200, response.json()
 
 
@@ -182,7 +182,7 @@ def test_patch_reset_password(client, active_user, logged_in_headers):
 
     response = client.patch(
         f"/api/v1/users/{user_id}/reset-password",
-        json=update_data.dict(),
+        json=update_data.model_dump(),
         headers=logged_in_headers,
     )
     assert response.status_code == 200, response.json()
@@ -198,7 +198,7 @@ def test_patch_user_wrong_id(client, active_user, logged_in_headers):
         username="newname",
     )
 
-    response = client.patch(f"/api/v1/users/{user_id}", json=update_data.dict(), headers=logged_in_headers)
+    response = client.patch(f"/api/v1/users/{user_id}", json=update_data.model_dump(), headers=logged_in_headers)
     assert response.status_code == 422, response.json()
     json_response = response.json()
     detail = json_response["detail"]