fix for pydantic<2

Author: hamdan-27

trubrics/integrations/streamlit/collect.py

@@ -96,7 +96,7 @@ def st_feedback(
                     else:
                         if success_fail_message:
                             st.success("Feedback saved to Trubrics.")
-                        return feedback.model_dump()
+                        return self._pydantic_to_dict(feedback)
                 else:
                     user_response = Response(**user_response)
                     feedback = Feedback(
@@ -108,15 +108,15 @@ def st_feedback(
                         tags=tags,
                         metadata=metadata,
                     )
-                    return feedback.model_dump()
+                    return self._pydantic_to_dict(feedback)
         elif feedback_type in ("thumbs", "faces"):
 
             def _log_feedback_trubrics(user_response, **kwargs):
                 feedback = self.log_feedback(user_response=user_response, **kwargs)
                 if success_fail_message:
                     if feedback:
                         st.toast("Feedback saved to [Trubrics](https://trubrics.streamlit.app/).", icon="✅")
-                        return feedback.model_dump()
+                        return self._pydantic_to_dict(feedback)
                     else:
                         st.toast("Error in saving feedback to [Trubrics](https://trubrics.streamlit.app/).", icon="❌")
 
@@ -147,12 +147,20 @@ def _log_feedback_trubrics(user_response, **kwargs):
                     tags=tags,
                     metadata=metadata,
                 )
-                return feedback.model_dump()
+                return self._pydantic_to_dict(feedback)
             return user_response
         else:
             raise ValueError("feedback_type must be one of ['textbox', 'faces', 'thumbs'].")
         return None
 
+    @staticmethod
+    def _pydantic_to_dict(feedback: Feedback) -> dict:
+        """Support for pydantic v1 and v2."""
+        try:
+            return feedback.model_dump()
+        except AttributeError:
+            return feedback.dict()
+
     @staticmethod
     def st_textbox_ui(
         type: str = "text-input", key: Optional[str] = None, label: Optional[str] = None
@@ -188,6 +196,7 @@ def clear_session_state():
                 )
             elif type == "text-area":
                 title = st.text_area(label=label or "Provide some feedback", key=f"{key}_title")
+
         if title:
             st.button("Save feedback", on_click=clear_session_state, key=f"{key}_save_button")
         if st.session_state[f"{key}_save_button"]: