@coze/taro-api/dist/event-source/index.tt.js

Official Coze Taro SDK for seamless AI integration into your applications | 扣子官方 Taro SDK，助您轻松集成 AI 能力到应用中

import { EventName } from './types';
import { BaseEventSource } from './base';
/**
 * Process streaming requests for ByteDance Mini Program
 */
export class EventSource extends BaseEventSource {
    options;
    task = null;
    isAborted = false;
    constructor(options) {
        super();
        this.options = options;
    }
    start() {
        const { url, method, headers, data, timeout } = this.options;
        this.task = tt.createEventSource({
            url,
            method,
            header: headers,
            data,
            timeout,
        });
        this.task.onOpen(() => {
            this.trigger(EventName.Open);
        });
        this.task.onMessage(msg => {
            this.trigger(EventName.Chunk, { data: msg });
        });
        this.task.onError(e => {
            const errMsg = e instanceof Error ? e.message : ((e && e.errMsg) ?? 'fail');
            if (this.isAborted) {
                return;
            }
            this.trigger(EventName.Fail, { errMsg });
        });
        this.task.onClose(() => {
            this.trigger(EventName.Success);
        });
    }
    abort() {
        if (!this.isAborted) {
            this.isAborted = true;
            this.task?.close();
            // fire "fail" manualy
            this.trigger(EventName.Fail, { errMsg: 'abort' });
        }
    }
}