diff --git a/packages/client/src/api/hermes/group-chat.ts b/packages/client/src/api/hermes/group-chat.ts index 7b427c3a8..49f6dd485 100644 --- a/packages/client/src/api/hermes/group-chat.ts +++ b/packages/client/src/api/hermes/group-chat.ts @@ -11,6 +11,7 @@ export interface RoomInfo { maxHistoryTokens?: number tailMessageCount?: number totalTokens?: number + defaultAgentId?: string | null } export interface RoomAgent { @@ -226,3 +227,11 @@ export async function forceCompress(roomId: string): Promise<{ success: boolean; method: 'POST', }) } + +export async function setDefaultAgent(roomId: string, agentId: string): Promise<{ success: boolean }> { + return request(`/api/hermes/group-chat/rooms/${roomId}/default-agent`, { + method: 'PUT', + headers: { 'Content-Type': 'application/json' }, + body: JSON.stringify({ agentId }), + }) +} diff --git a/packages/client/src/components/hermes/group-chat/GroupChatPanel.vue b/packages/client/src/components/hermes/group-chat/GroupChatPanel.vue index 902a0ee6e..7ed0c1a72 100644 --- a/packages/client/src/components/hermes/group-chat/GroupChatPanel.vue +++ b/packages/client/src/components/hermes/group-chat/GroupChatPanel.vue @@ -26,6 +26,7 @@ const isCompressing = ref(false) const selectedProfile = ref(null) const agentName = ref('') const agentDescription = ref('') +const setAsDefaultAgent = ref(false) const cloneSourceRoomId = ref(null) const cloneRoomName = ref('') const cloneInviteCode = ref('') @@ -186,11 +187,13 @@ async function confirmAddAgent() { profile: selectedProfile.value, name: agentName.value.trim() || undefined, description: agentDescription.value.trim() || undefined, + setDefaultAgent: setAsDefaultAgent.value, }) showAddAgentModal.value = false selectedProfile.value = null agentName.value = '' agentDescription.value = '' + setAsDefaultAgent.value = false message.success(t('groupChat.agentAdded')) } catch (err: any) { if (err.message?.includes('already')) { @@ -252,6 +255,16 @@ async function handleRemoveAgent(agentId: string) { } } +async function handleSetDefaultAgent(agentId: string) { + if (!store.currentRoomId) return + try { + await store.setDefaultAgent(store.currentRoomId, agentId) + message.success(t('groupChat.defaultAgentSet')) + } catch { + message.error(t('common.saveFailed')) + } +} + async function handleInterruptAgent(agentName: string) { try { await store.interruptAgent(agentName) @@ -366,13 +379,16 @@ watch(() => store.sortedMessages.length, async () => {
{{ t('groupChat.agents') }} ({{ store.agents.length }})
-
+
- {{ agent.name }} + + {{ agent.name }} + {{ t('groupChat.default') }} + {{ agent.profile }}
-
@@ -519,6 +535,12 @@ watch(() => store.sortedMessages.length, async () => { :placeholder="t('groupChat.agentDescPlaceholder')" />
+
+ +
{{ t('common.cancel') }} @@ -1235,6 +1257,32 @@ export default defineComponent({ components: { CreateRoomForm } }) margin: 4px 0 0; } +.checkbox-label { + display: flex; + align-items: center; + gap: 8px; + font-size: 13px; + color: $text-secondary; + cursor: pointer; + + input[type="checkbox"] { + width: 16px; + height: 16px; + cursor: pointer; + } +} + +.default-badge { + display: inline-block; + margin-left: 6px; + padding: 1px 6px; + font-size: 10px; + font-weight: 600; + color: var(--accent-primary); + background: rgba(var(--accent-primary-rgb), 0.1); + border-radius: 4px; +} + // ─── Connection Dot ────────────────────────────────────── .connection-dot { diff --git a/packages/client/src/i18n/locales/en.ts b/packages/client/src/i18n/locales/en.ts index 770c07603..ed4ad0198 100644 --- a/packages/client/src/i18n/locales/en.ts +++ b/packages/client/src/i18n/locales/en.ts @@ -1090,6 +1090,9 @@ export default { compressionSaved: 'Compression config saved', compressNow: 'Compress Now', compressingInProgress: 'Compression in progress, please wait', + setAsDefaultAgent: 'Set as default agent', + defaultAgentSet: 'Default agent updated', + default: 'Default', }, // Usage diff --git a/packages/client/src/i18n/locales/zh.ts b/packages/client/src/i18n/locales/zh.ts index 0ace8fcac..382b4136e 100644 --- a/packages/client/src/i18n/locales/zh.ts +++ b/packages/client/src/i18n/locales/zh.ts @@ -1092,6 +1092,9 @@ export default { compressionSaved: '压缩配置已保存', compressNow: '立即压缩', compressingInProgress: '正在压缩中，请稍后', + setAsDefaultAgent: '设为默认智能体', + defaultAgentSet: '默认智能体已更新', + default: '默认', }, // 用量统计 diff --git a/packages/client/src/stores/hermes/group-chat.ts b/packages/client/src/stores/hermes/group-chat.ts index e43c9b7df..2c909e7a1 100644 --- a/packages/client/src/stores/hermes/group-chat.ts +++ b/packages/client/src/stores/hermes/group-chat.ts @@ -23,6 +23,7 @@ import { cloneRoom as cloneRoomApi, deleteRoom as deleteRoomApi, clearRoomContext, + setDefaultAgent as setDefaultAgentApi, } from '@/api/hermes/group-chat' async function uploadGroupFiles(attachments: Attachment[]): Promise<{ name: string; path: string }[]> { @@ -438,6 +439,13 @@ export const useGroupChatStore = defineStore('groupChat', () => { messages.value = res.messages agents.value = res.agents members.value = res.members || [] + + const roomIdx = rooms.value.findIndex(r => r.id === res.room.id) + if (roomIdx >= 0) { + rooms.value[roomIdx] = { ...rooms.value[roomIdx], ...res.room } + } else { + rooms.value.push(res.room) + } } catch (err: any) { error.value = err.message throw err @@ -457,6 +465,10 @@ export const useGroupChatStore = defineStore('groupChat', () => { if (!res?.error) { members.value = res.members || [] if (res.agents) agents.value = res.agents + if (res.defaultAgentId !== undefined) { + const roomIdx = rooms.value.findIndex(r => r.id === roomId) + if (roomIdx >= 0) rooms.value[roomIdx].defaultAgentId = res.defaultAgentId + } // Restore typing state from server if (res.typingUsers) { @@ -526,7 +538,7 @@ export const useGroupChatStore = defineStore('groupChat', () => { } } - async function createNewRoom(name: string, inviteCode: string, agentList?: { profile: string; name?: string; description?: string; invited?: boolean }[], compression?: { triggerTokens: number; maxHistoryTokens: number; tailMessageCount: number }) { + async function createNewRoom(name: string, inviteCode: string, agentList?: { profile: string; name?: string; description?: string; invited?: boolean; setDefaultAgent?: boolean }[], compression?: { triggerTokens: number; maxHistoryTokens: number; tailMessageCount: number }) { try { const res = await createRoom({ name, @@ -605,10 +617,16 @@ export const useGroupChatStore = defineStore('groupChat', () => { } catch { /* ignore */ } } - async function addAgentToRoom(roomId: string, data: { profile: string; name?: string; description?: string; invited?: boolean }) { + async function addAgentToRoom(roomId: string, data: { profile: string; name?: string; description?: string; invited?: boolean; setDefaultAgent?: boolean }) { try { const res = await addAgent(roomId, data) agents.value.push(res.agent) + if (data.setDefaultAgent) { + const idx = rooms.value.findIndex(r => r.id === roomId) + if (idx >= 0) { + rooms.value[idx] = { ...rooms.value[idx], defaultAgentId: res.agent.id } + } + } return res.agent } catch (err: any) { error.value = err.message @@ -627,6 +645,17 @@ export const useGroupChatStore = defineStore('groupChat', () => { } } + async function setDefaultAgent(roomId: string, agentId: string) { + try { + await setDefaultAgentApi(roomId, agentId) + const room = rooms.value.find(r => r.id === roomId) + if (room) room.defaultAgentId = agentId + } catch (err: any) { + error.value = err.message + throw err + } + } + // ─── Typing ──────────────────────────────────────────── let _typingTimer: ReturnType | null = null @@ -717,6 +746,7 @@ export const useGroupChatStore = defineStore('groupChat', () => { loadAgents, addAgentToRoom, removeAgentFromRoom, + setDefaultAgent, } }) diff --git a/packages/server/src/db/hermes/schemas.ts b/packages/server/src/db/hermes/schemas.ts index d39031818..fee9c1f98 100644 --- a/packages/server/src/db/hermes/schemas.ts +++ b/packages/server/src/db/hermes/schemas.ts @@ -119,6 +119,7 @@ export const GC_ROOMS_SCHEMA: Record = { tailMessageCount: 'INTEGER NOT NULL DEFAULT 10', totalTokens: 'INTEGER NOT NULL DEFAULT 0', sessionSeed: "TEXT NOT NULL DEFAULT '0'", + defaultAgentId: 'TEXT', } export const GC_MESSAGES_TABLE = 'gc_messages' diff --git a/packages/server/src/routes/hermes/group-chat.ts b/packages/server/src/routes/hermes/group-chat.ts index 5159c6741..5c8b079b6 100644 --- a/packages/server/src/routes/hermes/group-chat.ts +++ b/packages/server/src/routes/hermes/group-chat.ts @@ -79,7 +79,7 @@ groupChatRoutes.post('/api/hermes/group-chat/rooms', async (ctx) => { const { name, inviteCode, agents, compression } = ctx.request.body as { name?: string inviteCode?: string - agents?: { profile: string; name?: string; description?: string; invited?: boolean }[] + agents?: { profile: string; name?: string; description?: string; invited?: boolean; setDefaultAgent?: boolean }[] compression?: { triggerTokens?: number; maxHistoryTokens?: number; tailMessageCount?: number } } if (!name || !inviteCode) { @@ -98,24 +98,33 @@ groupChatRoutes.post('/api/hermes/group-chat/rooms', async (ctx) => { const storage = chatServer.getStorage() storage.saveRoom(roomId, name, inviteCode, compression) - const addedAgents = [] + let lastDefaultAgentId: string | null = null const agentResults = [] for (const a of agents || []) { + const agentId = generateId() try { const agent = await connectAndPersistRoomAgent(chatServer, roomId, { profile: a.profile, name: a.name || a.profile, description: a.description || '', invited: a.invited, - }) + }, agentId) addedAgents.push(agent) agentResults.push({ profile: a.profile, ok: true, agent }) + if (a.setDefaultAgent) { + lastDefaultAgentId = agentId + } } catch (err: any) { console.error(`[GroupChat] Failed to connect agent ${a.profile} to room ${roomId}: ${sanitizeAgentConnectReason(err.message)}`) agentResults.push({ ok: false, ...agentConnectFailureBody(a.profile, err) }) } } + if (lastDefaultAgentId) { + storage.setDefaultAgent(roomId, lastDefaultAgentId) + chatServer.agentClients.setAgentShouldAnswer(roomId, lastDefaultAgentId) + } + const room = storage.getRoom(roomId) ctx.body = { room, agents: addedAgents, agentResults } }) @@ -145,16 +154,18 @@ groupChatRoutes.post('/api/hermes/group-chat/rooms/:roomId/clone', async (ctx) = tailMessageCount: sourceRoom.tailMessageCount, }) - const addedAgents = [] + const agentIdMap = new Map() const agentResults = [] for (const sourceAgent of storage.getRoomAgents(sourceRoom.id)) { + const agentId = generateId() + agentIdMap.set(sourceAgent.id, agentId) try { const agent = await connectAndPersistRoomAgent(chatServer, roomId, { profile: sourceAgent.profile, name: sourceAgent.name, description: sourceAgent.description, invited: sourceAgent.invited, - }) + }, agentId) addedAgents.push(agent) agentResults.push({ profile: sourceAgent.profile, ok: true, agent }) } catch (err: any) { @@ -163,6 +174,12 @@ groupChatRoutes.post('/api/hermes/group-chat/rooms/:roomId/clone', async (ctx) = } } + if (sourceRoom.defaultAgentId && agentIdMap.has(sourceRoom.defaultAgentId)) { + const newAgentId = agentIdMap.get(sourceRoom.defaultAgentId)! + storage.setDefaultAgent(roomId, newAgentId) + chatServer.agentClients.setAgentShouldAnswer(roomId, newAgentId) + } + const room = storage.getRoom(roomId) ctx.body = { room, agents: addedAgents, agentResults } }) @@ -245,7 +262,7 @@ groupChatRoutes.post('/api/hermes/group-chat/rooms/:roomId/agents', async (ctx) return } - const { profile, name, description, invited } = ctx.request.body as { profile?: string; name?: string; description?: string; invited?: boolean } + const { profile, name, description, invited, setDefaultAgent } = ctx.request.body as { profile?: string; name?: string; description?: string; invited?: boolean; setDefaultAgent?: boolean } if (!profile) { ctx.status = 400 ctx.body = { error: 'profile is required' } @@ -265,13 +282,18 @@ groupChatRoutes.post('/api/hermes/group-chat/rooms/:roomId/agents', async (ctx) return } + const agentId = generateId() try { const agent = await connectAndPersistRoomAgent(chatServer, ctx.params.roomId, { profile, name: name || profile, description: description || '', invited, - }) + }, agentId) + if (setDefaultAgent) { + chatServer.getStorage().setDefaultAgent(ctx.params.roomId, agentId) + chatServer.agentClients.setAgentShouldAnswer(ctx.params.roomId, agentId) + } ctx.body = { agent } } catch (err: any) { console.error(`[GroupChat] Failed to connect agent ${profile} to room ${ctx.params.roomId}: ${sanitizeAgentConnectReason(err.message)}`) @@ -310,6 +332,11 @@ groupChatRoutes.delete('/api/hermes/group-chat/rooms/:roomId/agents/:agentId', a return } + const room = storage.getRoom(roomId) + if (room?.defaultAgentId === requestedAgentId) { + storage.setDefaultAgent(roomId, null) + } + storage.removeRoomMembersForAgent(roomId, agent) storage.removeRoomAgent(roomId, requestedAgentId) chatServer.agentClients.removeAgentFromRoom(roomId, agent.agentId) @@ -320,6 +347,40 @@ groupChatRoutes.delete('/api/hermes/group-chat/rooms/:roomId/agents/:agentId', a } }) +// Set default agent for room +groupChatRoutes.put('/api/hermes/group-chat/rooms/:roomId/default-agent', async (ctx) => { + if (!chatServer) { + ctx.status = 503 + ctx.body = { error: 'Group chat not initialized' } + return + } + + const { agentId } = ctx.request.body as { agentId?: string } + if (!agentId) { + ctx.status = 400 + ctx.body = { error: 'agentId is required' } + return + } + + const room = chatServer.getStorage().getRoom(ctx.params.roomId) + if (!room) { + ctx.status = 404 + ctx.body = { error: 'Room not found' } + return + } + + const agents = chatServer.getStorage().getRoomAgents(ctx.params.roomId) + if (!agents.find(a => a.id === agentId)) { + ctx.status = 400 + ctx.body = { error: 'Agent not in room' } + return + } + + chatServer.getStorage().setDefaultAgent(ctx.params.roomId, agentId) + chatServer.agentClients.setAgentShouldAnswer(ctx.params.roomId, agentId) + ctx.body = { success: true } +}) + // Delete room groupChatRoutes.delete('/api/hermes/group-chat/rooms/:roomId', async (ctx) => { if (!chatServer) { diff --git a/packages/server/src/services/hermes/group-chat/agent-clients.ts b/packages/server/src/services/hermes/group-chat/agent-clients.ts index c5d95dac7..f9ecb89fe 100644 --- a/packages/server/src/services/hermes/group-chat/agent-clients.ts +++ b/packages/server/src/services/hermes/group-chat/agent-clients.ts @@ -22,6 +22,7 @@ interface AgentConfig { name: string description: string invited: number + dbId?: string } interface MessageData { @@ -68,6 +69,7 @@ export interface AgentEventHandler { class AgentClient { readonly agentId: string + readonly dbId: string readonly profile: string readonly name: string readonly description: string @@ -82,6 +84,7 @@ class AgentClient { constructor(config: AgentConfig, handlers: AgentEventHandler = {}) { this.agentId = config.agentId || Date.now().toString(36) + Math.random().toString(36).slice(2, 8) + this.dbId = config.dbId || this.agentId this.profile = config.profile this.name = config.name this.description = config.description @@ -796,6 +799,13 @@ export class AgentClients { private _processingRooms = new Set() private _mentionQueue = new Map>() + // agent_should_answer tracking: roomId -> agentId + private _agentShouldAnswer = new Map() + // Timeout handles for auto-reset: roomId -> timer + private _answerTimeouts = new Map>() + // Default timeout: 5 minutes + private _answerTimeoutMs = 5 * 60 * 1000 + /** * Create an agent client and connect it to the server. * The agent will NOT auto-join any room — call addAgentToRoom separately. @@ -907,23 +917,6 @@ export class AgentClients { await agent.interrupt(roomId) } - /** - * Disconnect all agents in a room. - */ - disconnectRoom(roomId: string): void { - const room = this.rooms.get(roomId) - if (!room) return - - room.forEach((client) => client.disconnect()) - this.rooms.delete(roomId) - logger.info(`[AgentClients] All agents disconnected from room: ${roomId}`) - - // Invalidate context engine cache for this room - if (this._contextEngine) { - try { this._contextEngine.invalidateRoom(roomId) } catch { /* ignore */ } - } - } - resetRoomContext(roomId: string): void { this._mentionQueue.delete(roomId) for (const key of Array.from(this._mentionQueue.keys())) { @@ -968,12 +961,127 @@ export class AgentClients { }) } + /** + * Set the agent that should answer for a room. + * Clears any existing timeout and sets a new one to reset to default. + */ + setAgentShouldAnswer(roomId: string, agentId: string): void { + this._agentShouldAnswer.set(roomId, agentId) + this._resetAnswerTimeout(roomId) + } + + /** + * Get the agent that should answer for a room. + * Returns the tracked agent, or falls back to the room's default agent. + */ + getAgentShouldAnswer(roomId: string): AgentClient | null { + const agentId = this._agentShouldAnswer.get(roomId) + if (agentId) { + const agent = this._findAgentById(roomId, agentId) + if (agent) return agent + } + + const defaultAgentId = this._storage?.getRoom?.(roomId)?.defaultAgentId + if (defaultAgentId) { + const agent = this._findAgentById(roomId, defaultAgentId) + if (agent) { + this._agentShouldAnswer.set(roomId, defaultAgentId) + this._resetAnswerTimeout(roomId) + return agent + } + } + + return null + } + + /** + * Initialize agent_should_answer from the room's default agent (call after agents are restored). + */ + initAgentShouldAnswer(roomId: string): void { + const defaultAgentId = this._storage?.getRoom?.(roomId)?.defaultAgentId + if (defaultAgentId && !this._agentShouldAnswer.has(roomId)) { + this._agentShouldAnswer.set(roomId, defaultAgentId) + this._resetAnswerTimeout(roomId) + } + } + + /** + * Update a room's default agent and reset agent_should_answer to it. + */ + setDefaultAgent(roomId: string, agentId: string | null): void { + this._storage?.setDefaultAgent?.(roomId, agentId) + if (agentId) { + this._agentShouldAnswer.set(roomId, agentId) + this._resetAnswerTimeout(roomId) + } else { + this._agentShouldAnswer.delete(roomId) + this._clearAnswerTimeout(roomId) + } + } + + private _findAgentById(roomId: string, agentId: string): AgentClient | null { + const room = this.rooms.get(roomId) + if (!room) return null + for (const client of room.values()) { + if (client.dbId === agentId || client.agentId === agentId) return client + } + return null + } + + private _resetAnswerTimeout(roomId: string): void { + this._clearAnswerTimeout(roomId) + const timer = setTimeout(() => { + this._resetToDefaultAgent(roomId) + }, this._answerTimeoutMs) + this._answerTimeouts.set(roomId, timer) + } + + private _clearAnswerTimeout(roomId: string): void { + const existing = this._answerTimeouts.get(roomId) + if (existing) { + clearTimeout(existing) + this._answerTimeouts.delete(roomId) + } + } + + private _resetToDefaultAgent(roomId: string): void { + const defaultAgentId = this._storage?.getRoom?.(roomId)?.defaultAgentId + if (defaultAgentId) { + this._agentShouldAnswer.set(roomId, defaultAgentId) + logger.debug(`[AgentClients] agent_should_answer for room ${roomId} reset to default: ${defaultAgentId}`) + } else { + this._agentShouldAnswer.delete(roomId) + } + this._answerTimeouts.delete(roomId) + } + + /** + * Clean up timeouts when a room is disconnected or cleared. + */ + disconnectRoom(roomId: string): void { + this._clearAnswerTimeout(roomId) + this._agentShouldAnswer.delete(roomId) + + const room = this.rooms.get(roomId) + if (!room) return + + room.forEach((client) => client.disconnect()) + this.rooms.delete(roomId) + logger.info(`[AgentClients] All agents disconnected from room: ${roomId}`) + + // Invalidate context engine cache for this room + if (this._contextEngine) { + try { this._contextEngine.invalidateRoom(roomId) } catch { /* ignore */ } + } + } + /** * Server-side: parse @mentions and forward to matching agents directly. * If the room is already processing (compressing/replying), queue the mention. + * Returns the list of mentioned agent names. */ - async processMentions(roomId: string, msg: MentionMessage): Promise { + async processMentions(roomId: string, msg: MentionMessage): Promise { const agents = this.getAgents(roomId) const mentioned = resolveMentionTargets(agents, msg.content, msg.senderId) if (mentioned.length === 0) return @@ -985,6 +1093,39 @@ export class AgentClients { logger.error(`[AgentClients] error processing mention for ${agent.name}: ${err.message}`) }) } + + return mentioned.map(a => a.name) + } + + /** + * Route a message to the agent that should answer (no @mention case). + * Updates agent_should_answer timeout. + */ + async routeToAgentShouldAnswer(roomId: string, msg: MentionMessage): Promise { + const agent = this.getAgentShouldAnswer(roomId) + if (!agent) { + logger.debug(`[AgentClients] no agent_should_answer for room ${roomId}, skipping`) + return + } + + logger.debug(`[AgentClients] routing to agent_should_answer ${agent.name} in room ${roomId}`) + this._resetAnswerTimeout(roomId) + + this._processAgentMention(roomId, agent, msg).catch((err) => { + logger.error(`[AgentClients] error processing agent_should_answer for ${agent.name}: ${err.message}`) + }) + } + + /** + * Update agent_should_answer when user @mentions an agent. + */ + updateAgentShouldAnswerOnMention(roomId: string, mentionedAgentName: string): void { + const agent = this.getAgents(roomId).find(a => a.name === mentionedAgentName) + if (agent) { + this._agentShouldAnswer.set(roomId, agent.agentId) + this._resetAnswerTimeout(roomId) + logger.debug(`[AgentClients] agent_should_answer for room ${roomId} set to ${agent.name}`) + } } /** diff --git a/packages/server/src/services/hermes/group-chat/index.ts b/packages/server/src/services/hermes/group-chat/index.ts index 8265f6957..ad756e6a6 100644 --- a/packages/server/src/services/hermes/group-chat/index.ts +++ b/packages/server/src/services/hermes/group-chat/index.ts @@ -278,16 +278,16 @@ class ChatStorage { // ─── Rooms ──────────────────────────────────────────────── - getRoom(roomId: string): { id: string; name: string; inviteCode: string | null; triggerTokens: number; maxHistoryTokens: number; tailMessageCount: number; totalTokens: number; sessionSeed: string } | undefined { - return this.db()?.prepare('SELECT id, name, inviteCode, triggerTokens, maxHistoryTokens, tailMessageCount, totalTokens, sessionSeed FROM gc_rooms WHERE id = ?').get(roomId) as any + getRoom(roomId: string): { id: string; name: string; inviteCode: string | null; triggerTokens: number; maxHistoryTokens: number; tailMessageCount: number; totalTokens: number; sessionSeed: string; defaultAgentId: string | null } | undefined { + return this.db()?.prepare('SELECT id, name, inviteCode, triggerTokens, maxHistoryTokens, tailMessageCount, totalTokens, sessionSeed, defaultAgentId FROM gc_rooms WHERE id = ?').get(roomId) as any } - getRoomByInviteCode(code: string): { id: string; name: string; inviteCode: string | null; triggerTokens: number; maxHistoryTokens: number; tailMessageCount: number; totalTokens: number; sessionSeed: string } | undefined { - return this.db()?.prepare('SELECT id, name, inviteCode, triggerTokens, maxHistoryTokens, tailMessageCount, totalTokens, sessionSeed FROM gc_rooms WHERE inviteCode = ?').get(code) as any + getRoomByInviteCode(code: string): { id: string; name: string; inviteCode: string | null; triggerTokens: number; maxHistoryTokens: number; tailMessageCount: number; totalTokens: number; sessionSeed: string; defaultAgentId: string | null } | undefined { + return this.db()?.prepare('SELECT id, name, inviteCode, triggerTokens, maxHistoryTokens, tailMessageCount, totalTokens, sessionSeed, defaultAgentId FROM gc_rooms WHERE inviteCode = ?').get(code) as any } - getAllRooms(): { id: string; name: string; inviteCode: string | null; triggerTokens: number; maxHistoryTokens: number; tailMessageCount: number; totalTokens: number; sessionSeed: string }[] { - return (this.db()?.prepare('SELECT id, name, inviteCode, triggerTokens, maxHistoryTokens, tailMessageCount, totalTokens, sessionSeed FROM gc_rooms ORDER BY id').all() || []) as any[] + getAllRooms(): { id: string; name: string; inviteCode: string | null; triggerTokens: number; maxHistoryTokens: number; tailMessageCount: number; totalTokens: number; sessionSeed: string; defaultAgentId: string | null }[] { + return (this.db()?.prepare('SELECT id, name, inviteCode, triggerTokens, maxHistoryTokens, tailMessageCount, totalTokens, sessionSeed, defaultAgentId FROM gc_rooms ORDER BY id').all() || []) as any[] } saveRoom(id: string, name: string, inviteCode?: string, config?: { triggerTokens?: number; maxHistoryTokens?: number; tailMessageCount?: number }): void { @@ -315,6 +315,10 @@ class ChatStorage { this.db()?.prepare('UPDATE gc_rooms SET totalTokens = ? WHERE id = ?').run(tokens, roomId) } + setDefaultAgent(roomId: string, agentId: string | null): void { + this.db()?.prepare('UPDATE gc_rooms SET defaultAgentId = ? WHERE id = ?').run(agentId, roomId) + } + rotateRoomSessionSeed(roomId: string): string { const seed = `${Date.now().toString(36)}${Math.random().toString(36).slice(2, 8)}` this.db()?.prepare('UPDATE gc_rooms SET sessionSeed = ? WHERE id = ?').run(seed, roomId) @@ -763,6 +767,7 @@ export class GroupChatServer { name: agent.name, description: agent.description, invited: agent.invited, + dbId: agent.id, }) await this.agentClients.addAgentToRoom(room.id, client) total++ @@ -770,6 +775,7 @@ export class GroupChatServer { logger.error(`[GroupChat] Failed to restore agent ${agent.name} in room ${room.id}: ${err.message}`) } } + this.agentClients.initAgentShouldAnswer(room.id) } if (total > 0) { @@ -875,6 +881,7 @@ export class GroupChatServer { // Load history from SQLite const messages = this.storage.getMessages(roomId) const agents = this.storage.getRoomAgents(roomId) + const roomInfo = this.storage.getRoom(roomId) ack?.({ roomId, @@ -885,6 +892,7 @@ export class GroupChatServer { rooms: this.getRoomIds(), typingUsers: this.getTypingUsers(roomId), contextStatuses: this.getContextStatuses(roomId), + defaultAgentId: roomInfo?.defaultAgentId || null, }) logger.debug(`[GroupChat] ${userName} (user=${userId}) joined room: ${roomId}`) @@ -933,17 +941,44 @@ export class GroupChatServer { const shouldRouteMentions = savedMsg.role === 'user' if (shouldRouteMentions) { - // Server-side @mention routing — parse user mentions and invoke agents directly. - this.agentClients.processMentions(roomId, { - content: contentToText(savedMsg.content), - input: Array.isArray(data.content) ? data.content : undefined, - senderName: savedMsg.senderName, - senderId: savedMsg.senderId, - timestamp: savedMsg.timestamp, - mentionDepth, - }).catch((err) => { - logger.error(`[GroupChat] processMentions error: ${err.message}`) + const contentText = contentToText(savedMsg.content) + const agents = this.storage.getRoomAgents(roomId) + const senderNameLower = savedMsg.senderName.toLowerCase() + + const mentionedAgents = agents.filter(a => { + if (a.name.toLowerCase() === senderNameLower) return false + const escaped = a.name.replace(/[.*+?^${}()|[\]\\]/g, '\\$&') + const pattern = new RegExp(`@${escaped}(?=$|\\s|[.,!?;:，。！？；：])`, 'i') + return pattern.test(contentText) }) + + if (mentionedAgents.length > 0 && savedMsg.role === 'user') { + this.agentClients.updateAgentShouldAnswerOnMention(roomId, mentionedAgents[0].name) + } + + if (mentionedAgents.length === 0 && savedMsg.role === 'user') { + this.agentClients.routeToAgentShouldAnswer(roomId, { + content: contentText, + input: Array.isArray(data.content) ? data.content : undefined, + senderName: savedMsg.senderName, + senderId: savedMsg.senderId, + timestamp: savedMsg.timestamp, + mentionDepth, + }).catch((err) => { + logger.error(`[GroupChat] routeToAgentShouldAnswer error: ${err.message}`) + }) + } else if (mentionedAgents.length > 0) { + this.agentClients.processMentions(roomId, { + content: contentText, + input: Array.isArray(data.content) ? data.content : undefined, + senderName: savedMsg.senderName, + senderId: savedMsg.senderId, + timestamp: savedMsg.timestamp, + mentionDepth, + }).catch((err) => { + logger.error(`[GroupChat] processMentions error: ${err.message}`) + }) + } } }