(session: SessionDep, record_id: int, answer: str)
| 918 | |
| 919 | |
| 920 | def save_predict_answer(session: SessionDep, record_id: int, answer: str) -> ChatRecord: |
| 921 | if not record_id: |
| 922 | raise Exception("Record id cannot be None") |
| 923 | |
| 924 | stmt = update(ChatRecord).where(and_(ChatRecord.id == record_id)).values( |
| 925 | predict=answer, |
| 926 | ) |
| 927 | |
| 928 | session.execute(stmt) |
| 929 | |
| 930 | session.commit() |
| 931 | |
| 932 | record = get_chat_record_by_id(session, record_id) |
| 933 | |
| 934 | return record |
| 935 | |
| 936 | |
| 937 | def save_select_datasource_answer(session: SessionDep, record_id: int, answer: str, |
no test coverage detected