try hello message

Html/apps/sockets/namespaces.py

@@ -8,6 +8,7 @@ from ..function_tools import next_chapter
 from ..services.backboard_service import realtime_response
 from ..services.markdown_service import load_full_markdown_file
 from ..extension_ase.ase_client import HSAEngineClient
+from ..services.user_service import get_or_load_current_user
 
 
 
@@ -53,8 +54,12 @@ class AgentNamespace(Namespace):
         user_id = uuid2username[user_uuid]
         session['user_uuid'] = user_uuid
         print(f'User connected with session user_uuid: {user_uuid}')
+
         user_uuid2chatmanager = ex["user_uuid2chatmanager"]
         user_uuid2ase_client = ex["user_uuid2ase_client"]
+        # if (user_uuid2ase_client[user_uuid] is not None and user_uuid2ase_client[user_uuid].connected):
+        #     user_uuid2ase_client[user_uuid].disconnect()
+        chatmanager = user_uuid2chatmanager[user_uuid]
         user_uuid2ase_client[user_uuid] = HSAEngineClient(current_app.config["ASE_ENGINE_URL"], current_app.config["ASE_ENGINE_NAMESPACE"],id=user_id)
         user_uuid2ase_client[user_uuid].connect()
         user_uuid2ase_client[user_uuid].register_route_with_entry(
@@ -75,14 +80,22 @@ class AgentNamespace(Namespace):
                                             entry=self,               
                                             init_data={'room': user_uuid}          
                                         )
+        user_uuid2ase_client[user_uuid].register_route_with_entry(
+                                            route='chapter_score',
+                                            callback=self.receive_ase_chapter_score,
+                                            entry=(get_or_load_current_user(),chatmanager),
+                                            init_data={'room': user_uuid}
+                                        )                            
+        
         fmd,fmdp,fsp = load_full_markdown_file(course_id, chapter_name, lesson_name)
-        chatmanager = user_uuid2chatmanager[user_uuid]
-        chatmanager.init(user_uuid, user_uuid2ase_client[user_uuid], fmd, fmdp, fsp, max_iter=5, app=current_app, socketio=self)
+        
+        chatmanager.init(user_uuid, user_uuid2ase_client[user_uuid], fmd, fmdp, fsp, course_id, chapter_name, lesson_name, max_iter=5, app=current_app, socketio=self)
+        user_uuid2ase_client.chatmanager = chatmanager
         # 注册函数，以确保一些初始参数
         chatmanager.function_manager.add_function(judge, {'course_id': course_id, 'root_path': f'/home/{user_id}/{course_id}/{chapter_name}/{lesson_name}'})
         chatmanager.function_manager.add_function(next_chapter, {'user_uuid': user_uuid})
-        
-
+        chatmanager.chat_historys = []
+        chatmanager.scores = []
         join_room(user_uuid, namespace='/agent')  # 将该用户加入以user_id为名的room
         backboard_manager.add_backboard(user_uuid, user_id, course_id, lesson_name, root_path=f'/home/{user_id}/{course_id}/{chapter_name}/{lesson_name}')
         chatmanager.bb = backboard_manager.get_backboard(user_uuid)
@@ -102,13 +115,8 @@ class AgentNamespace(Namespace):
             data = json.loads(data)
         if data['type'] == 'text':
             res = user_uuid2ase_client[uuid].send_text('dialog', data['data'])
+            chatmanager.chat_historys.append({'role': 'user', 'content': data['data']})
             print(f"Send text to route 'dialog' success: {res}")
-            # print('=*='*20)
-            # print(res.content)
-            # reply = f"{res.content['speak']}"
-            # with current_app.app_context(): 
-            #     emit('message',reply, room=uuid, namespace='/agent')
-            #     emit('request_function',res.content['function'], room=uuid, namespace='/agent')
 
         if data['type'] == 'function': # 在这里 前端会发送允许执行的function与参数
             print("function_call", data['data'])
@@ -117,11 +125,13 @@ class AgentNamespace(Namespace):
             d['data'] = res.to_dict()
             print("function_call_res", d['data'])
             emit(d['route'], d, room=uuid, namespace='/agent')
+            chatmanager.chat_historys.append({'role': 'user', 'content': data['data']})
             user_uuid2ase_client[uuid].send_text(d['route'], data['data'])
 
 
     def receive_ase_dialog(client_entry: HSAEngineClient, namespace_entry: Namespace, data, init_data):
         print("receive_ase_dialog", data)
+        client_entry.chatmanager.chat_historys.append({'role': 'assistant', 'content': data['data']})
         namespace_entry.emit('message', data['data'], room=init_data['room'], namespace='/agent')
 
     def receive_ase_judge(client_entry: HSAEngineClient, namespace_entry: Namespace, data, init_data):
@@ -132,6 +142,16 @@ class AgentNamespace(Namespace):
         print("receive_ase_next_chapter", data)
         namespace_entry.emit('request_function',data, room=init_data['room'], namespace='/agent')  # data 是一个列表 /dict也支持
 
+    def receive_ase_chapter_score(client_entry: HSAEngineClient, user_and_manager, data, init_data):
+        print("load_next_chapter")
+        user, chatmanager = user_and_manager
+        if len(chatmanager.scores) >= chatmanager.chapter_chain_now:
+            chatmanager.scores[chatmanager.chapter_chain_now - 1] = data.get('data', None)
+        else: chatmanager.scores.append(data.get('data', None))
+        user.set_course_progress(chatmanager.chat_historys, chatmanager.scores, chatmanager.material_id, chatmanager.chapter_name, chatmanager.lesson_name)
+        chatmanager.load_next_chapter()
+
+
     def on_initiative(self,data): # 主动call
         print("User active function call")
         ex = current_app.extensions