107 lines
4.8 KiB
Python
107 lines
4.8 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
|
|
uuid2username = ex["uuid2username"]
|
|
backboard_manager = ex["backboard_manager"]
|
|
print("VSCode client connected")
|
|
print(data)
|
|
dataconfig = data['config']
|
|
user_uuid = dataconfig.get('user_uuid')
|
|
path = dataconfig.get('path')
|
|
course_id = dataconfig.get('course_id')
|
|
lesson_id = dataconfig.get('chapter_id')
|
|
print(f"User {user_uuid} connected with path: {path}")
|
|
join_room(user_uuid, namespace='/vscode')
|
|
if backboard_manager.get_backboard(user_uuid) == None:
|
|
backboard_manager.add_backboard(user_uuid,uuid2username[user_uuid], 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
|
|
uuid2username = ex["uuid2username"]
|
|
backboard_manager = ex["backboard_manager"]
|
|
agent_manager = ex["agent_manager"]
|
|
data = json.loads(data)
|
|
user_uuid = data['user_uuid']
|
|
course_id = data['course_id']
|
|
chapter_id = data['chapter_id']
|
|
user_id = uuid2username[user_uuid]
|
|
session['user_uuid'] = user_uuid
|
|
print(f'User connected with session user_uuid: {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_id}/{course_id}/{chapter_id}')
|
|
print(user_uuid)
|
|
join_room(user_uuid, namespace='/agent') # 将该用户加入以user_id为名的room
|
|
backboard_manager.add_backboard(user_uuid, user_id, course_id, chapter_id, root_path=f'../../study/{user_id}/{course_id}/{chapter_id}')
|
|
|
|
def on_language(self, language):
|
|
ex = current_app.extensions
|
|
agent_manager = ex["agent_manager"]
|
|
uuid2username = ex["uuid2username"]
|
|
uuid = session.get('user_uuid')
|
|
user_id = uuid2username[uuid]
|
|
agent_manager.change_language(uuid, 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"]
|
|
uuid = session.get('user_uuid')
|
|
uuid2username = ex["uuid2username"]
|
|
user_id = uuid2username[uuid]
|
|
if (type(data)==str):
|
|
data = json.loads(data)
|
|
if data['type'] == 'text':
|
|
res = agent_manager.invoke(uuid, data['data'], backboard_manager.get_backboard(user_id).get_info_prompt())
|
|
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':
|
|
agent_manager.function_call(uuid, 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"]
|
|
uuid2username = ex["uuid2username"]
|
|
user_id = uuid2username[session['user_uuid']]
|
|
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)) |