refac: prompts

This commit is contained in:
Timothy Jaeryang Baek 2024-11-16 20:52:57 -08:00
parent 41bad9abcb
commit b81cd15ced

View File

@ -71,16 +71,14 @@ class PromptsTable:
prompt = PromptModel( prompt = PromptModel(
**{ **{
"user_id": user_id, "user_id": user_id,
"command": form_data.command, **form_data.model_dump(),
"title": form_data.title,
"content": form_data.content,
"timestamp": int(time.time()), "timestamp": int(time.time()),
} }
) )
try: try:
with get_db() as db: with get_db() as db:
result = Prompt(**prompt.dict()) result = Prompt(**prompt.model_dump())
db.add(result) db.add(result)
db.commit() db.commit()
db.refresh(result) db.refresh(result)