Skip to content

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

License

Notifications You must be signed in to change notification settings

zhengxs2018/erniebot-sdk-for-js

Repository files navigation

ERNIE Bot SDK for JavaScript

Typescriptcode style: prettiernpm packagenpm downloadsLicense

非官方 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

关联项目

参考

待办事项

  • 后端支持
  • 功能支持
    • ChatCompletion
    • Embedding
    • Images
    • Files
    • FineTuning
  • 运行时支持
    • NodeJS
    • Deno.js
    • 浏览器
    • 非标准 JS 环境,如:小程序

License

MIT

About

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

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published