'commit'

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")

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