| | |
| | | f"&file_type={file_type}") |
| | | excel_url = build_file_url(excel_name, 'excel') |
| | | image_url = build_file_url(image_name, 'image') |
| | | try: |
| | | SessionService(db).update_session(chat_id, message={"content": output, "role": "assistant"}) |
| | | except Exception as e: |
| | | logger.error(f"Unexpected error when update_session: {e}") |
| | | if excel_url or data.get("e", ""): |
| | | try: |
| | | SessionService(db).update_session(chat_id, |
| | | message={ |
| | | "content": output, |
| | | "excel_url": excel_url, |
| | | "image_url": image_url, |
| | | "sql": data.get("sql", ""), |
| | | "code": data.get("code", ""), |
| | | "e": data.get("e", ""), |
| | | "role": "assistant"}) |
| | | except Exception as e: |
| | | logger.error(f"Unexpected error when update_session: {e}") |
| | | # 发送结果给客户端 |
| | | data["type"] = "message" |
| | | data["message"] = output |