Compare commits
22 Commits
a8bccd7c3f
...
main
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
36bbedd3cf | ||
|
|
0e3703b1f4 | ||
|
|
52696707dd | ||
|
|
0a812e496f | ||
|
|
de1ba08d7e | ||
|
|
3f14cbd6c1 | ||
|
|
c51565199c | ||
|
|
8d29a26d7d | ||
|
|
decf24020a | ||
|
|
3dc219adc2 | ||
|
|
40d1341685 | ||
|
|
0d606fb171 | ||
|
|
e1f32ec980 | ||
|
|
ad8d7d069c | ||
|
|
687b316db6 | ||
|
|
60248d5c51 | ||
|
|
befec5543c | ||
|
|
af24c51a2f | ||
|
|
fa91696aae | ||
|
|
e698c50a4b | ||
|
|
75795dd853 | ||
|
|
5b8180b48e |
@@ -39,16 +39,19 @@ class ASEngineClient:
|
||||
# 连接事件
|
||||
@self.sio.on('connect', namespace=self.namespace)
|
||||
def on_connect():
|
||||
print("---------------ASEngineClient on_connect----------------")
|
||||
self.sid = self.sio.sid
|
||||
print(f"Connected to server with sid: {self.sid}")
|
||||
|
||||
self.connected = True
|
||||
self.sid = self.sio.get_sid(namespace=self.namespace)
|
||||
print(f"Connected to server. SID: {self.sid}")
|
||||
self.sio.emit('join', {'id': self.id}, namespace=self.namespace)
|
||||
if self._on_connect_callback is not None:
|
||||
self._on_connect_callback(
|
||||
self,
|
||||
self._on_connect_callback_entry,
|
||||
init_data=self._on_connect_callback_initData
|
||||
)
|
||||
|
||||
|
||||
# 断开连接事件
|
||||
@self.sio.on('disconnect', namespace=self.namespace)
|
||||
def on_disconnect():
|
||||
|
||||
@@ -14,7 +14,7 @@ from typing import Optional
|
||||
mongo = PyMongo()
|
||||
|
||||
|
||||
socketio = SocketIO(cors_allowed_origins="*",max_payload_size=52428800, async_mode="eventlet") # 不直接传 app
|
||||
socketio = SocketIO(cors_allowed_origins="*",max_payload_size=52428800, async_mode="eventlet",always_connect=False) # 不直接传 app
|
||||
cors = CORS()
|
||||
# ===== 你的全局对象 =====
|
||||
# Backboard
|
||||
@@ -38,7 +38,7 @@ def init_extensions(app):
|
||||
cors_allowed_origins='*',#app.config["VSCODE_WEB_URL"],
|
||||
ping_timeout=app.config["SOCKETIO_PING_TIMEOUT"],
|
||||
ping_interval=app.config["SOCKETIO_PING_INTERVAL"],
|
||||
max_payload_size=50000000, async_mode="eventlet"
|
||||
max_payload_size=50000000, async_mode="eventlet",always_connect=False
|
||||
)
|
||||
cors.init_app(
|
||||
app,
|
||||
|
||||
@@ -1,6 +1,4 @@
|
||||
import requests
|
||||
from flask_socketio import Namespace
|
||||
|
||||
|
||||
def clear_user_session(asengine_url, asengine_token, user_id):
|
||||
'''
|
||||
@@ -28,9 +26,12 @@ def on_connect_to_ase(client, entry, init_data, **kwargs):
|
||||
'''
|
||||
此时刚刚建立好aseclient连接,并向前端发送打开code-server-like指令。
|
||||
'''
|
||||
namespace_entry, ase_client = entry
|
||||
namespace_entry, ase_client, logger = entry
|
||||
logger.info(f"on_connect_to_ase with init_data: {init_data}")
|
||||
namespace_entry.emit('open-iframe',"", room=init_data['room'], namespace='/agent')
|
||||
namespace_entry.emit('message', "Agents服务已连接, 加载教案中", room=init_data['room'], namespace='/agent')
|
||||
namespace_entry.emit('message', f'{init_data}', room=init_data['room'], namespace='/agent')
|
||||
|
||||
|
||||
restart = init_data.get("restart", False)
|
||||
# 不再需要手动next_chapter,因为已经在on_login中根据mongo进度恢复了正确的章节
|
||||
|
||||
@@ -93,7 +93,6 @@ def get_multiagents_dialog_list(user_id: str) -> List[Dict]:
|
||||
|
||||
# 检查响应
|
||||
if response.status_code == 200:
|
||||
current_app.logger.info(f"获取MultiAgents对话列表成功: user_id={user_id}, json={response.json()}")
|
||||
return response.json()
|
||||
else:
|
||||
current_app.logger.error(f"获取MultiAgents对话列表失败: HTTP {response.status_code}")
|
||||
|
||||
@@ -63,6 +63,12 @@ class VSCodeNamespace(Namespace):
|
||||
|
||||
|
||||
class AgentNamespace(Namespace):
|
||||
def __init__(self, namespace):
|
||||
super().__init__(namespace)
|
||||
# 使用字典存储每个用户的chatmanager和ase_client,避免多用户冲突
|
||||
self.user_chatmanagers = {}
|
||||
self.user_ase_clients = {}
|
||||
|
||||
def _load_learning_progress(self, chatmanager, user_id, course_id, chapter_name, lesson_name, continue_learn):
|
||||
"""加载并恢复用户学习进度
|
||||
|
||||
@@ -120,8 +126,8 @@ class AgentNamespace(Namespace):
|
||||
chapter_name = data['chapter_name']
|
||||
user_id = uuid2username[user_uuid]
|
||||
session['user_uuid'] = user_uuid
|
||||
join_room(user_uuid, namespace='/agent') # 将该用户加入以user_id为名的room
|
||||
print(f'User connected with session user_uuid: {user_uuid}')
|
||||
join_room(user_uuid)
|
||||
|
||||
user_uuid2chatmanager = ex["user_uuid2chatmanager"]
|
||||
user_uuid2ase_client = ex["user_uuid2ase_client"]
|
||||
@@ -152,8 +158,9 @@ class AgentNamespace(Namespace):
|
||||
if chatmanager.chat_historys:
|
||||
emit('messages', chatmanager.chat_historys, room=user_uuid, namespace='/agent')
|
||||
|
||||
self.chatmanager = chatmanager
|
||||
self.ase_client = user_uuid2ase_client[user_uuid]
|
||||
# 存储每个用户的chatmanager和ase_client到字典中,使用用户UUID作为键
|
||||
self.user_chatmanagers[user_uuid] = chatmanager
|
||||
self.user_ase_clients[user_uuid] = user_uuid2ase_client[user_uuid]
|
||||
|
||||
# 将restart设置为False,避免后续重复处理
|
||||
restart = chatmanager.restart
|
||||
@@ -161,14 +168,14 @@ class AgentNamespace(Namespace):
|
||||
|
||||
user_uuid2ase_client[user_uuid].register_on_connect_entry(
|
||||
callback=on_connect_to_ase,
|
||||
entry=(self, user_uuid2ase_client[user_uuid]),
|
||||
entry=(self, user_uuid2ase_client[user_uuid], current_app.logger),
|
||||
init_data={'room':user_uuid, 'restart':restart}
|
||||
)
|
||||
user_uuid2ase_client[user_uuid].register_route_with_entry(
|
||||
route='dialog',
|
||||
callback=receive_ase_dialog,
|
||||
entry=self,
|
||||
init_data={'room': user_uuid}
|
||||
init_data={'room':user_uuid}
|
||||
)
|
||||
user_uuid2ase_client[user_uuid].register_route_with_entry(
|
||||
route='dialog-hint',
|
||||
@@ -186,13 +193,13 @@ class AgentNamespace(Namespace):
|
||||
route='judge',
|
||||
callback=receive_ase_judge,
|
||||
entry=self,
|
||||
init_data={'room': user_uuid}
|
||||
init_data={'room':user_uuid}
|
||||
)
|
||||
user_uuid2ase_client[user_uuid].register_route_with_entry(
|
||||
route='next_chapter',
|
||||
callback=receive_ase_next_chapter,
|
||||
entry=self,
|
||||
init_data={'room': user_uuid}
|
||||
init_data={'room':user_uuid}
|
||||
)
|
||||
user_uuid2ase_client[user_uuid].register_route_with_entry(
|
||||
route='chapter_score',
|
||||
@@ -286,4 +293,4 @@ class AgentNamespace(Namespace):
|
||||
leave_room(uuid, namespace='/agent')
|
||||
|
||||
self.app.logger.info("VSCode client disconnected")
|
||||
self.app.logger.info(f"Disconnect reason: {str(data)}")
|
||||
self.app.logger.info(f"Disconnect reason: {str(data)}")
|
||||
@@ -66,25 +66,23 @@ function cleanupExpiredApprovals() {
|
||||
// 定时清理过期记录
|
||||
setInterval(cleanupExpiredApprovals, 3000); // 每3秒检查一次
|
||||
document.addEventListener('DOMContentLoaded', function () {
|
||||
data = window.appData;
|
||||
user_data = window.appData;
|
||||
const host = window.location.host;
|
||||
console.log(`Connecting to WebSocket server at wss://${host}/agent`);
|
||||
socket = io(`wss://${host}/agent`, {
|
||||
query: {
|
||||
user_uuid: data.user_uuid,
|
||||
folder: data.folder
|
||||
user_uuid: user_data.user_uuid,
|
||||
folder: user_data.folder
|
||||
}
|
||||
});
|
||||
|
||||
// 初始化聊天消息管理器
|
||||
chatManager = new ChatMessageManager();
|
||||
|
||||
socket.on('connect', function () {
|
||||
console.log('Connected to server');
|
||||
socket.emit('login',JSON.stringify(data));
|
||||
console.log('Connected to WebSocket server'+socket.id+"with user"+JSON.stringify(user_data));
|
||||
socket.emit('login',JSON.stringify(user_data));
|
||||
});
|
||||
socket.on('open-iframe', function(data) {
|
||||
OpenIframe();
|
||||
socket.on('open-iframe', function() {
|
||||
OpenIframe(user_data);
|
||||
});
|
||||
// 监听来自服务器的消息
|
||||
socket.on('voiceMessage', function (data) {
|
||||
|
||||
@@ -208,7 +208,7 @@
|
||||
console.error('Error fetching session:', error);
|
||||
});
|
||||
let lastiframe = null;
|
||||
function OpenIframe(){
|
||||
function OpenIframe(data){
|
||||
// 等待1s再加载
|
||||
setTimeout(() => {
|
||||
// 在成功获取 session 后创建并插入 iframe
|
||||
|
||||
@@ -187,7 +187,7 @@
|
||||
|
||||
|
||||
let lastIframe = null;
|
||||
function OpenIframe(){
|
||||
function OpenIframe(data){
|
||||
// 在成功获取 session 后创建并插入 iframe
|
||||
const iframe = document.createElement('iframe');
|
||||
const div=document.getElementById('vscodeWeb')
|
||||
|
||||
@@ -91,38 +91,6 @@ def desktop(user_uuid, course_id, chapter_name, lesson_name):
|
||||
|
||||
|
||||
|
||||
caddy_conf = f"""
|
||||
handle_path /vsc{user_id}/* {{
|
||||
reverse_proxy localhost:{code_server_port}
|
||||
}}
|
||||
"""
|
||||
caddy_config_path = "/etc/caddy/Caddyfile"
|
||||
with open(caddy_config_path, "r") as f:
|
||||
lines = f.readlines()
|
||||
pattern = re.compile(
|
||||
r'handle_path\s+/vsc' + re.escape(user_id) + r'/\* {.*?reverse_proxy.*?}',
|
||||
re.DOTALL
|
||||
)
|
||||
for i in range(len(lines)):
|
||||
if (lines[i].strip()==''): lines[i]=''
|
||||
content = "".join(lines)
|
||||
if pattern.search(content):
|
||||
new_content = pattern.sub('', content)
|
||||
lines = new_content.splitlines(keepends=True)
|
||||
if len(lines) >= 2:
|
||||
insert_position = len(lines) - 2
|
||||
else:
|
||||
insert_position = len(lines)
|
||||
if caddy_conf[-1] not in ['\n', '\r']:
|
||||
caddy_conf += '\n'
|
||||
lines.insert(insert_position, caddy_conf)
|
||||
with open(caddy_config_path, "w") as f:
|
||||
f.writelines(lines)
|
||||
|
||||
try:
|
||||
subprocess.run(["sudo", "caddy", "reload", "--config", "/etc/caddy/Caddyfile"])
|
||||
except subprocess.CalledProcessError as e:
|
||||
print(f"Error reload caddy {e}")
|
||||
|
||||
cfg = current_app.config["VSCODE_WEB_PATH"]
|
||||
path_for_vscode = path_dir
|
||||
|
||||
@@ -3,12 +3,12 @@ from flask_socketio import SocketIO
|
||||
from .config import Config
|
||||
# 初始化扩展
|
||||
cors = CORS()
|
||||
socketio = SocketIO(path='vsc-like-ws',cors_allowed_origins="*",max_payload_size=5242880, async_mode="eventlet") # 不直接传 app
|
||||
socketio = SocketIO(path='vsc-like-ws',cors_allowed_origins="*",max_payload_size=5242880, async_mode="eventlet",always_connect=False) # 不直接传 app
|
||||
|
||||
session_paths = {}
|
||||
def init_extensions(app):
|
||||
cors.init_app(app, resources={r"/*": {"origins": "*"}}, supports_credentials=True)
|
||||
socketio.init_app(app, cors_allowed_origins="*", max_payload_size=5242880, async_mode="eventlet")
|
||||
socketio.init_app(app, cors_allowed_origins="*", max_payload_size=5242880, async_mode="eventlet",always_connect=False)
|
||||
app.extensions["session_paths"] = session_paths
|
||||
|
||||
def register_namespaces(app):
|
||||
|
||||
Reference in New Issue
Block a user