Agent Chat fixed

Html/app.py

@@ -98,9 +98,10 @@ agent_manager = AgentManager()
 
 
 class AgentNamespace(Namespace):
-    def on_connect(self):
-        user = request.args.get('username')
-        folder_name = request.args.get('folder')
+    def on_login(self, data):
+        data = json.loads(data)
+        user = data['username']
+        folder_name = data['folder']
         user_uuid = user_id2uuid[user]
         session['user_id'] = user_uuid
         print(f'User connected with session user_id: {user_uuid}')
@@ -114,6 +115,7 @@ class AgentNamespace(Namespace):
             user_uuid, agent = agent_manager.new_agent(markdown, markdown_prompts, score_prompts, id=user_uuid)
             join_room(user_uuid)  # 将该用户加入以user_id为名的room
 
+        
     def on_message(self, data):
         print(f"Message from client: {data}")
         id = session.get('user_id')
@@ -122,7 +124,7 @@ class AgentNamespace(Namespace):
         res = agent_manager.invoke(id, data, backboard_manager.get_backboard(id).get_info_prompt())
         reply = f"AI: {res.content['speak']}"
         # 将回复发送回前端
-        emit('receive_message', reply)
+        emit('message', reply)
 
 
     def on_disconnect(self):