(session: SessionDep, record_id: int, sql: str)
| 1008 | |
| 1009 | |
| 1010 | def save_sql(session: SessionDep, record_id: int, sql: str) -> ChatRecord: |
| 1011 | if not record_id: |
| 1012 | raise Exception("Record id cannot be None") |
| 1013 | |
| 1014 | record = get_chat_record_by_id(session, record_id) |
| 1015 | |
| 1016 | record.sql = sql |
| 1017 | |
| 1018 | result = ChatRecord(**record.model_dump()) |
| 1019 | |
| 1020 | stmt = update(ChatRecord).where(and_(ChatRecord.id == record.id)).values( |
| 1021 | sql=record.sql |
| 1022 | ) |
| 1023 | |
| 1024 | session.execute(stmt) |
| 1025 | |
| 1026 | session.commit() |
| 1027 | |
| 1028 | return result |
| 1029 | |
| 1030 | |
| 1031 | def save_chart_answer(session: SessionDep, record_id: int, answer: str) -> ChatRecord: |
no test coverage detected