glm.js 614 B

1234567891011121314151617181920212223
  1. const { BaseModelProvider } = require('../base/BaseModelProvider');
  2. const { ChatZhipuAI } = require('@langchain/community/chat_models/zhipuai');
  3. const { ZHIPU_AI_KEY } = require('../../../../config');
  4. class GlmModelProvider extends BaseModelProvider {
  5. async createModel() {
  6. const aiKey = ZHIPU_AI_KEY;
  7. const model_name = 'glm-4';
  8. const model = new ChatZhipuAI({
  9. apiKey: aiKey,
  10. model: model_name,
  11. temperature: 0.95, // never use 1.0, some models do not support it
  12. maxRetries: 3,
  13. verbose: true,
  14. });
  15. return model;
  16. }
  17. }
  18. module.exports = {
  19. GlmModelProvider,
  20. };