From 9de133b6726511ba1ad6ce5ab8cb9fb8a7b7e933 Mon Sep 17 00:00:00 2001 From: Adam Ladachowski Date: Tue, 3 Mar 2026 16:13:45 +0100 Subject: [PATCH] Include trigger words in /api/db/files response Co-Authored-By: Claude Opus 4.5 --- tensors/db.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/tensors/db.py b/tensors/db.py index 3ac4cce..7df0ba3 100644 --- a/tensors/db.py +++ b/tensors/db.py @@ -164,7 +164,7 @@ class Database: session.add(SafetensorMetadata(local_file_id=local_file_id, key=key, value=str_value)) def list_local_files(self) -> list[dict[str, Any]]: - """List all local files with CivitAI info.""" + """List all local files with CivitAI info and trigger words.""" with self.session() as session: files = session.exec(select(LocalFile)).all() results = [] @@ -178,6 +178,13 @@ class Database: creator = None if model and model.creator_id: creator = session.exec(select(Creator).where(Creator.id == model.creator_id)).first() + # Get trigger words for this version + triggers: list[str] = [] + if version: + words = session.exec( + select(TrainedWord).where(TrainedWord.version_id == version.id).order_by(col(TrainedWord.position)) + ).all() + triggers = [w.word for w in words] results.append( { "id": f.id, @@ -192,6 +199,7 @@ class Database: "version_name": version.name if version else None, "base_model": version.base_model if version else None, "creator": creator.username if creator else None, + "triggers": triggers, } ) return results