From 7a3f8b630c4e20bc8a43686f79b16eee8ed30d4e Mon Sep 17 00:00:00 2001
From: HuangHai <10402852@qq.com>
Date: Thu, 27 Mar 2025 11:13:37 +0800
Subject: [PATCH] 'commit'

---
 AI/WxMini/Start.py           | 5 ++---
 AI/WxMini/Utils/MySQLUtil.py | 3 +++
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/AI/WxMini/Start.py b/AI/WxMini/Start.py
index 51b4b411..5d73ab14 100644
--- a/AI/WxMini/Start.py
+++ b/AI/WxMini/Start.py
@@ -1,6 +1,5 @@
 import asyncio
 import logging
-import time
 import uuid
 from contextlib import asynccontextmanager
 from datetime import datetime, timedelta
@@ -17,12 +16,12 @@ from WxMini.Milvus.Config.MulvusConfig import *
 from WxMini.Milvus.Utils.MilvusCollectionManager import MilvusCollectionManager
 from WxMini.Milvus.Utils.MilvusConnectionPool import *
 from WxMini.Utils.EmbeddingUtil import text_to_embedding
-from WxMini.Utils.MySQLUtil import init_mysql_pool, save_chat_to_mysql, get_chat_log_by_session, get_user_by_login_name, \
+from WxMini.Utils.ImageUtil import *
+from WxMini.Utils.MySQLUtil import init_mysql_pool, get_chat_log_by_session, get_user_by_login_name, \
     get_chat_logs_by_risk_flag, get_chat_logs_summary
 from WxMini.Utils.MySQLUtil import update_risk, get_last_chat_log_id
 from WxMini.Utils.OssUtil import upload_mp3_to_oss_from_memory, get_sts_token
 from WxMini.Utils.TtsUtil import TTS
-from WxMini.Utils.ImageUtil import *
 
 # 配置日志
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
diff --git a/AI/WxMini/Utils/MySQLUtil.py b/AI/WxMini/Utils/MySQLUtil.py
index 9ffa4234..24d911ce 100644
--- a/AI/WxMini/Utils/MySQLUtil.py
+++ b/AI/WxMini/Utils/MySQLUtil.py
@@ -101,6 +101,9 @@ async def get_chat_log_by_session(mysql_pool, person_id, page=1, page_size=10):
                     "model_response": record['model_response'],
                     "audio_url": record['audio_url'],
                     "duration": record['duration'],
+                    "input_type": record['input_type'],
+                    "output_type": record['output_type'],
+                    "input_image_type": record['input_image_type'],
                     "create_time": record['create_time'].strftime("%Y-%m-%d %H:%M:%S")
                 }
                 for record in records