@hlw-zbzs/ai 1.0.2 → 1.0.3

package/src/engines/base.ts

@@ -1,142 +0,0 @@
-/**
- * @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

@@ -1,12 +0,0 @@
-/**
- * @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

@@ -1,158 +0,0 @@
-/**
- * @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

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