From 74d7d7c4b3560551c49b72644b5d39e30e377838 Mon Sep 17 00:00:00 2001 From: 4shen0ne <4shen.01@gmail.com> Date: Sun, 10 Aug 2025 18:39:24 +0800 Subject: [PATCH] Fix error message in upsert action --- .../autogenstudio/database/db_manager.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/python/packages/autogen-studio/autogenstudio/database/db_manager.py b/python/packages/autogen-studio/autogenstudio/database/db_manager.py index f3eceebbe891..02f134e20214 100644 --- a/python/packages/autogen-studio/autogenstudio/database/db_manager.py +++ b/python/packages/autogen-studio/autogenstudio/database/db_manager.py @@ -174,6 +174,7 @@ def upsert(self, model: BaseDBModel, return_json: bool = True) -> Response: Response: Contains status, message and data (either dict or SQLModel based on return_json) """ status = True + status_message = "" model_class = type(model) existing_model = None @@ -194,13 +195,16 @@ def upsert(self, model: BaseDBModel, return_json: bool = True) -> Response: session.rollback() logger.error("Error while updating/creating " + str(model_class.__name__) + ": " + str(e)) status = False + status_message = f"Error while updating/creating {model_class.__name__}" + else: + status_message = ( + f"{model_class.__name__} Updated Successfully" + if existing_model + else f"{model_class.__name__} Created Successfully" + ) return Response( - message=( - f"{model_class.__name__} Updated Successfully" - if existing_model - else f"{model_class.__name__} Created Successfully" - ), + message=status_message, status=status, data=model.model_dump() if return_json else model, )