(session: SessionDep, base_record: ChatRecord, action_type: str)
| 806 | |
| 807 | |
| 808 | def save_analysis_predict_record(session: SessionDep, base_record: ChatRecord, action_type: str) -> ChatRecord: |
| 809 | record = ChatRecord() |
| 810 | record.question = base_record.question |
| 811 | record.chat_id = base_record.chat_id |
| 812 | record.datasource = base_record.datasource |
| 813 | record.engine_type = base_record.engine_type |
| 814 | record.ai_modal_id = base_record.ai_modal_id |
| 815 | record.create_time = datetime.datetime.now() |
| 816 | record.create_by = base_record.create_by |
| 817 | record.chart = base_record.chart |
| 818 | record.data = base_record.data |
| 819 | |
| 820 | if action_type == 'analysis': |
| 821 | record.analysis_record_id = base_record.id |
| 822 | elif action_type == 'predict': |
| 823 | record.predict_record_id = base_record.id |
| 824 | |
| 825 | result = ChatRecord(**record.model_dump()) |
| 826 | |
| 827 | session.add(record) |
| 828 | session.flush() |
| 829 | session.refresh(record) |
| 830 | result.id = record.id |
| 831 | session.commit() |
| 832 | |
| 833 | return result |
| 834 | |
| 835 | |
| 836 | def start_log(session: SessionDep, ai_modal_id: int = None, ai_modal_name: str = None, operate: OperationEnum = None, |
no test coverage detected