wechat-bot/src/doubao/index.js

wechat-bot

import { remark } from 'remark'
import stripMarkdown from 'strip-markdown'
import OpenAI from 'openai'
import dotenv from 'dotenv'
const env = dotenv.config().parsed // 环境参数
import fs from 'fs'
import path from 'path'

const __dirname = path.resolve()
// 判断是否有 .env 文件, 没有则报错
const envPath = path.join(__dirname, '.env')
if (!fs.existsSync(envPath)) {
  console.log('❌ 请先根据文档，创建并配置.env文件！')
  process.exit(1)
}

let config = {
  apiKey: env.DOUBAO_API_KEY,
  baseURL: env.DOUBAO_URL,
}
const openai = new OpenAI(config)
const chosen_model = env.DOUBAO_MODEL
export async function getDoubaoReply(prompt, img_url = '') {
  const only_text = img_url == ''
  console.log('🚀🚀🚀 / prompt', prompt)
  let response
  if (only_text) {
    response = await openai.chat.completions.create({
      messages: [{ role: 'user', content: [{ type: 'text', text: prompt }] }],
      model: chosen_model,
    })
  } else {
    response = await openai.chat.completions.create({
      messages: [
        {
          role: 'user',
          content: [
            {
              type: 'image_url',
              image_url: {
                url: img_url,
              },
            },
            { type: 'text', text: prompt },
          ],
        },
      ],
      model: chosen_model,
    })
  }
  console.log('🚀🚀🚀 / reply', response.choices[0].message.content)
  return `${response.choices[0].message.content}`
}