update: 封装了豆包模型

main.py

@@ -5,7 +5,7 @@ from app.concrete import Agent
 from app.model import Assistant, User, get_db
 from app.schemas import *
 from app.dependency import get_logger
-from app.exception import AsrResultNoneError, AbnormalLLMFrame, SideNoiseError, SessionNotFoundError
+from app.exception import *
 import uvicorn
 import uuid
 import json
@@ -222,13 +222,15 @@ async def streaming_chat(ws: WebSocket,db=Depends(get_db)):
         agent.save()
         logger.debug("音频保存成功")
     except AsrResultNoneError:
-        await ws.send_text(json.dumps({"type":"close","code":201,"msg":"asr结果为空"}, ensure_ascii=False))
+        await ws.send_text(json.dumps({"type":"error","code":501,"msg":"asr结果为空"}, ensure_ascii=False))
     except AbnormalLLMFrame as e:
-        await ws.send_text(json.dumps({"type":"close","code":202,"msg":str(e)}, ensure_ascii=False))
+        await ws.send_text(json.dumps({"type":"error","code":502,"msg":str(e)}, ensure_ascii=False))
     except SideNoiseError as e:
-        await ws.send_text(json.dumps({"type":"close","code":203,"msg":str(e)}, ensure_ascii=False))
+        await ws.send_text(json.dumps({"type":"error","code":503,"msg":str(e)}, ensure_ascii=False))
     except SessionNotFoundError:
-        await ws.send_text(json.dumps({"type":"close","code":204,"msg":"session未找到"}, ensure_ascii=False))
+        await ws.send_text(json.dumps({"type":"error","code":504,"msg":"session不存在"}, ensure_ascii=False))
+    except UnknownVolcEngineModelError:
+        await ws.send_text(json.dumps({"type":"error","code":505,"msg":"未知的火山引擎模型"}, ensure_ascii=False))
     logger.debug("WebSocket连接断开")
     await ws.close()
 # --------------------------------------------------------------------------------------------------------------------------------------------------------------------------------