Update app.py

app.py

@@ -112,7 +112,7 @@ async def get_db(request: Request) -> Database:
 async def get_chat(database: Database = Depends(get_db)) -> Response:
     msgs = await database.get_messages()
     return Response(
-        b'\n'.join(json.dumps(to_ds_message(m)).encode('utf-8') for m in msgs),
+        b'\n'.join(json.dumps(to_chat_message(m)).encode('utf-8') for m in msgs),
         media_type='text/plain',
     )
 
@@ -435,12 +435,12 @@ async def post_chat(
         nn = len(prompt)
         prompt = prompt[1:nn]
         print(">>>", prompt, "<<<")
-        return StreamingResponse(stream_messages(prompt), media_type='text/plain')
+        return StreamingResponse(stream_messages(), media_type='text/plain')
     elif prompt[0] != "@" :    
         #print("biasa")
-        return StreamingResponse(stream_messages(prompt), media_type='text/plain')
+        return StreamingResponse(stream_messages(), media_type='text/plain')
     print("** selesai **")
-    return StreamingResponse(stream_messages(prompt), media_type='text/plain')
+    return StreamingResponse(stream_messages(), media_type='text/plain')
 
 P = ParamSpec('P')
 R = TypeVar('R')