laoxong/AstrBot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

modified: astrbot/core/pipeline/process_stage/method/llm_request.py

Browse Source 
new file:   astrbot/core/star/session_llm_manager.py
	modified:   astrbot/dashboard/routes/session_management.py
	modified:   dashboard/src/views/SessionManagementPage.vue

 增加了精确到会话的LLM启停管理以及插件启停管理
This commit is contained in:
advent259141
2025-06-14 03:42:21 +08:00
parent f26cf6ed6f
commit bfc8024119
4 changed files with 175 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
astrbot/core/pipeline/process_stage/method/llm_request.py
+7
View File
@@ -26,6 +26,7 @@ from astrbot.core.provider.entities import (
)
from astrbot.core.star.star_handler import star_handlers_registry, EventType
from astrbot.core.star.star import star_map
from astrbot.core.star.session_llm_manager import SessionLLMManager
from mcp.types import (
TextContent,
ImageContent,
@@ -71,6 +72,12 @@ class LLMRequestSubStage(Stage):
if not self.ctx.astrbot_config["provider_settings"]["enable"]:
logger.debug("未启用 LLM 能力，跳过处理。")
return
# 检查会话级别的LLM启停状态
if not SessionLLMManager.should_process_llm_request(event):
logger.debug(f"会话 {event.unified_msg_origin} 禁用了 LLM，跳过处理。")
return
umo = event.unified_msg_origin
provider = self.ctx.plugin_manager.context.get_using_provider(umo=umo)
if provider is None:
astrbot/core/star/session_llm_manager.py
+82
View File
@@ -0,0 +1,82 @@
"""
会话LLM管理器 - 负责管理每个会话的LLM启停状态
"""
from astrbot.core import sp, logger
from typing import Dict, Optional
from astrbot.core.platform.astr_message_event import AstrMessageEvent
class SessionLLMManager:
"""管理会话级别的LLM启停状态"""
@staticmethod
def is_llm_enabled_for_session(session_id: str) -> bool:
"""检查LLM是否在指定会话中启用
Args:
session_id: 会话ID (unified_msg_origin)
Returns:
bool: True表示启用，False表示禁用
"""
# 获取会话LLM配置
session_llm_config = sp.get("session_llm_config", {})
session_config = session_llm_config.get(session_id, {})
# 如果配置了该会话的LLM状态，返回该状态
llm_enabled = session_config.get("llm_enabled")
if llm_enabled is not None:
return llm_enabled
# 如果没有配置，默认为启用（兼容性考虑）
return True
@staticmethod
def set_llm_status_for_session(session_id: str, enabled: bool) -> None:
"""设置LLM在指定会话中的启停状态
Args:
session_id: 会话ID (unified_msg_origin)
enabled: True表示启用，False表示禁用
"""
# 获取当前配置
session_llm_config = sp.get("session_llm_config", {})
if session_id not in session_llm_config:
session_llm_config[session_id] = {}
# 设置LLM状态
session_llm_config[session_id]["llm_enabled"] = enabled
# 保存配置
sp.put("session_llm_config", session_llm_config)
logger.info(f"会话 {session_id} 的LLM状态已更新为: {'启用' if enabled else '禁用'}")
@staticmethod
def get_session_llm_config(session_id: str) -> Dict[str, bool]:
"""获取指定会话的LLM配置
Args:
session_id: 会话ID (unified_msg_origin)
Returns:
Dict[str, bool]: 包含llm_enabled的字典
"""
session_llm_config = sp.get("session_llm_config", {})
return session_llm_config.get(session_id, {
"llm_enabled": True # 默认启用
})
@staticmethod
def should_process_llm_request(event: AstrMessageEvent) -> bool:
"""检查是否应该处理LLM请求
Args:
event: 消息事件
Returns:
bool: True表示应该处理，False表示跳过
"""
session_id = event.unified_msg_origin
return SessionLLMManager.is_llm_enabled_for_session(session_id)
astrbot/dashboard/routes/session_management.py
+33 -4
View File
@@ -6,6 +6,7 @@ from astrbot.core.db import BaseDatabase
from astrbot.core.core_lifecycle import AstrBotCoreLifecycle
from astrbot.core.provider.entities import ProviderType
from astrbot.core.star.session_plugin_manager import SessionPluginManager
from astrbot.core.star.session_llm_manager import SessionLLMManager
class SessionManagementRoute(Route):
@@ -23,6 +24,7 @@ class SessionManagementRoute(Route):
"/session/get_session_info": ("POST", self.get_session_info),
"/session/plugins": ("GET", self.get_session_plugins),
"/session/update_plugin": ("POST", self.update_session_plugin),
"/session/update_llm": ("POST", self.update_session_llm),
}
self.db_helper = db_helper
self.core_lifecycle = core_lifecycle
@@ -47,8 +49,7 @@ class SessionManagementRoute(Route):
provider_manager = self.core_lifecycle.star_context.provider_manager
sessions = []
# 构建会话信息
# 构建会话信息
for session_id, conversation_id in session_conversations.items():
session_info = {
"session_id": session_id,
@@ -61,6 +62,7 @@ class SessionManagementRoute(Route):
"stt_provider_name": None,
"tts_provider_id": None,
"tts_provider_name": None,
"llm_enabled": SessionLLMManager.is_llm_enabled_for_session(session_id),
"platform": session_id.split(":")[0] if ":" in session_id else "unknown",
"message_type": session_id.split(":")[1] if session_id.count(":") >= 1 else "unknown",
"session_name": session_id.split(":")[2] if session_id.count(":") >= 2 else session_id,
@@ -260,8 +262,7 @@ class SessionManagementRoute(Route):
if not session_id:
return Response().error("缺少必要参数: session_id").__dict__
# 获取会话对话信息
# 获取会话对话信息
session_conversations = sp.get("session_conversation", {})
conversation_id = session_conversations.get(session_id)
@@ -279,6 +280,7 @@ class SessionManagementRoute(Route):
"stt_provider_name": None,
"tts_provider_id": None,
"tts_provider_name": None,
"llm_enabled": SessionLLMManager.is_llm_enabled_for_session(session_id),
"platform": session_id.split(":")[0] if ":" in session_id else "unknown",
"message_type": session_id.split(":")[1] if session_id.count(":") >= 1 else "unknown",
"session_name": session_id.split(":")[2] if session_id.count(":") >= 2 else session_id,
@@ -441,3 +443,30 @@ class SessionManagementRoute(Route):
error_msg = f"更新会话插件状态失败: {str(e)}\n{traceback.format_exc()}"
logger.error(error_msg)
return Response().error(f"更新会话插件状态失败: {str(e)}").__dict__
async def update_session_llm(self):
"""更新指定会话的LLM启停状态"""
try:
data = await request.get_json()
session_id = data.get("session_id")
enabled = data.get("enabled")
if not session_id:
return Response().error("缺少必要参数: session_id").__dict__
if enabled is None:
return Response().error("缺少必要参数: enabled").__dict__
# 使用 SessionLLMManager 更新LLM状态
SessionLLMManager.set_llm_status_for_session(session_id, enabled)
return Response().ok({
"message": f"LLM已{'启用' if enabled else '禁用'}",
"session_id": session_id,
"llm_enabled": enabled,
}).__dict__
except Exception as e:
error_msg = f"更新会话LLM状态失败: {str(e)}\n{traceback.format_exc()}"
logger.error(error_msg)
return Response().error(f"更新会话LLM状态失败: {str(e)}").__dict__
dashboard/src/views/SessionManagementPage.vue
+53 -5
View File
@@ -154,7 +154,25 @@
{{ selection.raw.label }}
</v-chip>
</template>
</v-select>
</v-select> </template>
<!-- LLM启停 -->
<template v-slot:item.llm_enabled="{ item }">
<v-switch
:model-value="item.llm_enabled"
@update:model-value="(value) => updateLLM(item, value)"
:loading="item.updating"
hide-details
density="compact"
color="primary"
inset
>
<template v-slot:label>
<span class="text-caption">
{{ item.llm_enabled ? '已启用' : '已禁用' }}
</span>
</template>
</v-switch>
</template>
<!-- 插件管理 -->
@@ -307,8 +325,7 @@
</v-chip>
</v-list-item-subtitle>
</v-list-item>
<v-list-item v-if="selectedSession.tts_provider_name">
<v-list-item v-if="selectedSession.tts_provider_name">
<v-list-item-title>TTS Provider</v-list-item-title>
<v-list-item-subtitle>
<v-chip size="small" color="warning">
@@ -316,6 +333,18 @@
</v-chip>
</v-list-item-subtitle>
</v-list-item>
<v-list-item>
<v-list-item-title>LLM状态</v-list-item-title>
<v-list-item-subtitle>
<v-chip
size="small"
:color="selectedSession.llm_enabled ? 'success' : 'error'"
>
{{ selectedSession.llm_enabled ? '已启用' : '已禁用' }}
</v-chip>
</v-list-item-subtitle>
</v-list-item>
</v-list>
</v-card-text>
</v-card>
@@ -424,14 +453,14 @@ export default {
snackbar: false,
snackbarText: '',
snackbarColor: 'success',
// 表格头部
// 表格头部
headers: [
{ title: '会话信息', key: 'session_info', sortable: false, width: '200px' },
{ title: '人格', key: 'persona', sortable: false, width: '180px' },
{ title: 'Chat Provider', key: 'chat_provider', sortable: false, width: '180px' },
{ title: 'STT Provider', key: 'stt_provider', sortable: false, width: '150px' },
{ title: 'TTS Provider', key: 'tts_provider', sortable: false, width: '150px' },
{ title: 'LLM启停', key: 'llm_enabled', sortable: false, width: '120px' },
{ title: '插件管理', key: 'plugins', sortable: false, width: '120px' },
{ title: '操作', key: 'actions', sortable: false, width: '80px' },
],
@@ -585,6 +614,25 @@ export default {
}
} catch (error) {
this.showError(error.response?.data?.message || 'Provider 更新失败');
} session.updating = false;
},
async updateLLM(session, enabled) {
session.updating = true;
try {
const response = await axios.post('/api/session/update_llm', {
session_id: session.session_id,
enabled: enabled
});
if (response.data.status === 'ok') {
session.llm_enabled = enabled;
this.showSuccess(`LLM ${enabled ? '已启用' : '已禁用'}`);
} else {
this.showError(response.data.message || 'LLM状态更新失败');
}
} catch (error) {
this.showError(error.response?.data?.message || 'LLM状态更新失败');
}
session.updating = false;
},