(session: SessionDep, record_id: int, data: str = '')
| 1066 | |
| 1067 | |
| 1068 | def save_predict_data(session: SessionDep, record_id: int, data: str = '') -> ChatRecord: |
| 1069 | if not record_id: |
| 1070 | raise Exception("Record id cannot be None") |
| 1071 | record = get_chat_record_by_id(session, record_id) |
| 1072 | |
| 1073 | record.predict_data = data |
| 1074 | |
| 1075 | result = ChatRecord(**record.model_dump()) |
| 1076 | |
| 1077 | stmt = update(ChatRecord).where(and_(ChatRecord.id == record.id)).values( |
| 1078 | predict_data=record.predict_data |
| 1079 | ) |
| 1080 | |
| 1081 | session.execute(stmt) |
| 1082 | |
| 1083 | session.commit() |
| 1084 | |
| 1085 | return result |
| 1086 | |
| 1087 | |
| 1088 | def save_error_message(session: SessionDep, record_id: int, message: str) -> ChatRecord: |
no test coverage detected