flowmind 1.0.1 → 1.1.0

package/core/ai/providers/ernie.js

@@ -0,0 +1,111 @@
+/**
+ * ERNIE Provider - 百度文心一言模型接入
+ * 支持 ERNIE-4.0-Turbo-8K、ERNIE-3.5-8K 等模型
+ */
+
+const BaseModel = require('../base-model');
+
+class ERNIEProvider extends BaseModel {
+  constructor(config = {}) {
+    super('ernie', config);
+    this.apiKey = config.apiKey || process.env.BAIDU_API_KEY;
+    this.secretKey = config.secretKey || process.env.BAIDU_SECRET_KEY;
+    this.model = config.model || 'ernie-4.0-turbo-8k';
+    this.baseUrl = config.baseUrl || 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop';
+    this.temperature = config.temperature ?? 0.3;
+    this.maxTokens = config.maxTokens ?? 2000;
+    this.accessToken = null;
+  }
+
+  async init() {
+    if (!this.apiKey || !this.secretKey) {
+      throw new Error('Baidu API key and secret key are required. Set BAIDU_API_KEY and BAIDU_SECRET_KEY environment variables or provide in config.');
+    }
+    await this.refreshAccessToken();
+    this.initialized = true;
+  }
+
+  async refreshAccessToken() {
+    const response = await fetch(
+      `https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=${this.apiKey}&client_secret=${this.secretKey}`,
+      { method: 'POST' }
+    );
+
+    if (!response.ok) {
+      throw new Error('Failed to get Baidu access token');
+    }
+
+    const data = await response.json();
+    this.accessToken = data.access_token;
+  }
+
+  validateConfig() {
+    return !!this.apiKey && !!this.secretKey;
+  }
+
+  async chat(messages, options = {}) {
+    if (!this.initialized) {
+      await this.init();
+    }
+
+    const model = options.model || this.model;
+    const endpoint = this.getEndpoint(model);
+
+    const response = await fetch(`${this.baseUrl}${endpoint}?access_token=${this.accessToken}`, {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json'
+      },
+      body: JSON.stringify({
+        messages: messages,
+        temperature: options.temperature ?? this.temperature,
+        max_output_tokens: options.maxTokens ?? this.maxTokens
+      })
+    });
+
+    if (!response.ok) {
+      const error = await response.text();
+      throw new Error(`ERNIE API error: ${response.status} - ${error}`);
+    }
+
+    const data = await response.json();
+    return data.result;
+  }
+
+  getEndpoint(model) {
+    const endpoints = {
+      'ernie-4.0-turbo-8k': '/chat/ernie-4.0-turbo-8k',
+      'ernie-4.0-8k': '/chat/ernie-4.0-8k',
+      'ernie-3.5-8k': '/chat/ernie-3.5-8k',
+      'ernie-3.5-4k-0205': '/chat/ernie-3.5-4k-0205',
+      'ernie-speed-8k': '/chat/ernie-speed-8k',
+      'ernie-lite-8k': '/chat/ernie-lite-8k'
+    };
+    return endpoints[model] || `/chat/${model}`;
+  }
+
+  async complete(prompt, options = {}) {
+    return this.chat([{ role: 'user', content: prompt }], options);
+  }
+
+  async isAvailable() {
+    try {
+      if (!this.apiKey || !this.secretKey) return false;
+      await this.refreshAccessToken();
+      return true;
+    } catch {
+      return false;
+    }
+  }
+
+  getInfo() {
+    return {
+      ...super.getInfo(),
+      model: this.model,
+      baseUrl: this.baseUrl,
+      provider: 'Baidu'
+    };
+  }
+}
+
+module.exports = ERNIEProvider;

package/core/ai/providers/glm.js

@@ -0,0 +1,80 @@
+/**
+ * GLM Provider - 智谱AI GLM 模型接入
+ * 支持 GLM-4、GLM-4-Flash、ChatGLM 等模型
+ */
+
+const BaseModel = require('../base-model');
+
+class GLMProvider extends BaseModel {
+  constructor(config = {}) {
+    super('glm', config);
+    this.apiKey = config.apiKey || process.env.ZHIPU_API_KEY;
+    this.model = config.model || 'glm-4-flash';
+    this.baseUrl = config.baseUrl || 'https://open.bigmodel.cn/api/paas/v4';
+    this.temperature = config.temperature ?? 0.3;
+    this.maxTokens = config.maxTokens ?? 2000;
+  }
+
+  async init() {
+    if (!this.apiKey) {
+      throw new Error('Zhipu API key is required. Set ZHIPU_API_KEY environment variable or provide in config.');
+    }
+    this.initialized = true;
+  }
+
+  validateConfig() {
+    return !!this.apiKey;
+  }
+
+  async chat(messages, options = {}) {
+    if (!this.initialized) {
+      await this.init();
+    }
+
+    const response = await fetch(`${this.baseUrl}/chat/completions`, {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json',
+        'Authorization': `Bearer ${this.apiKey}`
+      },
+      body: JSON.stringify({
+        model: options.model || this.model,
+        messages: messages,
+        temperature: options.temperature ?? this.temperature,
+        max_tokens: options.maxTokens ?? this.maxTokens
+      })
+    });
+
+    if (!response.ok) {
+      const error = await response.text();
+      throw new Error(`GLM API error: ${response.status} - ${error}`);
+    }
+
+    const data = await response.json();
+    return data.choices[0].message.content;
+  }
+
+  async complete(prompt, options = {}) {
+    return this.chat([{ role: 'user', content: prompt }], options);
+  }
+
+  async isAvailable() {
+    try {
+      if (!this.apiKey) return false;
+      return true;
+    } catch {
+      return false;
+    }
+  }
+
+  getInfo() {
+    return {
+      ...super.getInfo(),
+      model: this.model,
+      baseUrl: this.baseUrl,
+      provider: 'Zhipu AI'
+    };
+  }
+}
+
+module.exports = GLMProvider;

package/core/ai/providers/mimo.js

@@ -0,0 +1,80 @@
+/**
+ * MiMo Provider - 小米 MiMo 模型接入
+ * 支持 MiMo-7B 等模型
+ */
+
+const BaseModel = require('../base-model');
+
+class MiMoProvider extends BaseModel {
+  constructor(config = {}) {
+    super('mimo', config);
+    this.apiKey = config.apiKey || process.env.MIMO_API_KEY;
+    this.model = config.model || 'mimo-7b';
+    this.baseUrl = config.baseUrl || 'https://api.mimo.ai/v1';
+    this.temperature = config.temperature ?? 0.3;
+    this.maxTokens = config.maxTokens ?? 2000;
+  }
+
+  async init() {
+    if (!this.apiKey) {
+      throw new Error('MiMo API key is required. Set MIMO_API_KEY environment variable or provide in config.');
+    }
+    this.initialized = true;
+  }
+
+  validateConfig() {
+    return !!this.apiKey;
+  }
+
+  async chat(messages, options = {}) {
+    if (!this.initialized) {
+      await this.init();
+    }
+
+    const response = await fetch(`${this.baseUrl}/chat/completions`, {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json',
+        'Authorization': `Bearer ${this.apiKey}`
+      },
+      body: JSON.stringify({
+        model: options.model || this.model,
+        messages: messages,
+        temperature: options.temperature ?? this.temperature,
+        max_tokens: options.maxTokens ?? this.maxTokens
+      })
+    });
+
+    if (!response.ok) {
+      const error = await response.text();
+      throw new Error(`MiMo API error: ${response.status} - ${error}`);
+    }
+
+    const data = await response.json();
+    return data.choices[0].message.content;
+  }
+
+  async complete(prompt, options = {}) {
+    return this.chat([{ role: 'user', content: prompt }], options);
+  }
+
+  async isAvailable() {
+    try {
+      if (!this.apiKey) return false;
+      return true;
+    } catch {
+      return false;
+    }
+  }
+
+  getInfo() {
+    return {
+      ...super.getInfo(),
+      model: this.model,
+      baseUrl: this.baseUrl,
+      provider: 'Xiaomi'
+    };
+  }
+}
+
+module.exports = MiMoProvider;

package/core/ai/providers/ollama.js

@@ -0,0 +1,147 @@
+/**
+ * Ollama Provider - 本地模型接入
+ * 支持 Ollama 运行的本地模型（Llama2、Mistral、CodeLlama 等）
+ */
+
+const BaseModel = require('../base-model');
+
+class OllamaProvider extends BaseModel {
+  constructor(config = {}) {
+    super('ollama', config);
+    this.baseUrl = config.baseUrl || 'http://localhost:11434';
+    this.model = config.model || 'llama2';
+    this.temperature = config.temperature ?? 0.3;
+  }
+
+  async init() {
+    // 检查 Ollama 服务是否可用
+    const available = await this.isAvailable();
+    if (!available) {
+      throw new Error(`Ollama service is not available at ${this.baseUrl}. Please start Ollama first.`);
+    }
+    this.initialized = true;
+  }
+
+  validateConfig() {
+    return !!this.baseUrl && !!this.model;
+  }
+
+  async chat(messages, options = {}) {
+    if (!this.initialized) {
+      await this.init();
+    }
+
+    // 转换消息格式为 Ollama 格式
+    const prompt = this._convertMessagesToPrompt(messages);
+
+    const response = await fetch(`${this.baseUrl}/api/generate`, {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json'
+      },
+      body: JSON.stringify({
+        model: options.model || this.model,
+        prompt: prompt,
+        stream: false,
+        options: {
+          temperature: options.temperature ?? this.temperature,
+          num_predict: options.maxTokens ?? 2000
+        }
+      })
+    });
+
+    if (!response.ok) {
+      const error = await response.text();
+      throw new Error(`Ollama API error: ${response.status} - ${error}`);
+    }
+
+    const data = await response.json();
+    return data.response;
+  }
+
+  async complete(prompt, options = {}) {
+    if (!this.initialized) {
+      await this.init();
+    }
+
+    const response = await fetch(`${this.baseUrl}/api/generate`, {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json'
+      },
+      body: JSON.stringify({
+        model: options.model || this.model,
+        prompt: prompt,
+        stream: false,
+        options: {
+          temperature: options.temperature ?? this.temperature,
+          num_predict: options.maxTokens ?? 2000
+        }
+      })
+    });
+
+    if (!response.ok) {
+      const error = await response.text();
+      throw new Error(`Ollama API error: ${response.status} - ${error}`);
+    }
+
+    const data = await response.json();
+    return data.response;
+  }
+
+  async isAvailable() {
+    try {
+      const response = await fetch(`${this.baseUrl}/api/tags`, {
+        method: 'GET',
+        signal: AbortSignal.timeout(5000) // 5秒超时
+      });
+      return response.ok;
+    } catch {
+      return false;
+    }
+  }
+
+  /**
+   * 获取可用模型列表
+   * @returns {Promise<Array>}
+   */
+  async listModels() {
+    try {
+      const response = await fetch(`${this.baseUrl}/api/tags`);
+      if (!response.ok) return [];
+      const data = await response.json();
+      return data.models || [];
+    } catch {
+      return [];
+    }
+  }
+
+  /**
+   * 将消息数组转换为单个提示词
+   * @private
+   */
+  _convertMessagesToPrompt(messages) {
+    let prompt = '';
+    for (const msg of messages) {
+      if (msg.role === 'system') {
+        prompt += `[System]: ${msg.content}\n\n`;
+      } else if (msg.role === 'user') {
+        prompt += `[User]: ${msg.content}\n\n`;
+      } else if (msg.role === 'assistant') {
+        prompt += `[Assistant]: ${msg.content}\n\n`;
+      }
+    }
+    prompt += '[Assistant]: ';
+    return prompt;
+  }
+
+  getInfo() {
+    return {
+      ...super.getInfo(),
+      model: this.model,
+      baseUrl: this.baseUrl
+    };
+  }
+}
+
+module.exports = OllamaProvider;

package/core/ai/providers/openai.js

@@ -0,0 +1,82 @@
+/**
+ * OpenAI Provider - OpenAI GPT 模型接入
+ */
+
+const BaseModel = require('../base-model');
+
+class OpenAIProvider extends BaseModel {
+  constructor(config = {}) {
+    super('openai', config);
+    this.apiKey = config.apiKey || process.env.OPENAI_API_KEY;
+    this.model = config.model || 'gpt-4';
+    this.baseUrl = config.baseUrl || 'https://api.openai.com/v1';
+    this.temperature = config.temperature ?? 0.3;
+    this.maxTokens = config.maxTokens ?? 2000;
+  }
+
+  async init() {
+    if (!this.apiKey) {
+      throw new Error('OpenAI API key is required. Set OPENAI_API_KEY environment variable or provide in config.');
+    }
+    this.initialized = true;
+  }
+
+  validateConfig() {
+    return !!this.apiKey;
+  }
+
+  async chat(messages, options = {}) {
+    if (!this.initialized) {
+      await this.init();
+    }
+
+    const response = await fetch(`${this.baseUrl}/chat/completions`, {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json',
+        'Authorization': `Bearer ${this.apiKey}`
+      },
+      body: JSON.stringify({
+        model: options.model || this.model,
+        messages: messages,
+        temperature: options.temperature ?? this.temperature,
+        max_tokens: options.maxTokens ?? this.maxTokens,
+        response_format: options.responseFormat || undefined
+      })
+    });
+
+    if (!response.ok) {
+      const error = await response.text();
+      throw new Error(`OpenAI API error: ${response.status} - ${error}`);
+    }
+
+    const data = await response.json();
+    return data.choices[0].message.content;
+  }
+
+  async complete(prompt, options = {}) {
+    return this.chat([{ role: 'user', content: prompt }], options);
+  }
+
+  async isAvailable() {
+    try {
+      if (!this.apiKey) return false;
+      const response = await fetch(`${this.baseUrl}/models`, {
+        headers: { 'Authorization': `Bearer ${this.apiKey}` }
+      });
+      return response.ok;
+    } catch {
+      return false;
+    }
+  }
+
+  getInfo() {
+    return {
+      ...super.getInfo(),
+      model: this.model,
+      baseUrl: this.baseUrl
+    };
+  }
+}
+
+module.exports = OpenAIProvider;

package/core/ai/providers/qwen.js

@@ -0,0 +1,80 @@
+/**
+ * Qwen Provider - 阿里云通义千问模型接入
+ * 支持 Qwen-Turbo、Qwen-Plus、Qwen-Max 等模型
+ */
+
+const BaseModel = require('../base-model');
+
+class QwenProvider extends BaseModel {
+  constructor(config = {}) {
+    super('qwen', config);
+    this.apiKey = config.apiKey || process.env.DASHSCOPE_API_KEY;
+    this.model = config.model || 'qwen-turbo';
+    this.baseUrl = config.baseUrl || 'https://dashscope.aliyuncs.com/compatible-mode/v1';
+    this.temperature = config.temperature ?? 0.3;
+    this.maxTokens = config.maxTokens ?? 2000;
+  }
+
+  async init() {
+    if (!this.apiKey) {
+      throw new Error('DashScope API key is required. Set DASHSCOPE_API_KEY environment variable or provide in config.');
+    }
+    this.initialized = true;
+  }
+
+  validateConfig() {
+    return !!this.apiKey;
+  }
+
+  async chat(messages, options = {}) {
+    if (!this.initialized) {
+      await this.init();
+    }
+
+    const response = await fetch(`${this.baseUrl}/chat/completions`, {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json',
+        'Authorization': `Bearer ${this.apiKey}`
+      },
+      body: JSON.stringify({
+        model: options.model || this.model,
+        messages: messages,
+        temperature: options.temperature ?? this.temperature,
+        max_tokens: options.maxTokens ?? this.maxTokens
+      })
+    });
+
+    if (!response.ok) {
+      const error = await response.text();
+      throw new Error(`Qwen API error: ${response.status} - ${error}`);
+    }
+
+    const data = await response.json();
+    return data.choices[0].message.content;
+  }
+
+  async complete(prompt, options = {}) {
+    return this.chat([{ role: 'user', content: prompt }], options);
+  }
+
+  async isAvailable() {
+    try {
+      if (!this.apiKey) return false;
+      return true;
+    } catch {
+      return false;
+    }
+  }
+
+  getInfo() {
+    return {
+      ...super.getInfo(),
+      model: this.model,
+      baseUrl: this.baseUrl,
+      provider: 'Alibaba Cloud'
+    };
+  }
+}
+
+module.exports = QwenProvider;