fix retrival resource miss in chatflow (#18307)

2025-04-17 18:05:15 +08:00
parent 397e2a8522
commit e90c532c3a
3 changed files with 2 additions and 30 deletions
--- a/api/models/model.py
+++ b/api/models/model.py
@@ -1091,12 +1091,7 @@ class Message(db.Model):  # type: ignore[name-defined]

    @property
    def retriever_resources(self):
-        return (
-            db.session.query(DatasetRetrieverResource)
-            .filter(DatasetRetrieverResource.message_id == self.id)
-            .order_by(DatasetRetrieverResource.position.asc())
-            .all()
-        )
+        return self.message_metadata_dict.get("retriever_resources") if self.message_metadata else []

    @property
    def message_files(self):