@hlw-zbzs/ai 1.0.1

package/src/core/index.ts

@@ -0,0 +1,169 @@
+/**
+ * @file AI 核心响应式 Hook (单单词极简命名，零冗余代码)
+ */
+
+import { computed, ref } from 'vue'
+import { storeToRefs } from 'pinia'
+import { useAiStore } from '../store'
+import { Engine } from '../engines/base'
+import type {
+	AiConfig,
+	AiEngine,
+	AiHandler,
+	AiInfo,
+	AiMsg,
+	AiOption,
+	AiResult,
+	AiTest
+} from '../types'
+
+const engines = new Map<AiEngine, Engine>()
+
+function getEngine(type: AiEngine, conf: AiConfig): Engine {
+	let instance = engines.get(type)
+	if (!instance) {
+		instance = new Engine(type, conf)
+		engines.set(type, instance)
+	} else {
+		instance.save(conf)
+	}
+	return instance
+}
+
+export function useAi() {
+	const store = useAiStore()
+	const { engine, enabled } = storeToRefs(store)
+
+	const loading = ref(false)
+	const text = ref('')
+	const think = ref('')
+	let controller: AbortController | null = null
+
+	const current = computed<Engine>(() => getEngine(store.engine, store.conf))
+	const info = computed<AiInfo>(() => store.info)
+	const conf = computed<AiConfig>(() => store.conf)
+	const ready = computed<boolean>(() => store.ready)
+	const list = computed<AiInfo[]>(() => store.list)
+
+	function select(type: AiEngine) {
+		store.set(type)
+	}
+
+	function abort() {
+		if (controller) {
+			controller.abort()
+			controller = null
+		}
+		loading.value = false
+	}
+
+	async function chat(
+		input: string | AiMsg[],
+		option?: Partial<AiOption>,
+		type?: AiEngine
+	): Promise<string> {
+		if (!enabled.value) throw new Error('AI 功能已停用')
+		loading.value = true
+		text.value = ''
+		think.value = ''
+		controller = new AbortController()
+
+		const target = type ? getEngine(type, store.configs[type] || store.conf) : current.value
+
+		try {
+			const list: AiMsg[] =
+				typeof input === 'string' ? [{ role: 'user', text: input }] : input
+
+			const res: AiResult = await target.chat({
+				list,
+				signal: controller.signal,
+				...option
+			})
+
+			text.value = res.text
+			think.value = res.think || ''
+			return res.text
+		} finally {
+			loading.value = false
+			controller = null
+		}
+	}
+
+	async function stream(
+		input: string | AiMsg[],
+		handler: AiHandler = {},
+		type?: AiEngine
+	): Promise<string> {
+		if (!enabled.value) throw new Error('AI 功能已停用')
+		loading.value = true
+		text.value = ''
+		think.value = ''
+		controller = new AbortController()
+
+		const target = type ? getEngine(type, store.configs[type] || store.conf) : current.value
+
+		try {
+			const list: AiMsg[] =
+				typeof input === 'string' ? [{ role: 'user', text: input }] : input
+
+			const res: AiResult = await target.stream(
+				{
+					list,
+					signal: controller.signal
+				},
+				{
+					onToken: (delta) => {
+						text.value += delta
+						handler.onToken?.(delta)
+					},
+					onThink: (delta) => {
+						think.value += delta
+						handler.onThink?.(delta)
+					},
+					onDone: (fullText, fullThink) => {
+						text.value = fullText
+						think.value = fullThink || ''
+						handler.onDone?.(fullText, fullThink)
+					},
+					onError: (err) => {
+						handler.onError?.(err)
+					}
+				}
+			)
+
+			return res.text
+		} finally {
+			loading.value = false
+			controller = null
+		}
+	}
+
+	async function test(type?: AiEngine, tempConf?: Partial<AiConfig>): Promise<AiTest> {
+		const targetType = type || store.engine
+		const targetConf = store.configs[targetType] || store.conf
+		const target = getEngine(targetType, targetConf)
+		return target.test(tempConf)
+	}
+
+	return {
+		store,
+		engine,
+		enabled,
+		conf,
+		info,
+		list,
+		ready,
+		loading: computed(() => loading.value),
+		text: computed(() => text.value),
+		think: computed(() => think.value),
+		select,
+		chat,
+		stream,
+		test,
+		abort,
+		save: store.save.bind(store),
+		reset: store.reset.bind(store)
+	}
+}
+
+export default useAi

package/src/engines/base.ts

@@ -0,0 +1,142 @@
+/**
+ * @file 大模型底层适配引擎 (基于官方 OpenAI SDK，单单词极简命名)
+ */
+
+import OpenAI from 'openai'
+import type { AiConfig, AiEngine, AiHandler, AiMsg, AiOption, AiResult, AiTest } from '../types'
+
+export const URLS: Record<AiEngine, string> = {
+	deepseek: 'https://api.deepseek.com/v1',
+	qwen: 'https://dashscope.aliyuncs.com/compatible-mode/v1',
+	chatgpt: 'https://api.openai.com/v1',
+	zhipu: 'https://open.bigmodel.cn/api/paas/v4',
+	doubao: 'https://ark.cn-beijing.volces.com/api/v3',
+	moonshot: 'https://api.moonshot.cn/v1',
+	claude: 'https://api.anthropic.com/v1',
+	ollama: 'http://localhost:11434/v1',
+	custom: 'https://api.openai.com/v1'
+}
+
+export class Engine {
+	public type: AiEngine
+	public conf: AiConfig
+	private client: OpenAI | null = null
+
+	constructor(type: AiEngine, conf: AiConfig) {
+		this.type = type
+		this.conf = { ...conf }
+		this.init()
+	}
+
+	private init() {
+		const url = (this.conf.url || '').trim().replace(/\/+$/, '') || URLS[this.type]
+		const key = (this.conf.key || '').trim() || 'dummy-key'
+		this.client = new OpenAI({
+			baseURL: url,
+			apiKey: key,
+			dangerouslyAllowBrowser: true
+		})
+	}
+
+	public save(conf: Partial<AiConfig>) {
+		Object.assign(this.conf, conf)
+		this.init()
+	}
+
+	private build(option: AiOption): OpenAI.Chat.ChatCompletionMessageParam[] {
+		const list: OpenAI.Chat.ChatCompletionMessageParam[] = []
+		const sys = option.sys ?? this.conf.sys
+		if (sys && sys.trim()) {
+			list.push({ role: 'system', content: sys.trim() })
+		}
+		for (const item of option.list) {
+			if (item.role === 'system' && sys) continue
+			list.push({
+				role: item.role as 'user' | 'assistant' | 'system',
+				content: item.text
+			})
+		}
+		return list
+	}
+
+	public async chat(option: AiOption): Promise<AiResult> {
+		if (!this.client) this.init()
+		const list = this.build(option)
+		const res = await this.client!.chat.completions.create(
+			{
+				model: option.model || this.conf.model,
+				messages: list,
+				temperature: option.temp ?? this.conf.temp ?? 0.7,
+				max_tokens: option.tokens ?? this.conf.tokens ?? 2048,
+				stream: false
+			},
+			{ signal: option.signal }
+		)
+		const choice = res.choices?.[0]
+		const text = choice?.message?.content || ''
+		const think = (choice?.message as any)?.reasoning_content || ''
+		return { text, think }
+	}
+
+	public async stream(option: AiOption, handler: AiHandler = {}): Promise<AiResult> {
+		if (!this.client) this.init()
+		const list = this.build(option)
+		let text = ''
+		let think = ''
+		try {
+			const stream = await this.client!.chat.completions.create(
+				{
+					model: option.model || this.conf.model,
+					messages: list,
+					temperature: option.temp ?? this.conf.temp ?? 0.7,
+					max_tokens: option.tokens ?? this.conf.tokens ?? 2048,
+					stream: true
+				},
+				{ signal: option.signal }
+			)
+			for await (const chunk of stream) {
+				const delta = chunk.choices?.[0]?.delta
+				if (!delta) continue
+				const thought = (delta as any)?.reasoning_content
+				if (thought) {
+					think += thought
+					handler.onThink?.(thought)
+				}
+				if (delta.content) {
+					text += delta.content
+					handler.onToken?.(delta.content)
+				}
+			}
+			handler.onDone?.(text, think)
+			return { text, think }
+		} catch (error: any) {
+			handler.onError?.(error)
+			throw error
+		}
+	}
+
+	public async test(tempConfig?: Partial<AiConfig>): Promise<AiTest> {
+		const prev = { ...this.conf }
+		if (tempConfig) this.save(tempConfig)
+		const time = Date.now()
+		try {
+			const res = await this.chat({
+				list: [{ role: 'user', text: 'Hi' }],
+				tokens: 5
+			})
+			return {
+				success: true,
+				msg: '连接成功',
+				latency: Date.now() - time
+			}
+		} catch (error: any) {
+			return {
+				success: false,
+				msg: error?.message || '连接失败',
+				latency: Date.now() - time
+			}
+		} finally {
+			this.save(prev)
+		}
+	}
+}

package/src/engines/index.ts

@@ -0,0 +1,12 @@
+/**
+ * @file 大模型引擎聚合与工厂
+ */
+
+import { Engine } from './base'
+import type { AiConfig, AiEngine } from '../types'
+
+export * from './base'
+
+export function createEngine(type: AiEngine, conf: AiConfig): Engine {
+	return new Engine(type, conf)
+}

package/src/store/index.ts

@@ -0,0 +1,158 @@
+/**
+ * @file AI 状态管理 (单单词极简设计，支持各主流模型配置持久化)
+ */
+
+import { defineStore } from 'pinia'
+import { electronStorage } from '@hlw-zbzs/core'
+import { URLS } from '../engines/base'
+import type { AiConfig, AiEngine, AiInfo } from '../types'
+
+export const ENGINES: AiInfo[] = [
+	{
+		type: 'deepseek',
+		name: 'DeepSeek',
+		tag: '超高性价比 / 深度思考',
+		enable: true,
+		models: ['deepseek-chat', 'deepseek-reasoner'],
+		model: 'deepseek-chat',
+		url: 'https://platform.deepseek.com/api_keys'
+	},
+	{
+		type: 'qwen',
+		name: '通义千问',
+		tag: '阿里百炼 / 国产领先',
+		enable: true,
+		models: ['qwen-max', 'qwen-plus', 'qwen-turbo', 'qwen-long', 'qwen2.5-72b-instruct'],
+		model: 'qwen-plus',
+		url: 'https://bailian.console.aliyun.com/?apiKey=1#/api-key'
+	},
+	{
+		type: 'chatgpt',
+		name: 'ChatGPT',
+		tag: 'OpenAI / 强大通用',
+		enable: true,
+		models: ['gpt-4o', 'gpt-4o-mini', 'o1', 'o3-mini', 'gpt-4-turbo', 'gpt-3.5-turbo'],
+		model: 'gpt-4o-mini',
+		url: 'https://platform.openai.com/api-keys'
+	},
+	{
+		type: 'zhipu',
+		name: '智谱 GLM',
+		tag: '清华基座 / 智能体',
+		enable: true,
+		models: ['glm-4-plus', 'glm-4-air', 'glm-4-flash', 'glm-4-long', 'glm-zero-preview'],
+		model: 'glm-4-flash',
+		url: 'https://bigmodel.cn/usercenter/apikeys'
+	},
+	{
+		type: 'doubao',
+		name: '字节豆包',
+		tag: '火山引擎 / 低延迟',
+		enable: true,
+		models: ['doubao-pro-32k', 'doubao-lite-32k', 'doubao-pro-128k'],
+		model: 'doubao-pro-32k',
+		url: 'https://console.volcengine.com/ark/region:ark+cn-beijing/apiKey'
+	},
+	{
+		type: 'moonshot',
+		name: 'Moonshot',
+		tag: 'Kimi / 超长上下文',
+		enable: true,
+		models: ['moonshot-v1-8k', 'moonshot-v1-32k', 'moonshot-v1-128k'],
+		model: 'moonshot-v1-8k',
+		url: 'https://platform.moonshot.cn/console/api-keys'
+	},
+	{
+		type: 'claude',
+		name: 'Claude',
+		tag: 'Anthropic / 顶尖逻辑',
+		enable: true,
+		models: ['claude-3-5-sonnet-20241022', 'claude-3-5-haiku-20241022', 'claude-3-opus-20240229'],
+		model: 'claude-3-5-sonnet-20241022',
+		url: 'https://console.anthropic.com/settings/keys'
+	},
+	{
+		type: 'ollama',
+		name: 'Ollama',
+		tag: '本地私有 / 免密免Key',
+		free: true,
+		enable: true,
+		models: ['deepseek-r1', 'llama3.3', 'qwen2.5', 'gemma2', 'mistral'],
+		model: 'deepseek-r1',
+		url: 'https://ollama.com'
+	},
+	{
+		type: 'custom',
+		name: '自定义 API',
+		tag: '第三方中转 / 兼容协议',
+		enable: true,
+		models: ['default'],
+		model: 'default'
+	}
+]
+
+export function getDefaults(): Record<AiEngine, AiConfig> {
+	const result: Partial<Record<AiEngine, AiConfig>> = {}
+	for (const item of ENGINES) {
+		result[item.type] = {
+			key: '',
+			url: URLS[item.type],
+			model: item.model,
+			temp: 0.7,
+			tokens: 2048,
+			sys: '你是一名专业热情的直播间智能助手，负责协助主播进行弹幕互动与话术生成。'
+		}
+	}
+	return result as Record<AiEngine, AiConfig>
+}
+
+export const useAiStore = defineStore('ai', {
+	state: () => ({
+		engine: 'deepseek' as AiEngine,
+		enabled: true,
+		configs: getDefaults()
+	}),
+
+	getters: {
+		conf: (state): AiConfig => state.configs[state.engine] || getDefaults()[state.engine],
+		ready: (state): boolean => {
+			if (state.engine === 'ollama') return true
+			const target = state.configs[state.engine]
+			return Boolean(target?.key?.trim())
+		},
+		list: () => ENGINES,
+		info: (state): AiInfo => ENGINES.find((item) => item.type === state.engine) || ENGINES[0]
+	},
+
+	actions: {
+		set(type: AiEngine) {
+			this.engine = type
+		},
+		toggle() {
+			this.enabled = !this.enabled
+		},
+		save(type: AiEngine, conf: Partial<AiConfig>) {
+			if (!this.configs[type]) {
+				this.configs[type] = getDefaults()[type]
+			}
+			Object.assign(this.configs[type], conf)
+		},
+		reset(type?: AiEngine) {
+			const defaults = getDefaults()
+			if (type) {
+				this.configs[type] = defaults[type]
+			} else {
+				this.configs = defaults
+				this.engine = 'deepseek'
+				this.enabled = true
+			}
+		}
+	},
+
+	persist: {
+		key: 'ai.config',
+		storage: electronStorage
+	}
+})
+
+export default useAiStore

package/src/types.d.ts

@@ -0,0 +1 @@
+export type * from '../types'