diff --git a/api/models/model.py b/api/models/model.py index de095d0319..45b48c623b 100644 --- a/api/models/model.py +++ b/api/models/model.py @@ -1322,12 +1322,12 @@ class MessageAnnotation(Base): db.Index("message_annotation_message_idx", "message_id"), ) - id = mapped_column(StringUUID, server_default=db.text("uuid_generate_v4()")) - app_id = mapped_column(StringUUID, nullable=False) - conversation_id = mapped_column(StringUUID, db.ForeignKey("conversations.id"), nullable=True) - message_id = mapped_column(StringUUID, nullable=True) - question = mapped_column(db.Text, nullable=True) - content = mapped_column(db.Text, nullable=False) + id: Mapped[str] = mapped_column(StringUUID, server_default=db.text("uuid_generate_v4()")) + app_id: Mapped[str] = mapped_column(StringUUID) + conversation_id: Mapped[Optional[str]] = mapped_column(StringUUID, db.ForeignKey("conversations.id")) + message_id: Mapped[Optional[str]] = mapped_column(StringUUID) + question: Mapped[Optional[str]] + content: Mapped[str] hit_count = mapped_column(db.Integer, nullable=False, server_default=db.text("0")) account_id = mapped_column(StringUUID, nullable=False) created_at = mapped_column(db.DateTime, nullable=False, server_default=func.current_timestamp()) diff --git a/api/tests/unit_tests/utils/position_helper/test_position_helper.py b/api/tests/unit_tests/utils/position_helper/test_position_helper.py index 29558a93c2..dbd8f05098 100644 --- a/api/tests/unit_tests/utils/position_helper/test_position_helper.py +++ b/api/tests/unit_tests/utils/position_helper/test_position_helper.py @@ -95,7 +95,7 @@ def test_included_position_data(prepare_example_positions_yaml): position_map = get_position_map(folder_path=prepare_example_positions_yaml, file_name="example_positions.yaml") pin_list = ["forth", "first"] include_set = {"forth", "first"} - exclude_set = {} + exclude_set = set() position_map = pin_position_map(original_position_map=position_map, pin_list=pin_list)