mirror of
https://github.com/hicccc77/WeFlow.git
synced 2026-04-11 23:15:51 +00:00
feat: ai功能的初次提交
This commit is contained in:
cc
444
electron/services/aiAssistantService.ts
Normal file
444
electron/services/aiAssistantService.ts
Normal file
@@ -0,0 +1,444 @@
|
||||
import { randomUUID } from 'crypto'
|
||||
import { existsSync } from 'fs'
|
||||
import { mkdir, readdir, readFile, rm, writeFile } from 'fs/promises'
|
||||
import { join } from 'path'
|
||||
import { ConfigService } from './config'
|
||||
|
||||
export type AssistantChatType = 'group' | 'private'
|
||||
export type AssistantToolCategory = 'core' | 'analysis'
|
||||
|
||||
export interface AssistantSummary {
|
||||
id: string
|
||||
name: string
|
||||
systemPrompt: string
|
||||
presetQuestions: string[]
|
||||
allowedBuiltinTools?: string[]
|
||||
builtinId?: string
|
||||
applicableChatTypes?: AssistantChatType[]
|
||||
supportedLocales?: string[]
|
||||
}
|
||||
|
||||
export interface AssistantConfigFull extends AssistantSummary {}
|
||||
|
||||
export interface BuiltinAssistantInfo {
|
||||
id: string
|
||||
name: string
|
||||
systemPrompt: string
|
||||
applicableChatTypes?: AssistantChatType[]
|
||||
supportedLocales?: string[]
|
||||
imported: boolean
|
||||
}
|
||||
|
||||
const GENERAL_CN_MD = `---
|
||||
id: general_cn
|
||||
name: 通用分析助手
|
||||
supportedLocales:
|
||||
- zh
|
||||
presetQuestions:
|
||||
- 最近都在聊什么？
|
||||
- 谁是最活跃的人？
|
||||
- 帮我总结一下最近一周的重要聊天
|
||||
- 帮我找一下关于“旅游”的讨论
|
||||
allowedBuiltinTools:
|
||||
- ai_query_time_window_activity
|
||||
- ai_query_session_candidates
|
||||
- ai_query_session_glimpse
|
||||
- ai_query_timeline
|
||||
- ai_fetch_message_briefs
|
||||
- ai_list_voice_messages
|
||||
- ai_transcribe_voice_messages
|
||||
- ai_query_topic_stats
|
||||
- ai_query_source_refs
|
||||
- ai_query_top_contacts
|
||||
---
|
||||
|
||||
你是 WeFlow 的全局聊天分析助手。请使用工具获取证据，给出简洁、准确、可执行的结论。
|
||||
|
||||
输出要求：
|
||||
1. 先结论，再证据。
|
||||
2. 若证据不足，明确说明不足并建议下一步。
|
||||
3. 涉及语音内容时，必须先列语音 ID，再按 ID 转写。
|
||||
4. 默认中文输出，除非用户明确指定其他语言。`
|
||||
|
||||
const GENERAL_EN_MD = `---
|
||||
id: general_en
|
||||
name: General Analysis Assistant
|
||||
supportedLocales:
|
||||
- en
|
||||
presetQuestions:
|
||||
- What have people been discussing recently?
|
||||
- Who are the most active contacts?
|
||||
- Summarize my key chat topics this week
|
||||
allowedBuiltinTools:
|
||||
- ai_query_time_window_activity
|
||||
- ai_query_session_candidates
|
||||
- ai_query_session_glimpse
|
||||
- ai_query_timeline
|
||||
- ai_fetch_message_briefs
|
||||
- ai_list_voice_messages
|
||||
- ai_transcribe_voice_messages
|
||||
- ai_query_topic_stats
|
||||
- ai_query_source_refs
|
||||
- ai_query_top_contacts
|
||||
---
|
||||
|
||||
You are WeFlow's global chat analysis assistant.
|
||||
Always ground your answers in tool evidence, stay concise, and clearly call out uncertainty when data is insufficient.`
|
||||
|
||||
const GENERAL_JA_MD = `---
|
||||
id: general_ja
|
||||
name: 汎用分析アシスタント
|
||||
supportedLocales:
|
||||
- ja
|
||||
presetQuestions:
|
||||
- 最近どんな話題が多い？
|
||||
- 一番アクティブな相手は誰？
|
||||
- 今週の重要な会話を要約して
|
||||
allowedBuiltinTools:
|
||||
- ai_query_time_window_activity
|
||||
- ai_query_session_candidates
|
||||
- ai_query_session_glimpse
|
||||
- ai_query_timeline
|
||||
- ai_fetch_message_briefs
|
||||
- ai_list_voice_messages
|
||||
- ai_transcribe_voice_messages
|
||||
- ai_query_topic_stats
|
||||
- ai_query_source_refs
|
||||
- ai_query_top_contacts
|
||||
---
|
||||
|
||||
あなたは WeFlow のグローバルチャット分析アシスタントです。
|
||||
ツールから得た根拠に基づき、簡潔かつ正確に回答してください。`
|
||||
|
||||
const BUILTIN_ASSISTANTS = [
|
||||
{ id: 'general_cn', raw: GENERAL_CN_MD },
|
||||
{ id: 'general_en', raw: GENERAL_EN_MD },
|
||||
{ id: 'general_ja', raw: GENERAL_JA_MD }
|
||||
] as const
|
||||
|
||||
function normalizeText(value: unknown, fallback = ''): string {
|
||||
const text = String(value ?? '').trim()
|
||||
return text || fallback
|
||||
}
|
||||
|
||||
function parseInlineList(text: string): string[] {
|
||||
const raw = normalizeText(text)
|
||||
if (!raw) return []
|
||||
return raw
|
||||
.split(',')
|
||||
.map((item) => item.trim())
|
||||
.filter(Boolean)
|
||||
}
|
||||
|
||||
function splitFrontmatter(raw: string): { frontmatter: string; body: string } {
|
||||
const normalized = String(raw || '')
|
||||
if (!normalized.startsWith('---')) {
|
||||
return { frontmatter: '', body: normalized.trim() }
|
||||
}
|
||||
const end = normalized.indexOf('\n---', 3)
|
||||
if (end < 0) return { frontmatter: '', body: normalized.trim() }
|
||||
return {
|
||||
frontmatter: normalized.slice(3, end).trim(),
|
||||
body: normalized.slice(end + 4).trim()
|
||||
}
|
||||
}
|
||||
|
||||
function parseAssistantMarkdown(raw: string): AssistantConfigFull {
|
||||
const { frontmatter, body } = splitFrontmatter(raw)
|
||||
const lines = frontmatter ? frontmatter.split('\n') : []
|
||||
const data: Record<string, unknown> = {}
|
||||
let currentArrayKey = ''
|
||||
for (const line of lines) {
|
||||
const trimmed = line.trim()
|
||||
if (!trimmed) continue
|
||||
const kv = trimmed.match(/^([A-Za-z0-9_]+)\s*:\s*(.*)$/)
|
||||
if (kv) {
|
||||
const key = kv[1]
|
||||
const value = kv[2]
|
||||
if (!value) {
|
||||
data[key] = []
|
||||
currentArrayKey = key
|
||||
} else {
|
||||
data[key] = value
|
||||
currentArrayKey = ''
|
||||
}
|
||||
continue
|
||||
}
|
||||
const arr = trimmed.match(/^- (.+)$/)
|
||||
if (arr && currentArrayKey) {
|
||||
const next = Array.isArray(data[currentArrayKey]) ? data[currentArrayKey] as string[] : []
|
||||
next.push(arr[1].trim())
|
||||
data[currentArrayKey] = next
|
||||
}
|
||||
}
|
||||
|
||||
const id = normalizeText(data.id)
|
||||
const name = normalizeText(data.name, id || 'assistant')
|
||||
const applicableChatTypes = Array.isArray(data.applicableChatTypes)
|
||||
? (data.applicableChatTypes as string[]).filter((item): item is AssistantChatType => item === 'group' || item === 'private')
|
||||
: parseInlineList(String(data.applicableChatTypes || '')).filter((item): item is AssistantChatType => item === 'group' || item === 'private')
|
||||
const supportedLocales = Array.isArray(data.supportedLocales)
|
||||
? (data.supportedLocales as string[]).map((item) => item.trim()).filter(Boolean)
|
||||
: parseInlineList(String(data.supportedLocales || ''))
|
||||
const presetQuestions = Array.isArray(data.presetQuestions)
|
||||
? (data.presetQuestions as string[]).map((item) => item.trim()).filter(Boolean)
|
||||
: parseInlineList(String(data.presetQuestions || ''))
|
||||
const allowedBuiltinTools = Array.isArray(data.allowedBuiltinTools)
|
||||
? (data.allowedBuiltinTools as string[]).map((item) => item.trim()).filter(Boolean)
|
||||
: parseInlineList(String(data.allowedBuiltinTools || ''))
|
||||
const builtinId = normalizeText(data.builtinId)
|
||||
|
||||
return {
|
||||
id,
|
||||
name,
|
||||
systemPrompt: body,
|
||||
presetQuestions,
|
||||
allowedBuiltinTools,
|
||||
builtinId: builtinId || undefined,
|
||||
applicableChatTypes,
|
||||
supportedLocales
|
||||
}
|
||||
}
|
||||
|
||||
function toMarkdown(config: AssistantConfigFull): string {
|
||||
const lines = [
|
||||
'---',
|
||||
`id: ${config.id}`,
|
||||
`name: ${config.name}`
|
||||
]
|
||||
if (config.builtinId) lines.push(`builtinId: ${config.builtinId}`)
|
||||
if (config.supportedLocales && config.supportedLocales.length > 0) {
|
||||
lines.push('supportedLocales:')
|
||||
config.supportedLocales.forEach((item) => lines.push(` - ${item}`))
|
||||
}
|
||||
if (config.applicableChatTypes && config.applicableChatTypes.length > 0) {
|
||||
lines.push('applicableChatTypes:')
|
||||
config.applicableChatTypes.forEach((item) => lines.push(` - ${item}`))
|
||||
}
|
||||
if (config.presetQuestions && config.presetQuestions.length > 0) {
|
||||
lines.push('presetQuestions:')
|
||||
config.presetQuestions.forEach((item) => lines.push(` - ${item}`))
|
||||
}
|
||||
if (config.allowedBuiltinTools && config.allowedBuiltinTools.length > 0) {
|
||||
lines.push('allowedBuiltinTools:')
|
||||
config.allowedBuiltinTools.forEach((item) => lines.push(` - ${item}`))
|
||||
}
|
||||
lines.push('---')
|
||||
lines.push('')
|
||||
lines.push(config.systemPrompt || '')
|
||||
return lines.join('\n')
|
||||
}
|
||||
|
||||
function defaultBuiltinToolCatalog(): Array<{ name: string; category: AssistantToolCategory }> {
|
||||
return [
|
||||
{ name: 'ai_query_time_window_activity', category: 'core' },
|
||||
{ name: 'ai_query_session_candidates', category: 'core' },
|
||||
{ name: 'ai_query_session_glimpse', category: 'core' },
|
||||
{ name: 'ai_query_timeline', category: 'core' },
|
||||
{ name: 'ai_fetch_message_briefs', category: 'core' },
|
||||
{ name: 'ai_list_voice_messages', category: 'core' },
|
||||
{ name: 'ai_transcribe_voice_messages', category: 'core' },
|
||||
{ name: 'ai_query_topic_stats', category: 'analysis' },
|
||||
{ name: 'ai_query_source_refs', category: 'analysis' },
|
||||
{ name: 'ai_query_top_contacts', category: 'analysis' },
|
||||
{ name: 'activate_skill', category: 'analysis' }
|
||||
]
|
||||
}
|
||||
|
||||
class AiAssistantService {
|
||||
private readonly config = ConfigService.getInstance()
|
||||
private initialized = false
|
||||
private readonly cache = new Map<string, AssistantConfigFull>()
|
||||
|
||||
private getRootDirCandidates(): string[] {
|
||||
const dbPath = normalizeText(this.config.get('dbPath'))
|
||||
const wxid = normalizeText(this.config.get('myWxid'))
|
||||
const roots: string[] = []
|
||||
if (dbPath && wxid) {
|
||||
roots.push(join(dbPath, wxid, 'db_storage', 'wf_ai_v2'))
|
||||
roots.push(join(dbPath, wxid, 'db_storage', 'wf_ai'))
|
||||
}
|
||||
roots.push(join(process.cwd(), 'data', 'wf_ai_v2'))
|
||||
return roots
|
||||
}
|
||||
|
||||
private async getRootDir(): Promise<string> {
|
||||
const roots = this.getRootDirCandidates()
|
||||
const dir = roots[0]
|
||||
await mkdir(dir, { recursive: true })
|
||||
return dir
|
||||
}
|
||||
|
||||
private async getAssistantsDir(): Promise<string> {
|
||||
const root = await this.getRootDir()
|
||||
const dir = join(root, 'assistants')
|
||||
await mkdir(dir, { recursive: true })
|
||||
return dir
|
||||
}
|
||||
|
||||
private async ensureInitialized(): Promise<void> {
|
||||
if (this.initialized) return
|
||||
const dir = await this.getAssistantsDir()
|
||||
|
||||
for (const builtin of BUILTIN_ASSISTANTS) {
|
||||
const filePath = join(dir, `${builtin.id}.md`)
|
||||
if (!existsSync(filePath)) {
|
||||
const parsed = parseAssistantMarkdown(builtin.raw)
|
||||
const config: AssistantConfigFull = {
|
||||
...parsed,
|
||||
builtinId: parsed.id
|
||||
}
|
||||
await writeFile(filePath, toMarkdown(config), 'utf8')
|
||||
}
|
||||
}
|
||||
|
||||
this.cache.clear()
|
||||
const files = await readdir(dir)
|
||||
for (const fileName of files) {
|
||||
if (!fileName.endsWith('.md')) continue
|
||||
const filePath = join(dir, fileName)
|
||||
try {
|
||||
const raw = await readFile(filePath, 'utf8')
|
||||
const parsed = parseAssistantMarkdown(raw)
|
||||
if (!parsed.id) continue
|
||||
this.cache.set(parsed.id, parsed)
|
||||
} catch {
|
||||
// ignore broken file
|
||||
}
|
||||
}
|
||||
this.initialized = true
|
||||
}
|
||||
|
||||
async getAll(): Promise<AssistantSummary[]> {
|
||||
await this.ensureInitialized()
|
||||
return Array.from(this.cache.values())
|
||||
.sort((a, b) => a.name.localeCompare(b.name, 'zh-Hans-CN'))
|
||||
.map((assistant) => ({ ...assistant }))
|
||||
}
|
||||
|
||||
async getConfig(id: string): Promise<AssistantConfigFull | null> {
|
||||
await this.ensureInitialized()
|
||||
const key = normalizeText(id)
|
||||
const config = this.cache.get(key)
|
||||
return config ? { ...config } : null
|
||||
}
|
||||
|
||||
async create(
|
||||
payload: Omit<AssistantConfigFull, 'id'> & { id?: string }
|
||||
): Promise<{ success: boolean; id?: string; error?: string }> {
|
||||
await this.ensureInitialized()
|
||||
const id = normalizeText(payload.id, `custom_${randomUUID().replace(/-/g, '').slice(0, 12)}`)
|
||||
if (this.cache.has(id)) return { success: false, error: '助手 ID 已存在' }
|
||||
const config: AssistantConfigFull = {
|
||||
id,
|
||||
name: normalizeText(payload.name, '新助手'),
|
||||
systemPrompt: normalizeText(payload.systemPrompt),
|
||||
presetQuestions: Array.isArray(payload.presetQuestions) ? payload.presetQuestions.map((item) => normalizeText(item)).filter(Boolean) : [],
|
||||
allowedBuiltinTools: Array.isArray(payload.allowedBuiltinTools) ? payload.allowedBuiltinTools.map((item) => normalizeText(item)).filter(Boolean) : [],
|
||||
builtinId: normalizeText(payload.builtinId) || undefined,
|
||||
applicableChatTypes: Array.isArray(payload.applicableChatTypes) ? payload.applicableChatTypes : [],
|
||||
supportedLocales: Array.isArray(payload.supportedLocales) ? payload.supportedLocales.map((item) => normalizeText(item)).filter(Boolean) : []
|
||||
}
|
||||
const dir = await this.getAssistantsDir()
|
||||
await writeFile(join(dir, `${id}.md`), toMarkdown(config), 'utf8')
|
||||
this.cache.set(id, config)
|
||||
return { success: true, id }
|
||||
}
|
||||
|
||||
async update(
|
||||
id: string,
|
||||
updates: Partial<AssistantConfigFull>
|
||||
): Promise<{ success: boolean; error?: string }> {
|
||||
await this.ensureInitialized()
|
||||
const key = normalizeText(id)
|
||||
const existing = this.cache.get(key)
|
||||
if (!existing) return { success: false, error: '助手不存在' }
|
||||
const next: AssistantConfigFull = {
|
||||
...existing,
|
||||
...updates,
|
||||
id: key,
|
||||
name: normalizeText(updates.name, existing.name),
|
||||
systemPrompt: updates.systemPrompt == null ? existing.systemPrompt : normalizeText(updates.systemPrompt),
|
||||
presetQuestions: Array.isArray(updates.presetQuestions) ? updates.presetQuestions.map((item) => normalizeText(item)).filter(Boolean) : existing.presetQuestions,
|
||||
allowedBuiltinTools: Array.isArray(updates.allowedBuiltinTools) ? updates.allowedBuiltinTools.map((item) => normalizeText(item)).filter(Boolean) : existing.allowedBuiltinTools,
|
||||
applicableChatTypes: Array.isArray(updates.applicableChatTypes) ? updates.applicableChatTypes : existing.applicableChatTypes,
|
||||
supportedLocales: Array.isArray(updates.supportedLocales) ? updates.supportedLocales.map((item) => normalizeText(item)).filter(Boolean) : existing.supportedLocales
|
||||
}
|
||||
const dir = await this.getAssistantsDir()
|
||||
await writeFile(join(dir, `${key}.md`), toMarkdown(next), 'utf8')
|
||||
this.cache.set(key, next)
|
||||
return { success: true }
|
||||
}
|
||||
|
||||
async delete(id: string): Promise<{ success: boolean; error?: string }> {
|
||||
await this.ensureInitialized()
|
||||
const key = normalizeText(id)
|
||||
if (key === 'general_cn' || key === 'general_en' || key === 'general_ja') {
|
||||
return { success: false, error: '默认助手不可删除' }
|
||||
}
|
||||
const dir = await this.getAssistantsDir()
|
||||
const filePath = join(dir, `${key}.md`)
|
||||
if (existsSync(filePath)) {
|
||||
await rm(filePath, { force: true })
|
||||
}
|
||||
this.cache.delete(key)
|
||||
return { success: true }
|
||||
}
|
||||
|
||||
async reset(id: string): Promise<{ success: boolean; error?: string }> {
|
||||
await this.ensureInitialized()
|
||||
const key = normalizeText(id)
|
||||
const existing = this.cache.get(key)
|
||||
if (!existing?.builtinId) {
|
||||
return { success: false, error: '该助手不支持重置' }
|
||||
}
|
||||
const builtin = BUILTIN_ASSISTANTS.find((item) => item.id === existing.builtinId)
|
||||
if (!builtin) return { success: false, error: '内置模板不存在' }
|
||||
const parsed = parseAssistantMarkdown(builtin.raw)
|
||||
const config: AssistantConfigFull = {
|
||||
...parsed,
|
||||
id: key,
|
||||
builtinId: existing.builtinId
|
||||
}
|
||||
const dir = await this.getAssistantsDir()
|
||||
await writeFile(join(dir, `${key}.md`), toMarkdown(config), 'utf8')
|
||||
this.cache.set(key, config)
|
||||
return { success: true }
|
||||
}
|
||||
|
||||
async getBuiltinCatalog(): Promise<BuiltinAssistantInfo[]> {
|
||||
await this.ensureInitialized()
|
||||
return BUILTIN_ASSISTANTS.map((builtin) => {
|
||||
const parsed = parseAssistantMarkdown(builtin.raw)
|
||||
const imported = Array.from(this.cache.values()).some((config) => config.builtinId === builtin.id || config.id === builtin.id)
|
||||
return {
|
||||
id: parsed.id,
|
||||
name: parsed.name,
|
||||
systemPrompt: parsed.systemPrompt,
|
||||
applicableChatTypes: parsed.applicableChatTypes,
|
||||
supportedLocales: parsed.supportedLocales,
|
||||
imported
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
async getBuiltinToolCatalog(): Promise<Array<{ name: string; category: AssistantToolCategory }>> {
|
||||
return defaultBuiltinToolCatalog()
|
||||
}
|
||||
|
||||
async importFromMd(rawMd: string): Promise<{ success: boolean; id?: string; error?: string }> {
|
||||
try {
|
||||
const parsed = parseAssistantMarkdown(rawMd)
|
||||
if (!parsed.id) return { success: false, error: '缺少 id' }
|
||||
if (this.cache.has(parsed.id)) return { success: false, error: '助手 ID 已存在' }
|
||||
const dir = await this.getAssistantsDir()
|
||||
await writeFile(join(dir, `${parsed.id}.md`), toMarkdown(parsed), 'utf8')
|
||||
this.cache.set(parsed.id, parsed)
|
||||
return { success: true, id: parsed.id }
|
||||
} catch (error) {
|
||||
return { success: false, error: String((error as Error)?.message || error) }
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
export const aiAssistantService = new AiAssistantService()
|
||||
Reference in New Issue
Block a user