mirror of
https://github.com/hellodigua/ChatLab.git
synced 2026-05-02 10:51:18 +08:00
f7c427df50
- Delete monolithic registry.ts (−1185 lines) - Add tools/definitions/ with 12 individual tool files + index.ts, each using AgentTool interface and TypeBox schemas - Add tools/utils/ with shared helpers (format.ts, schemas.ts, time-params.ts) - Rewrite tools/index.ts to provide getAllTools() factory - Clean up tools/types.ts, keep only ToolContext and OwnerInfo - Use i18n keys for tool descriptions, preserve Chinese as comments
68 lines
2.8 KiB
TypeScript
68 lines
2.8 KiB
TypeScript
import { Type } from '@mariozechner/pi-ai'
|
||
import type { AgentTool } from '@mariozechner/pi-agent-core'
|
||
import type { ToolContext } from '../types'
|
||
import { timeParamPropertiesNoHour } from '../utils/schemas'
|
||
import * as workerManager from '../../../worker/workerManager'
|
||
import { parseExtendedTimeParams } from '../utils/time-params'
|
||
import { formatTimeRange, formatMessageCompact, isChineseLocale } from '../utils/format'
|
||
|
||
const schema = Type.Object({
|
||
keywords: Type.Optional(Type.Array(Type.String(), { description: 'ai.tools.search_sessions.params.keywords' })),
|
||
limit: Type.Optional(Type.Number({ description: 'ai.tools.search_sessions.params.limit' })),
|
||
...timeParamPropertiesNoHour,
|
||
})
|
||
|
||
/** 搜索聊天会话（对话段落）。会话是根据消息时间间隔自动切分的对话单元。适用于查找特定话题的讨论、了解某个时间段内发生了几次对话等场景。返回匹配的会话列表及每个会话的前5条消息预览。 */
|
||
export function createTool(context: ToolContext): AgentTool<typeof schema> {
|
||
return {
|
||
name: 'search_sessions',
|
||
label: 'search_sessions',
|
||
description: 'ai.tools.search_sessions.desc',
|
||
parameters: schema,
|
||
execute: async (_toolCallId, params) => {
|
||
const { sessionId, timeFilter: contextTimeFilter, locale } = context
|
||
const limit = params.limit || 20
|
||
const effectiveTimeFilter = parseExtendedTimeParams(params, contextTimeFilter)
|
||
|
||
const sessions = await workerManager.searchSessions(sessionId, params.keywords, effectiveTimeFilter, limit, 5)
|
||
|
||
if (sessions.length === 0) {
|
||
const data = {
|
||
total: 0,
|
||
message: isChineseLocale(locale) ? '未找到匹配的会话' : 'No matching sessions found',
|
||
}
|
||
return {
|
||
content: [{ type: 'text', text: JSON.stringify(data) }],
|
||
details: data,
|
||
}
|
||
}
|
||
|
||
const localeStr = isChineseLocale(locale) ? 'zh-CN' : 'en-US'
|
||
const msgSuffix = isChineseLocale(locale) ? '条消息' : ' messages'
|
||
const completeLabel = isChineseLocale(locale) ? '完整会话' : 'complete'
|
||
|
||
const data = {
|
||
total: sessions.length,
|
||
timeRange: formatTimeRange(effectiveTimeFilter, locale),
|
||
sessions: sessions.map((s) => {
|
||
const startTime = new Date(s.startTs * 1000).toLocaleString(localeStr)
|
||
const endTime = new Date(s.endTs * 1000).toLocaleString(localeStr)
|
||
const completeTag = s.isComplete ? ` [${completeLabel}]` : ''
|
||
|
||
return {
|
||
sessionId: s.id,
|
||
time: `${startTime} ~ ${endTime}`,
|
||
messageCount: `${s.messageCount}${msgSuffix}${completeTag}`,
|
||
preview: s.previewMessages.map((m) => formatMessageCompact(m, locale)),
|
||
}
|
||
}),
|
||
}
|
||
|
||
return {
|
||
content: [{ type: 'text', text: JSON.stringify(data) }],
|
||
details: data,
|
||
}
|
||
},
|
||
}
|
||
}
|