104 lines
4.6 KiB
Python
104 lines
4.6 KiB
Python
# 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)) |