github/ChatLab
1
0
Fork 0
mirror of https://github.com/hellodigua/ChatLab.git synced 2026-04-29 08:12:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: 优化LLM配置逻辑

Browse Source
This commit is contained in:
digua
2025-12-06 01:11:30 +08:00
parent 77d0734ba3
commit 14c93bbed3
18 changed files with 222 additions and 139 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
electron/main/ai/tools/registry.ts
View File

@@ -53,9 +53,9 @@ async function searchMessagesExecutor(
params: { keywords: string[]; sender_id?: number; limit?: number; year?: number; month?: number },
context: ToolContext
): Promise<unknown> {
const { sessionId, timeFilter: contextTimeFilter } = context
// 默认 200，上限 5000
const limit = Math.min(params.limit || 200, 5000)
const { sessionId, timeFilter: contextTimeFilter, maxMessagesLimit } = context
// 优先使用 LLM 指定的 limit，其次使用用户配置，最后使用默认值 200，上限 5000
const limit = Math.min(params.limit || maxMessagesLimit || 200, 5000)
// 构建时间过滤器：优先使用 LLM 指定的年/月，否则使用 context 中的
let effectiveTimeFilter = contextTimeFilter
@@ -143,8 +143,9 @@ async function getRecentMessagesExecutor(
params: { limit?: number; year?: number; month?: number },
context: ToolContext
): Promise<unknown> {
const { sessionId, timeFilter: contextTimeFilter } = context
const limit = params.limit || 100
const { sessionId, timeFilter: contextTimeFilter, maxMessagesLimit } = context
// 优先使用 LLM 指定的 limit，其次使用用户配置，最后使用默认值 100
const limit = params.limit || maxMessagesLimit || 100
// 构建时间过滤器：优先使用 LLM 指定的年/月
let effectiveTimeFilter = contextTimeFilter
@@ -493,8 +494,9 @@ async function getConversationBetweenExecutor(
params: { member_id_1: number; member_id_2: number; limit?: number; year?: number; month?: number },
context: ToolContext
): Promise<unknown> {
const { sessionId, timeFilter: contextTimeFilter } = context
const limit = params.limit || 100
const { sessionId, timeFilter: contextTimeFilter, maxMessagesLimit } = context
// 优先使用 LLM 指定的 limit，其次使用用户配置，最后使用默认值 100
const limit = params.limit || maxMessagesLimit || 100
// 构建时间过滤器
let effectiveTimeFilter = contextTimeFilter

2
electron/main/ai/tools/types.ts
View File

@@ -17,6 +17,8 @@ export interface ToolContext {
startTs: number
endTs: number
}
/** 用户配置的消息条数限制（工具获取消息时使用） */
maxMessagesLimit?: number
}
/**