diff --git a/telegram-application/sql_scripts/create_table.sql b/telegram-application/sql_scripts/create_table.sql index 7cfa7f8..0cddd72 100644 --- a/telegram-application/sql_scripts/create_table.sql +++ b/telegram-application/sql_scripts/create_table.sql @@ -33,5 +33,6 @@ CREATE TABLE tg_messages ( CREATE TABLE prompts ( id SERIAL PRIMARY KEY , prompt TEXT NOT NULL, + role TEXT NOT NULL, created_at TIMESTAMP DEFAULT NOW() NOT NULL ) diff --git a/telegram-application/src/core/admin/promt_admin.py b/telegram-application/src/core/admin/promt_admin.py index de1e38f..b02ef73 100644 --- a/telegram-application/src/core/admin/promt_admin.py +++ b/telegram-application/src/core/admin/promt_admin.py @@ -12,11 +12,13 @@ class PromptAdmin(ModelView, model=Prompt): column_list = [ Prompt.prompt, - Prompt.created_at + Prompt.created_at, + Prompt.role, ] column_searchable_list = [ - Prompt.prompt + Prompt.prompt, + Prompt.role, ] column_sortable_list = [ diff --git a/telegram-application/src/core/database/prompt.py b/telegram-application/src/core/database/prompt.py index c87f48f..3e80eaa 100644 --- a/telegram-application/src/core/database/prompt.py +++ b/telegram-application/src/core/database/prompt.py @@ -12,6 +12,7 @@ class Prompt(Base): primary_key=True, autoincrement=True, ) + role: Mapped[str] prompt: Mapped[str] created_at: Mapped[datetime] = mapped_column( default=datetime.utcnow,