84 lines
4.0 KiB
Python
84 lines
4.0 KiB
Python
import requests
|
||
from flask_socketio import Namespace
|
||
|
||
|
||
def clear_user_session(asengine_url, asengine_token, user_id):
|
||
'''
|
||
快速清理用户会话的函数
|
||
'''
|
||
try:
|
||
base_url = asengine_url
|
||
clear_url = f"{base_url}/api/sync/clear_user_session"
|
||
payload = {
|
||
'namespace_url': asengine_token, # WebSocket命名空间URL
|
||
'user_id': user_id, # 用户ID，对应session_info.id
|
||
'clear_type': 'all' # 清理类型，默认为全部
|
||
}
|
||
headers = {
|
||
'Content-Type': 'application/json'
|
||
}
|
||
response = requests.post(clear_url, json=payload, headers=headers)
|
||
response.raise_for_status() # 检查请求是否成功
|
||
print(f"User session {user_id} cleared successfully. Response: {response.json()}")
|
||
except Exception as e:
|
||
print(f"Error clearing user session: {e}")
|
||
|
||
|
||
def on_connect_to_ase(client, entry, init_data, **kwargs):
|
||
'''
|
||
此时刚刚建立好aseclient连接，并向前端发送打开code-server-like指令。
|
||
'''
|
||
namespace_entry, ase_client = entry
|
||
namespace_entry.emit('open-iframe',"", room=init_data['room'], namespace='/agent')
|
||
namespace_entry.emit('message', "Agents服务已连接, 加载教案中", room=init_data['room'], namespace='/agent')
|
||
|
||
clear_user_session(init_data['asengine_url'], init_data['asengine_token'], init_data['room'])
|
||
|
||
restart = init_data.get("restart", False)
|
||
ase_client.chatmanager.load_now_chapter(load_code=restart)
|
||
print("load_now_chapter with restart:", restart)
|
||
if restart:
|
||
ase_client.send_text("chapter-start", "")
|
||
namespace_entry.emit('message', "教案加载完毕", room=init_data['room'], namespace='/agent')
|
||
|
||
|
||
#接受消息回来直接让message监听发送
|
||
def receive_ase_paste_detected(client_entry, namespace_entry, data, init_data):
|
||
print("receive_ase_paste_detected", 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_dialog(client_entry, namespace_entry, 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_message_hint(client_entry, namespace_entry, data, init_data):
|
||
print("receive_ase_message_hint", data)
|
||
namespace_entry.emit('message-hint', data['data'], room=init_data['room'], namespace='/agent')
|
||
|
||
#接受语音信息
|
||
def receive_ase_voice_out(client_entry, namespace_entry, data, init_data):
|
||
namespace_entry.emit('voiceMessage', data['data'], room=init_data['room'], namespace='/agent')
|
||
|
||
def receive_ase_process(client_entry, namespace_entry, data, init_data):
|
||
namespace_entry.emit('process', data['data'], room=init_data['room'], namespace='/agent')
|
||
|
||
def receive_ase_judge(client_entry, namespace_entry, data, init_data):
|
||
print("receive_ase_judge", data)
|
||
namespace_entry.emit('request_function', data, room=init_data['room'], namespace='/agent') # data 是一个列表 /dict也支持
|
||
|
||
def receive_ase_next_chapter(client_entry, namespace_entry, data, init_data):
|
||
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, 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.course_id, chatmanager.chapter_name, chatmanager.lesson_name)
|
||
chatmanager.load_now_chapter()
|
||
chatmanager.ase_client.send_text("chapter-start", "")
|