Skip to content

AI 基础配置

代码示例

使用 ChatGPT

ts
new AiEditor({
    element: "#aiEditor",
    ai: {
        models: {
            openai: {
                apiKey:"sk-alQ96zbDn*****"
            } as OpenaiModelConfig
        }
    },
})

或者星火大模型

typescript
new AiEditor({
    element: "#aiEditor",
    ai:{
        models:{
            spark:{
                appId:"****",
                apiKey:"****",
                apiSecret:"****",
                version:"v3.5" //支持 v2.1 v3.1 v3.5 v4.0
            }
        },
        bubblePanelEnable: true,
        bubblePanelModel: "auto",
        bubblePanelIcon: "<svg .....>",
        bubblePanelMenus: [],
        onCreateClientUrl: "...."
    },
})
  • models: 模型配置,支持 openai, spark, wenxin, custom,具体参考:这里
  • bubblePanelEnable: 是否开启 AI 弹窗功能
  • bubblePanelModel: AI 弹框使用的大语言模型名称
  • bubblePanelIcon: AI 弹框菜单的 自定义 ICON
  • bubblePanelMenus: 这个配置参考 这里
  • onCreateClientUrl: 自定义 URL 签名算法,一般情况下,如果编辑器涉及内容对外开放,则需要配置 onCreateURL,用于通过 server 端对 url 签名生成通信 url。
  • spark: 星火大模型配置,星火大模型支持配置的内容如下:
typescript
protocol?: string,
appId: string,
apiKey?: string,
apiSecret?: string,
version?: string,
  • protocol:通信协议,支持 ws 和 wss。
  • appId:应用 ID。
  • apiKey:api Key。
  • apiSecret:api 秘钥。
  • version: 版本,默认为 v3.5 支持 v2.1、 3.1、 3.5。

服务端签名

在使用 AiEditor 中,如果是用于内部用户,我们可以通过前端配置模型的 appIdapiKeyapiSecret ,这是没问题的。

但是,AiEditor 给普通互联网用户使用,则不应该把 appIdapiKeyapiSecret 配置在前端,而是通过后端服务对 Ai 请求路径进行签名, 需要自定义配置 onCreateClientUrl 才能保证秘钥安全。

示例代码如下:

typescript
new AiEditor({
    element: "#aiEditor",
    ai:{
        models:{
            spark:{
                appId: "****",
            }
        },
        onCreateClientUrl: (modelName, modelConfig, startFn)=>{
            //通过后端获取到 签名 url 后,执行 startFn 并传入 url。
            fetch("/your-path/getUrl?appId="+modelConfig.appId)
                .then(resp=>resp.json)
                .then(json=> {
                    startFn(json.url)
                })
        },
        onTokenConsume(_, __, count) {
            axios.post("/api/v1/resource/doTokenCounting", {
                count
            })
        }
    },
})

记录 Token 消耗情况

示例代码如下:

typescript
new AiEditor({
    element: "#aiEditor",
    ai: {
        models: {
            spark: {
                appId: "****",
            }
        },
        onTokenConsume(modelName, modelConfig, count) {
            //当有 token 消耗时,会回调此方法
            axios.post("/api/v1/resource/doTokenCounting", {
                count
            })
        }
    },
})