feat: init v3

models/chaite/channel_storage.js

@@ -0,0 +1,74 @@
+import ChatGPTStorage from '../storage.js'
+
+/**
+ * @returns {import('chaite').ChannelsStorage}
+ */
+export async function createChannelsStorage () {
+  return new LowDBChannelStorage(ChatGPTStorage)
+}
+
+class LowDBChannelStorage {
+  /**
+   *
+   * @param { LowDBStorage } storage
+   */
+  constructor (storage) {
+    this.storage = storage
+    /**
+     * 集合
+     * @type {LowDBCollection}
+     */
+    this.collection = this.storage.collection('channel')
+  }
+
+  async saveChannel (channel) {
+    await this.collection.insert(channel)
+  }
+
+  async getChannel (id) {
+    return this.collection.collection()
+  }
+
+  /**
+   *
+   * @param name
+   * @returns {Promise<import('chaite').Channel[]>}
+   */
+  async getChannelByName (name) {
+    return this.collection.find({ name })
+  }
+
+  async deleteChannel (name) {
+    await this.collection.delete({ name })
+  }
+
+  /**
+   * 获取所有渠道
+   * @param {string?} model
+   * @returns {Promise<import('chaite').Channel[]>}
+   */
+  async getAllChannels (model) {
+    if (model) {
+      return this.collection.find({ 'options.model': model })
+    }
+    return this.collection.findAll()
+  }
+
+  /**
+   *
+   * @param {import('chaite').ClientType} type
+   * @returns {Promise<Object[]>}
+   */
+  async getChannelByType (type) {
+    return this.collection.find({ type })
+  }
+
+  /**
+   *
+   * @param {'enabled' | 'disabled'} status
+   * @returns {Promise<*>}
+   */
+  async getChannelByStatus (status) {
+    return this.collection.find({ status })
+  }
+}

models/chaite/chat_preset_storage.js

@@ -0,0 +1,53 @@
+import ChatGPTStorage from '../storage.js'
+
+/**
+ * @returns {import('chaite').ChatPresetsStorage}
+ */
+export async function createChatPresetsStorage () {
+  return new LowDBChatPresetsStorage(ChatGPTStorage)
+}
+
+class LowDBChatPresetsStorage {
+  /**
+   *
+   * @param { LowDBStorage } storage
+   */
+  constructor (storage) {
+    this.storage = storage
+    /**
+     * 集合
+     * @type {LowDBCollection}
+     */
+    this.collection = this.storage.collection('chat_presets')
+  }
+
+  /**
+   *
+   * @param {import('chaite').ChatPreset} preset
+   * @returns {Promise<void>}
+   */
+  async savePreset (preset) {
+    await this.collection.insert(preset)
+  }
+
+  /**
+   *
+   * @param { string } name
+   * @returns {Promise<import('chaite').ChatPreset | null>}
+   */
+  async getPreset (name) {
+    return this.collection.findOne({ name })
+  }
+
+  /**
+   *
+   * @returns {Promise<import('chaite').ChatPreset[]>}
+   */
+  async getAllPresets () {
+    return this.collection.findAll()
+  }
+
+  async deletePreset (name) {
+    await this.collection.delete({ name })
+  }
+}

models/chaite/cloud.js

@@ -0,0 +1,28 @@
+import { DefaultToolCloudService, ToolManager } from 'chaite'
+import ChatGPTConfig from '../../config/config.js'
+import { createToolsSettingsStorage } from './tool_settings_storage.js'
+const ChatGPTToolCloudService = new DefaultToolCloudService(ChatGPTConfig.cloudBaseUrl, '', {})
+/**
+ * @type {import('chaite').ToolManager}
+ */
+let ChatGPTToolManager
+ToolManager.getInstance(ChatGPTConfig.toolsDirPath, createToolsSettingsStorage(), ChatGPTToolCloudService).then((manager) => {
+  ChatGPTToolManager = manager
+})
+
+/**
+ * 认证，以便共享上传
+ * @param apiKey
+ * @returns {Promise<import('chaite').User>}
+ */
+export async function authCloud (apiKey) {
+  const user = await ChatGPTToolCloudService.authenticate(apiKey)
+  ChatGPTToolManager.setCloudService(ChatGPTToolCloudService)
+  return user
+}
+
+export default {
+  ChatGPTToolCloudService,
+  ChatGPTToolManager
+}
+

models/chaite/tool_settings_storage.js

@@ -0,0 +1,53 @@
+import ChatGPTStorage from '../storage.js'
+
+/**
+ * @returns {import('chaite').ToolSettingsStorage}
+ */
+export function createToolsSettingsStorage () {
+  return new LowDBToolsSettingsStorage(ChatGPTStorage)
+}
+
+class LowDBToolsSettingsStorage {
+  /**
+   *
+   * @param { LowDBStorage } storage
+   */
+  constructor (storage) {
+    this.storage = storage
+    /**
+     * 集合
+     * @type {LowDBCollection}
+     */
+    this.collection = this.storage.collection('tool_settings')
+  }
+
+  /**
+   *
+   * @param {import('chaite').ToolSettings} settings
+   * @returns {Promise<void>}
+   */
+  async saveToolSettings (settings) {
+    await this.collection.insert(settings)
+  }
+
+  /**
+   *
+   * @param { string } name
+   * @returns {Promise<import('chaite').ToolSettings | null>}
+   */
+  async getToolSettings (name) {
+    return this.collection.findOne({ name })
+  }
+
+  /**
+   *
+   * @returns {Promise<import('chaite').ToolSettings[]>}
+   */
+  async getAllToolSettings () {
+    return this.collection.findAll()
+  }
+
+  async deleteToolSettings (name) {
+    await this.collection.delete({ name })
+  }
+}

models/chaite/vector_database.js

@@ -0,0 +1,36 @@
+// todo
+class FaissVectorDatabase {
+  constructor (index) {
+    this.index = index
+  }
+
+  async addVector (vector, text) {
+  }
+
+  async addVectors (vectors, texts) {
+  }
+
+  async search (queryVector, k) {
+  }
+
+  async getVector (id) {
+  }
+
+  async deleteVector (id) {
+  }
+
+  async updateVector (id, newVector, newText) {
+  }
+
+  async count () {
+  }
+
+  async clear () {
+  }
+}
+
+/**
+ * 默认向量库
+ * @type {import('chaite').VectorDatabase}
+ */
+export const ChatGPTVectorDatabase = new FaissVectorDatabase()

models/storage.js

@@ -0,0 +1,358 @@
+// storage.js written by sonnet
+import { Low } from 'lowdb'
+import { JSONFile } from 'lowdb/node'
+import path from 'path'
+import fs from 'fs'
+import ChatGPTConfig from '../config/config.js'
+
+/**
+ * 基于 LowDB 的简单存储类，提供 CRUD 和条件查询功能
+ */
+export class LowDBStorage {
+  /**
+   * 创建一个新的存储实例
+   * @param {Object} options 配置选项
+   * @param {string} options.filename 数据文件名称
+   * @param {string} options.directory 数据目录，默认为当前目录下的 data 文件夹
+   */
+  constructor (options = {}) {
+    const { filename = 'db.json', directory = path.join(process.cwd(), 'data') } = options
+
+    // 确保目录存在
+    if (!fs.existsSync(directory)) {
+      fs.mkdirSync(directory, { recursive: true })
+    }
+
+    this.filePath = path.join(directory, filename)
+    this.adapter = new JSONFile(this.filePath)
+    this.db = new Low(this.adapter)
+    this.initialized = false
+  }
+
+  /**
+   * 初始化存储
+   * @returns {Promise<LowDBStorage>} 当前存储实例
+   */
+  async init () {
+    // 读取数据文件，如果不存在则创建默认结构
+    await this.db.read()
+    this.db.data ||= { collections: {} }
+    await this.db.write()
+
+    this.initialized = true
+    return this
+  }
+
+  /**
+   * 获取或创建一个集合
+   * @param {string} name 集合名称
+   * @returns {LowDBCollection} 集合实例
+   */
+  collection (name) {
+    this._checkInit()
+
+    // 确保集合存在
+    if (!this.db.data.collections[name]) {
+      this.db.data.collections[name] = []
+      this.db.write()
+    }
+
+    return new LowDBCollection(this, name)
+  }
+
+  /**
+   * 列出所有集合名称
+   * @returns {string[]} 集合名称列表
+   */
+  listCollections () {
+    this._checkInit()
+    return Object.keys(this.db.data.collections)
+  }
+
+  /**
+   * 删除一个集合
+   * @param {string} name 要删除的集合名称
+   * @returns {Promise<boolean>} 是否成功删除
+   */
+  async dropCollection (name) {
+    this._checkInit()
+
+    if (this.db.data.collections[name]) {
+      delete this.db.data.collections[name]
+      await this.db.write()
+      return true
+    }
+    return false
+  }
+
+  /**
+   * 检查存储是否已初始化
+   * @private
+   */
+  _checkInit () {
+    if (!this.initialized) {
+      throw new Error('存储尚未初始化，请先调用 init() 方法')
+    }
+  }
+}
+
+/**
+ * 集合类，提供对特定数据集合的操作
+ */
+export class LowDBCollection {
+  /**
+   * 创建一个集合实例
+   * @param {LowDBStorage} storage 所属存储实例
+   * @param {string} name 集合名称
+   */
+  constructor (storage, name) {
+    this.storage = storage
+    this.name = name
+  }
+
+  /**
+   * 获取集合数据引用
+   * @private
+   */
+  get _collection () {
+    return this.storage.db.data.collections[this.name]
+  }
+
+  /**
+   * 保存数据到存储
+   * @private
+   */
+  async _save () {
+    return this.storage.db.write()
+  }
+
+  /**
+   * 生成唯一ID
+   * @private
+   */
+  _generateId () {
+    return Date.now().toString(36) + Math.random().toString(36).substring(2, 15)
+  }
+
+  /**
+   * 创建新文档
+   * @param {Object} doc 要插入的文档
+   * @returns {Promise<Object>} 插入的文档（带ID）
+   */
+  async insert (doc) {
+    // 生成唯一ID，如果没有提供
+    if (!doc.id) {
+      doc.id = this._generateId()
+    }
+
+    // 加上时间戳
+    if (!doc.createdAt) {
+      doc.createdAt = new Date().toISOString()
+    }
+
+    doc.updatedAt = new Date().toISOString()
+
+    // 添加到集合
+    this._collection.push(doc)
+    await this._save()
+
+    return doc
+  }
+
+  /**
+   * 批量插入多个文档
+   * @param {Object[]} docs 要插入的文档数组
+   * @returns {Promise<Object[]>} 插入的文档（带ID）
+   */
+  async insertMany (docs) {
+    const inserted = []
+
+    for (const doc of docs) {
+      inserted.push(await this.insert(doc))
+    }
+
+    return inserted
+  }
+
+  /**
+   * 根据ID查找单个文档
+   * @param {string} id 文档ID
+   * @returns {Promise<Object|null>} 查找到的文档或null
+   */
+  async findById (id) {
+    return this._collection.find(doc => doc.id === id) || null
+  }
+
+  /**
+   * 返回集合中的所有文档
+   * @returns {Promise<Object[]>} 文档数组
+   */
+  async findAll () {
+    return [...this._collection]
+  }
+
+  /**
+   * 根据条件查找文档
+   * @param {Object} query 查询条件（字段等值匹配）
+   * @returns {Promise<Object[]>} 匹配的文档数组
+   */
+  async find (query = {}) {
+    return this._collection.filter(doc => {
+      for (const key in query) {
+        const value = query[key]
+
+        // 处理嵌套属性 (例如 user.profile.name)
+        if (key.includes('.')) {
+          const parts = key.split('.')
+          let current = doc
+
+          for (let i = 0; i < parts.length; i++) {
+            if (current === undefined || current === null) return false
+            current = current[parts[i]]
+          }
+
+          if (current !== value) return false
+        } else if (doc[key] !== value) {
+          return false
+        }
+      }
+      return true
+    })
+  }
+
+  /**
+   * 根据条件查找单个文档
+   * @param {Object} query 查询条件
+   * @returns {Promise<Object|null>} 第一个匹配的文档或null
+   */
+  async findOne (query = {}) {
+    const results = await this.find(query)
+    return results.length > 0 ? results[0] : null
+  }
+
+  /**
+   * 使用自定义函数进行高级查询
+   * @param {Function} filterFn 过滤函数
+   * @returns {Promise<Object[]>} 匹配的文档数组
+   */
+  async findWhere (filterFn) {
+    return this._collection.filter(filterFn)
+  }
+
+  /**
+   * 根据ID更新文档
+   * @param {string} id 文档ID
+   * @param {Object} updates 要更新的字段
+   * @returns {Promise<Object|null>} 更新后的文档或null
+   */
+  async updateById (id, updates) {
+    const index = this._collection.findIndex(doc => doc.id === id)
+
+    if (index === -1) return null
+
+    // 防止覆盖ID
+    const { id: _, ...safeUpdates } = updates
+
+    // 更新文档
+    const updatedDoc = {
+      ...this._collection[index],
+      ...safeUpdates,
+      updatedAt: new Date().toISOString()
+    }
+
+    this._collection[index] = updatedDoc
+    await this._save()
+
+    return updatedDoc
+  }
+
+  /**
+   * 根据条件更新文档
+   * @param {Object} query 查询条件
+   * @param {Object} updates 要更新的字段
+   * @returns {Promise<number>} 更新的文档数量
+   */
+  async update (query, updates) {
+    const matches = await this.find(query)
+    let updated = 0
+
+    for (const doc of matches) {
+      await this.updateById(doc.id, updates)
+      updated++
+    }
+
+    return updated
+  }
+
+  /**
+   * 根据ID删除文档
+   * @param {string} id 文档ID
+   * @returns {Promise<boolean>} 是否成功删除
+   */
+  async deleteById (id) {
+    const index = this._collection.findIndex(doc => doc.id === id)
+
+    if (index === -1) return false
+
+    this._collection.splice(index, 1)
+    await this._save()
+
+    return true
+  }
+
+  /**
+   * 根据条件删除文档
+   * @param {Object} query 查询条件
+   * @returns {Promise<number>} 删除的文档数量
+   */
+  async delete (query) {
+    const before = this._collection.length
+
+    const remaining = this._collection.filter(doc => {
+      for (const key in query) {
+        if (doc[key] !== query[key]) {
+          return true // 保留不匹配的
+        }
+      }
+      return false // 删除匹配的
+    })
+
+    this.storage.db.data.collections[this.name] = remaining
+    await this._save()
+
+    return before - remaining.length
+  }
+
+  /**
+   * 清空集合中的所有文档
+   * @returns {Promise<number>} 删除的文档数量
+   */
+  async deleteAll () {
+    const count = this._collection.length
+    this.storage.db.data.collections[this.name] = []
+    await this._save()
+    return count
+  }
+
+  /**
+   * 返回集合中文档的数量
+   * @returns {Promise<number>} 文档数量
+   */
+  async count (query = {}) {
+    if (Object.keys(query).length === 0) {
+      return this._collection.length
+    }
+
+    const matches = await this.find(query)
+    return matches.length
+  }
+}
+
+const ChatGPTStorage = new LowDBStorage({
+  filename: 'storage.json',
+  directory: ChatGPTConfig.dataDir
+})
+
+ChatGPTStorage.init()
+
+export default ChatGPTStorage