zhengxs2018 / erniebot-sdk-for-js

非官方 JS-SDK，可以调用文心大模型的能力，包含文本创作、通用对话、语义向量、AI作图等。

ERNIE Bot SDK for JavaScript

非官方 JS-SDK，请勿在生产中使用

可以调用文心大模型的能力，包含文本创作、通用对话、语义向量、AI作图等。

注意： 后续兼容 openai，可能会调整 API 的输出格式。

安装

# With NPM
$ npm i -S @zhengxs/erniebot

# With Yarn
$ yarn add @zhengxs/erniebot

# With PNPM
$ pnpm add @zhengxs/erniebot

使用

AI Studio

默认是 AI Studio 后端。

import ERNIEBot from '@zhengxs/erniebot'

const erniebot = new ERNIEBot({
  apiType: 'aistudio',
  token: 'My API Access Token', // defaults to process.env["EB_ACCESS_TOKEN"]
})

async function main() {
  const chatCompletion = await erniebot.chat.completions.create({
    model: 'ernie-bot',
    messages: [{ role: 'user', content: 'Say this is a test' }],
  })

  console.log(chatCompletion.result)
}

main()

文心千帆

可以切换为 文心千帆 后端。

import ERNIEBot from '@zhengxs/erniebot'

const erniebot = new ERNIEBot({
  apiType: 'qianfan',
  ak: 'My APP Access Token', // defaults to process.env["EB_AK"]
  sk: 'My APP Secret Token', // defaults to process.env["EB_SK"]
})

async function main() {
  const chatCompletion = await erniebot.chat.completions.create({
    model: 'ernie-bot',
    messages: [{ role: 'user', content: 'Say this is a test' }],
  })

  console.log(chatCompletion.result)
}

main()

CLI

使用全局安装

# With NPM
$ npm i -g @zhengxs/erniebot-cli

# With Yarn
$ yarn global @zhengxs/erniebot-cli

# With PNPM
$ pnpm add --global @zhengxs/erniebot-cli

使用 erniebot 命令运行。

$ erniebot

关联项目

• 基于 wechaty 的聊天助手插件

参考

• openai-node
• ERNIE-Bot-SDK

待办事项

• 后端支持
  • AI Studio
  • 文心千帆
  • 文心一言
• 功能支持
  • ChatCompletion
  • Embedding
  • Images
  • Files
  • FineTuning
• 运行时支持
  • NodeJS
  • Deno.js
  • 浏览器
  • 非标准 JS 环境，如：小程序

License

MIT