Browse Source

feat: 修复bug

zhangwl 4 weeks ago
parent
commit
3e4a3756d9
1 changed files with 2 additions and 2 deletions
  1. 2 2
      app/routers/chat.py

+ 2 - 2
app/routers/chat.py

@@ -146,7 +146,7 @@ async def voice_to_text(
     return {"text": text}
     return {"text": text}
 
 
 
 
-async def generate_stream_response(request: ChatRequest, user_context: UserContext):
+async def generate_stream_response(request: ChatRequest, user_context: UserContext, promptType: str):
     session_id = request.session_id
     session_id = request.session_id
     latest_user_msg = None
     latest_user_msg = None
     try:
     try:
@@ -577,7 +577,7 @@ async def chat(
     try:
     try:
         if request.stream:
         if request.stream:
             return StreamingResponse(
             return StreamingResponse(
-                generate_stream_response(request, user_context),
+                generate_stream_response(request, user_context, promptType),
                 media_type="text/plain",
                 media_type="text/plain",
                 headers={
                 headers={
                     "Cache-Control": "no-cache",
                     "Cache-Control": "no-cache",