Version 0.1.1 rise to flask app factory

Html/apps/sockets/namespaces.py

@@ -0,0 +1,104 @@
+# myapp/sockets/namespaces.py
+import json
+from flask import current_app, request, session
+from flask_socketio import Namespace, join_room, leave_room, emit
+from ..services.memory_service import save_chapter_memory_by_uuid
+from ..services.backboard_service import realtime_response
+class VSCodeNamespace(Namespace):
+    def on_login(self,data):        
+        ex = current_app.extensions
+        username2uuid    = ex["username2uuid"]
+        backboard_manager = ex["backboard_manager"]
+        print("VSCode client connected")
+        print(data)
+        dataconfig = data['config']
+        user_id = dataconfig.get('user_id')
+        path = dataconfig.get('path')
+        course_id = dataconfig.get('course_id')
+        lesson_id = dataconfig.get('chapter_id')
+        print(f"User {user_id} connected with path: {path}")
+        useruuid = username2uuid[user_id]
+        join_room(useruuid, namespace='/vscode')
+        if backboard_manager.get_backboard(useruuid) == None:
+            backboard_manager.add_backboard(useruuid,user_id, course_id, lesson_id, path)
+
+
+    def on_message(self, data):
+        print(f"Received from VSCode client: {data}")
+        dataconfig = data['config']
+        realtime_response(dataconfig, data)
+        # emit('response', {'message': 'Config data received and connection established'})
+    def on_disconnect(self, data):
+        print("VSCode client disconnected")
+        print("Disconnect reason:"+str(data))
+
+
+class AgentNamespace(Namespace):
+    def on_login(self, data):
+        ex = current_app.extensions
+        username2uuid    = ex["username2uuid"]
+        backboard_manager = ex["backboard_manager"]
+        agent_manager = ex["agent_manager"]
+        data = json.loads(data)
+        user = data['username']
+        course_id = data['course_id']
+        chapter_id = data['chapter_id']
+        user_uuid = username2uuid[user]
+        session['user_id'] = user_uuid
+        print(f'User connected with session user_id: {user_uuid}')
+        # 从markdown、markdown_prompts、score_prompts中各读取数据folder.md并进行new_agent
+        with open(f'books/markdown/{course_id}/{chapter_id}.md','r',encoding='UTF-8') as fmd,\
+            open(f'books/markdown_prompts/{course_id}/{chapter_id}.md','r',encoding='UTF-8')as fmdp,\
+            open(f'books/score_prompts/{course_id}/{chapter_id}.md','r',encoding='UTF-8') as fsp:
+            markdown = fmd.read()
+            markdown_prompts = fmdp.read()
+            score_prompts = fsp.read()
+            user_uuid, agent = agent_manager.new_agent(course_id, chapter_id, markdown, markdown_prompts, score_prompts, id=user_uuid, 
+                                                       root_path=f'../../study/{user}/{course_id}/{chapter_id}')
+            print(user_uuid)
+            join_room(user_uuid, namespace='/agent')  # 将该用户加入以user_id为名的room
+        backboard_manager.add_backboard(user_uuid, user, course_id, chapter_id, root_path=f'../../study/{user}/{course_id}/{chapter_id}')
+
+    def on_language(self, language):
+        ex = current_app.extensions
+        agent_manager = ex["agent_manager"]
+        id = session.get('user_id')
+        agent_manager.change_language(id, language)
+
+    def on_message(self, data):
+        print(f"Message from client: {data}")
+        ex = current_app.extensions
+        agent_manager = ex["agent_manager"]
+        backboard_manager = ex["backboard_manager"]
+        id = session.get('user_id')
+        if (type(data)==str):
+            data = json.loads(data)
+        print(id)
+        if data['type'] == 'text':
+            res = agent_manager.invoke(id, data['data'], backboard_manager.get_backboard(id).get_info_prompt())
+            print('=*='*20)
+            print(res.content)
+            reply = f"{res.content['speak']}"
+            with current_app.app_context(): 
+                emit('message',reply, room=id, namespace='/agent')
+                emit('request_function',res.content['function'], room=id, namespace='/agent')
+
+        if data['type'] == 'function':
+            agent_manager.function_call(id, data['data'])
+
+    def on_initiative(self,data):
+        print("User active function call")
+        ex = current_app.extensions
+        agent_manager = ex["agent_manager"]
+        backboard_manager = ex["backboard_manager"]
+        user_id = session['user_id']
+        if data['name'] == 'sample_judge':
+            agent_manager.sample_judge(user_id, backboard_manager.get_backboard(user_id))
+        if data['name'] == 'judge':
+            agent_manager.judge(user_id, backboard_manager.get_backboard(user_id))
+
+
+
+    def on_disconnect(self,data):
+        print("VSCode client disconnected")
+        print("Disconnect reason:"+str(data))