ctod 0.2.0 → 0.3.1

package/lib/broker/35.ts

@@ -1,207 +0,0 @@
-import { flow } from 'power-helper'
-import { BaseBroker } from './index'
-import { Translator } from '../core/translator'
-import { Broker35Plugin } from '../core/plugin'
-import { ValidateCallback, ValidateCallbackOutputs } from '../utils/validate'
-import { ChatGPT35, ChatGPT35Message, ChatGPT35TalkResponse } from '../service/chatgpt35'
-
-export class ChatGPT35Broker<
-    S extends ValidateCallback<any>,
-    O extends ValidateCallback<any>,
-    P extends Broker35Plugin<any, any>,
-    PS extends Record<string, ReturnType<P['use']>>
-    > extends BaseBroker<ChatGPT35, S, O, P, PS, {
-
-        /**
-         * @zh 第一次聊天的時候觸發
-         * @en Triggered when chatting for the first time
-         */
-
-        talkFirst: {
-            id: string
-            data: ValidateCallbackOutputs<S>
-            plugins: {
-                [K in keyof PS]: {
-                    send: (data: PS[K]['__receiveData']) => void
-                }
-            }
-            messages: ChatGPT35Message[]
-            setPreMessages: (messages: ChatGPT35Message[]) => void
-            changeMessages: (messages: ChatGPT35Message[]) => void
-        }
-
-        /**
-         * @zh 發送聊天訊息給機器人前觸發
-         * @en Triggered before sending chat message to bot
-         */
-
-        talkBefore: {
-            id: string
-            data: ValidateCallbackOutputs<S>
-            messages: ChatGPT35Message[]
-            lastUserMessage: string
-        }
-
-        /**
-         * @zh 當聊天機器人回傳資料的時候觸發
-         * @en Triggered when the chatbot returns data
-         */
-
-        talkAfter: {
-            id: string
-            data: ValidateCallbackOutputs<S>
-            response: ChatGPT35TalkResponse
-            messages: ChatGPT35Message[]
-            parseText: string
-            lastUserMessage: string
-            changeParseText: (text: string) => void
-        }
-
-        /**
-         * @zh 當回傳資料符合規格時觸發
-         * @en Triggered when the returned data meets the specifications
-         */
-
-        succeeded: {
-            id: string
-            output: ValidateCallbackOutputs<O>
-        }
-    
-        /**
-         * @zh 當回傳資料不符合規格，或是解析錯誤時觸發
-         * @en Triggered when the returned data does not meet the specifications or parsing errors
-         */
-
-        parseFailed: {
-            id: string
-            error: any
-            retry: () => void
-            count: number
-            response: ChatGPT35TalkResponse
-            parserFails: { name: string, error: any }[]
-            messages: ChatGPT35Message[]
-            lastUserMessage: string
-            changeMessages: (messages: ChatGPT35Message[]) => void
-        }
-
-        /**
-         * @zh 不論成功失敗，執行結束的時候會執行。
-         * @en It will be executed when the execution is completed, regardless of success or failure.
-         */
-
-        done: {
-            id: string
-        }
-    }> {
-    bot: ChatGPT35 = new ChatGPT35()
-
-    /**
-     * @zh 將請求發出至聊天機器人。
-     * @en Send request to chatbot.
-     */
-
-    async request<T extends Translator<S, O>>(data: T['__schemeType']): Promise<T['__outputType']> {
-        this._install()
-        let id = flow.createUuid()
-        let output: any = null
-        let plugins = {} as any
-        let question = await this.translator.compile(data)
-        let messages: ChatGPT35Message[] = [
-            {
-                role: 'user',
-                content: question.prompt
-            }
-        ]
-        for (let key in this.plugins) {
-            plugins[key] = {
-                send: (data: any) => this.plugins[key].send({
-                    id,
-                    data
-                })
-            }
-        }
-        await this.hook.notify('talkFirst', {
-            id,
-            data,
-            plugins,
-            messages,
-            setPreMessages: ms => {
-                messages = [
-                    ...ms,
-                    {
-                        role: 'user',
-                        content: question.prompt
-                    }
-                ]
-            },
-            changeMessages: ms => {
-                messages = ms
-            }
-        })
-        await flow.asyncWhile(async ({ count, doBreak }) => {
-            if (count >= 10) {
-                return doBreak()
-            }
-            let response: ChatGPT35TalkResponse = null as any
-            let parseText = ''
-            let retryFlag = false
-            let lastUserMessage = messages.filter(e => e.role === 'user').slice(-1)[0]?.content || ''
-            try {
-                await this.hook.notify('talkBefore', {
-                    id,
-                    data,
-                    messages,
-                    lastUserMessage
-                })
-                response = await this.bot.talk(messages)
-                parseText = response.text
-                await this.hook.notify('talkAfter', {
-                    id,
-                    data,
-                    response,
-                    parseText,
-                    messages: response.newMessages,
-                    lastUserMessage,
-                    changeParseText: text => {
-                        parseText = text
-                    }
-                })
-                messages = response.newMessages
-                output = (await this.translator.parse(parseText)).output
-                await this.hook.notify('succeeded', {
-                    id,
-                    output
-                })
-                await this.hook.notify('done', { id })
-                doBreak()
-            } catch (error: any) {
-                // 如果解析錯誤，可以選擇是否重新解讀
-                if (error.isParserError) {
-                    await this.hook.notify('parseFailed', {
-                        id,
-                        error: error.error,
-                        count,
-                        response,
-                        messages,
-                        lastUserMessage,
-                        parserFails: error.parserFails,
-                        retry: () => {
-                            retryFlag = true
-                        },
-                        changeMessages: ms => {
-                            messages = ms
-                        }
-                    })
-                    if (retryFlag === false) {
-                        await this.hook.notify('done', { id })
-                        throw error
-                    }
-                } else {
-                    await this.hook.notify('done', { id })
-                    throw error
-                }
-            }
-        })
-        return output
-    }
-}

package/lib/broker/4.ts

@@ -1,207 +0,0 @@
-import { flow } from 'power-helper'
-import { BaseBroker } from './index'
-import { Translator } from '../core/translator'
-import { Broker4Plugin } from '../core/plugin'
-import { ValidateCallback, ValidateCallbackOutputs } from '../utils/validate'
-import { ChatGPT4, ChatGPT4Message, ChatGPT4TalkResponse } from '../service/chatgpt4'
-
-export class ChatGPT4Broker<
-    S extends ValidateCallback<any>,
-    O extends ValidateCallback<any>,
-    P extends Broker4Plugin<any, any>,
-    PS extends Record<string, ReturnType<P['use']>>
-    > extends BaseBroker<ChatGPT4, S, O, P, PS, {
-
-        /**
-         * @zh 第一次聊天的時候觸發
-         * @en Triggered when chatting for the first time
-         */
-
-        talkFirst: {
-            id: string
-            data: ValidateCallbackOutputs<S>
-            plugins: {
-                [K in keyof PS]: {
-                    send: (data: PS[K]['__receiveData']) => void
-                }
-            }
-            messages: ChatGPT4Message[]
-            setPreMessages: (messages: ChatGPT4Message[]) => void
-            changeMessages: (messages: ChatGPT4Message[]) => void
-        }
-
-        /**
-         * @zh 發送聊天訊息給機器人前觸發
-         * @en Triggered before sending chat message to bot
-         */
-
-        talkBefore: {
-            id: string
-            data: ValidateCallbackOutputs<S>
-            messages: ChatGPT4Message[]
-            lastUserMessage: string
-        }
-
-        /**
-         * @zh 當聊天機器人回傳資料的時候觸發
-         * @en Triggered when the chatbot returns data
-         */
-
-        talkAfter: {
-            id: string
-            data: ValidateCallbackOutputs<S>
-            response: ChatGPT4TalkResponse
-            messages: ChatGPT4Message[]
-            parseText: string
-            lastUserMessage: string
-            changeParseText: (text: string) => void
-        }
-
-        /**
-         * @zh 當回傳資料符合規格時觸發
-         * @en Triggered when the returned data meets the specifications
-         */
-
-        succeeded: {
-            id: string
-            output: ValidateCallbackOutputs<O>
-        }
-    
-        /**
-         * @zh 當回傳資料不符合規格，或是解析錯誤時觸發
-         * @en Triggered when the returned data does not meet the specifications or parsing errors
-         */
-
-        parseFailed: {
-            id: string
-            error: any
-            retry: () => void
-            count: number
-            response: ChatGPT4TalkResponse
-            parserFails: { name: string, error: any }[]
-            messages: ChatGPT4Message[]
-            lastUserMessage: string
-            changeMessages: (messages: ChatGPT4Message[]) => void
-        }
-
-        /**
-         * @zh 不論成功失敗，執行結束的時候會執行。
-         * @en It will be executed when the execution is completed, regardless of success or failure.
-         */
-
-        done: {
-            id: string
-        }
-    }> {
-    bot: ChatGPT4 = new ChatGPT4()
-
-    /**
-     * @zh 將請求發出至聊天機器人。
-     * @en Send request to chatbot.
-     */
-
-    async request<T extends Translator<S, O>>(data: T['__schemeType']): Promise<T['__outputType']> {
-        this._install()
-        let id = flow.createUuid()
-        let output: any = null
-        let plugins = {} as any
-        let question = await this.translator.compile(data)
-        let messages: ChatGPT4Message[] = [
-            {
-                role: 'user',
-                content: question.prompt
-            }
-        ]
-        for (let key in this.plugins) {
-            plugins[key] = {
-                send: (data: any) => this.plugins[key].send({
-                    id,
-                    data
-                })
-            }
-        }
-        await this.hook.notify('talkFirst', {
-            id,
-            data,
-            plugins,
-            messages,
-            setPreMessages: ms => {
-                messages = [
-                    ...ms,
-                    {
-                        role: 'user',
-                        content: question.prompt
-                    }
-                ]
-            },
-            changeMessages: ms => {
-                messages = ms
-            }
-        })
-        await flow.asyncWhile(async ({ count, doBreak }) => {
-            if (count >= 10) {
-                return doBreak()
-            }
-            let response: ChatGPT4TalkResponse = null as any
-            let parseText = ''
-            let retryFlag = false
-            let lastUserMessage = messages.filter(e => e.role === 'user').slice(-1)[0]?.content || ''
-            try {
-                await this.hook.notify('talkBefore', {
-                    id,
-                    data,
-                    messages,
-                    lastUserMessage
-                })
-                response = await this.bot.talk(messages)
-                parseText = response.text
-                await this.hook.notify('talkAfter', {
-                    id,
-                    data,
-                    response,
-                    parseText,
-                    messages: response.newMessages,
-                    lastUserMessage,
-                    changeParseText: text => {
-                        parseText = text
-                    }
-                })
-                messages = response.newMessages
-                output = (await this.translator.parse(parseText)).output
-                await this.hook.notify('succeeded', {
-                    id,
-                    output
-                })
-                await this.hook.notify('done', { id })
-                doBreak()
-            } catch (error: any) {
-                // 如果解析錯誤，可以選擇是否重新解讀
-                if (error.isParserError) {
-                    await this.hook.notify('parseFailed', {
-                        id,
-                        error: error.error,
-                        count,
-                        response,
-                        messages,
-                        lastUserMessage,
-                        parserFails: error.parserFails,
-                        retry: () => {
-                            retryFlag = true
-                        },
-                        changeMessages: ms => {
-                            messages = ms
-                        }
-                    })
-                    if (retryFlag === false) {
-                        await this.hook.notify('done', { id })
-                        throw error
-                    }
-                } else {
-                    await this.hook.notify('done', { id })
-                    throw error
-                }
-            }
-        })
-        return output
-    }
-}

package/lib/broker/index.ts

@@ -1,86 +0,0 @@
-import { Hook, Log } from 'power-helper'
-import { ChatGPT3 } from '../service/chatgpt3'
-import { ChatGPT35 } from '../service/chatgpt35'
-import { ChatGPT4 } from '../service/chatgpt4'
-import { TextParser } from '../core/parser'
-import { ValidateCallback } from '../utils/validate'
-import { Translator, TranslatorParams } from '../core/translator'
-import { Broker3Plugin, Broker35Plugin, Broker4Plugin } from '../core/plugin'
-
-export type Params<
-    B extends ChatGPT3 | ChatGPT35 | ChatGPT4,
-    S extends ValidateCallback<any>,
-    O extends ValidateCallback<any>,
-    C extends Record<string, any>,
-    P extends Broker3Plugin<any, any> | Broker35Plugin<any, any> | Broker4Plugin<any, any>,
-    PS extends Record<string, ReturnType<P['use']>>
-> = Omit<TranslatorParams<S, O>, 'parsers'> & {
-    name?: string
-    plugins?: PS | (() => PS)
-    install: (context: {
-        log: Log
-        bot: B
-        attach: Hook<C>['attach']
-        attachAfter: Hook<C>['attachAfter']
-        translator: Translator<S, O>
-    }) => void
-}
-
-export class BaseBroker<
-    B extends ChatGPT3 | ChatGPT35 | ChatGPT4,
-    S extends ValidateCallback<any>,
-    O extends ValidateCallback<any>,
-    P extends Broker3Plugin<any, any> | Broker35Plugin<any, any> | Broker4Plugin<any, any>,
-    PS extends Record<string, ReturnType<P['use']>>,
-    C extends Record<string, any>
-> {
-    protected __hookType!: C
-    protected log: Log
-    protected hook = new Hook<C>()
-    protected bot!: B
-    protected params: Params<B, S, O, C, P, PS>
-    protected plugins = {} as PS
-    protected installed = false
-    protected translator: Translator<S, O>
-
-    constructor(params: Params<B, S, O, C, P, PS>) {
-        this.log = new Log(params.name ?? 'no name')
-        this.params = params
-        this.translator = new Translator({
-            ...params,
-            parsers: [
-                TextParser.JsonMessage()
-            ]
-        })
-    }
-
-    protected _install(): any {
-        if (this.installed === false) {
-            this.installed = true
-            if (this.bot) {
-                const context = {
-                    bot: this.bot,
-                    log: this.log,
-                    attach: this.hook.attach.bind(this.hook),
-                    attachAfter: this.hook.attachAfter.bind(this.hook),
-                    translator: this.translator
-                }
-                this.params.install(context)
-                if (this.params.plugins) {
-                    this.plugins = typeof this.params.plugins === 'function' ? this.params.plugins() : this.params.plugins
-                    for (let key in this.plugins) {
-                        this.plugins[key].instance._params.onInstall({
-                            ...context,
-                            params: this.plugins[key].params,
-                            receive: this.plugins[key].receive
-                        })
-                    }
-                }
-            }
-        }
-    }
-
-    protected request(_data: any): any {
-        throw Error('DON\'T CALL THIS!')
-    }
-}

package/lib/service/chatgpt4.ts

@@ -1,137 +0,0 @@
-import axios, { AxiosInstance } from 'axios'
-import { json } from 'power-helper'
-import { PromiseResponseType } from '../types'
-
-export type ChatGPT4Message = {
-    role: 'system' | 'user' | 'assistant'
-    name?: string
-    content: string
-}
-
-type ApiResponse = {
-    id: string
-    object: string
-    created: number
-    choices: Array<{
-        index: number
-        finish_reason: string
-        message: {
-            role: 'system' | 'user' | 'assistant'
-            name?: string
-            content: string
-        }
-    }>
-    usage: {
-        prompt_tokens: number
-        completion_tokens: number
-        total_tokens: number
-    }
-}
-
-type Config = {
-    /**
-     * @zh 一次回應數量
-     * @en How many chat completion choices to generate for each input message.
-     */
-    n: number
-    /**
-     * @zh 選擇運行的模型，32k意味著能處理長度為 32768 的文本，而預設為 8192。
-     * @en How many chat completion choices to generate for each input message.
-     */
-    model: 'gpt-4' | 'gpt-4-32k'
-    /**
-     * @zh 冒險指數，數值由 0 ~ 2 之間。
-     * @en What sampling temperature to use, between 0 and 2. Higher values like 0.8 will make the output more random, while lower values like 0.2 will make it more focused and deterministic.
-     */
-    temperature: number
-}
-
-export class ChatGPT4 {
-    axios = axios.create()
-    apiKey = ''
-    config: Config = {
-        n: 1,
-        model: 'gpt-4',
-        temperature: 1
-    }
-
-    /**
-     * @zh 如果你有需要特別設定 axios，請使用這方法
-     * @en If you need to set axios, use this method
-     */
-
-    setAxios(axios: AxiosInstance) {
-        this.axios = axios
-    }
-
-    /**
-     * @zh 設定 api key
-     * @en Set api key
-     */
-
-    setConfiguration(apiKey: string) {
-        this.apiKey = apiKey
-    }
-
-    /**
-     * @zh 改變對話的一些設定
-     * @en Change some settings of the conversation
-     */
-
-    setConfig(options: Partial<Config>) {
-        Object.assign(this.config, options)
-    }
-
-    /**
-     * @zh 進行對話
-     * @en Talk to the AI
-     */
-
-    async talk(messages: ChatGPT4Message[] = []) {
-        const newMessages = json.jpjs(messages)
-        const result = await this.axios.post<ApiResponse>('https://api.openai.com/v1/chat/completions', {
-            model: this.config.model,
-            n: this.config.n,
-            messages: newMessages,
-            temperature: this.config.temperature
-        }, {
-            headers: {
-                'Content-Type': 'application/json',
-                'Authorization': `Bearer ${this.apiKey}`
-            }
-        })
-        const choices = result.data.choices || []
-        const message = choices[0]?.message || {
-            role: 'assistant',
-            content: ''
-        }
-        newMessages.push(message)
-        return {
-            id: result?.data.id as string,
-            text: message.content as string,
-            isDone: choices[0]?.finish_reason === 'stop',
-            newMessages,
-            apiReseponse: result.data
-        }
-    }
-
-    /**
-     * @zh 開啟持續性對話
-     */
-
-    async chat(prompt: string | string[], oldMessages: ChatGPT4Message[] = []) {
-        const result = await this.talk([
-            ...oldMessages,
-            {
-                role: 'user',
-                content: Array.isArray(prompt) ? prompt.join('\n') : prompt
-            }
-        ])
-        return {
-            result,
-            nextTalk: (prompt: string | string[]) => this.chat(prompt, result.newMessages)
-        }
-    }
-}
-
-export type ChatGPT4TalkResponse = PromiseResponseType<ChatGPT4['talk']>

package/types/lib/broker/3.d.ts

@@ -1,71 +0,0 @@
-import { Translator } from '../core/translator';
-import { BaseBroker } from './index';
-import { Broker3Plugin } from '../core/plugin';
-import { ChatGPT3, ChatGPT3TalkResponse } from '../service/chatgpt3';
-import { ValidateCallback, ValidateCallbackOutputs } from '../utils/validate';
-export declare class ChatGPT3Broker<S extends ValidateCallback<any>, O extends ValidateCallback<any>, P extends Broker3Plugin<any, any>, PS extends Record<string, ReturnType<P['use']>>> extends BaseBroker<ChatGPT3, S, O, P, PS, {
-    /**
-     * @zh 發送聊天訊息給機器人前觸發
-     * @en Triggered before sending chat message to bot
-     */
-    talkBefore: {
-        id: string;
-        data: ValidateCallbackOutputs<S>;
-        prompt: string;
-        plugins: {
-            [K in keyof PS]: {
-                send: (data: PS[K]['__receiveData']) => void;
-            };
-        };
-    };
-    /**
-     * @zh 當聊天機器人回傳資料的時候觸發
-     * @en Triggered when the chatbot returns data
-     */
-    talkAfter: {
-        id: string;
-        data: ValidateCallbackOutputs<S>;
-        prompt: string;
-        response: ChatGPT3TalkResponse;
-        parseText: string;
-        changeParseText: (text: string) => void;
-    };
-    /**
-     * @zh 當回傳資料符合規格時觸發
-     * @en Triggered when the returned data meets the specifications
-     */
-    succeeded: {
-        id: string;
-        output: ValidateCallbackOutputs<O>;
-    };
-    /**
-     * @zh 當回傳資料不符合規格，或是解析錯誤時觸發
-     * @en Triggered when the returned data does not meet the specifications or parsing errors
-     */
-    parseFailed: {
-        id: string;
-        error: any;
-        count: number;
-        retry: () => void;
-        response: ChatGPT3TalkResponse;
-        parserFails: {
-            name: string;
-            error: any;
-        }[];
-        changePrompt: (text: string) => void;
-    };
-    /**
-     * @zh 不論成功失敗，執行結束的時候會執行。
-     * @en It will be executed when the execution is completed, regardless of success or failure.
-     */
-    done: {
-        id: string;
-    };
-}> {
-    bot: ChatGPT3;
-    /**
-     * @zh 將請求發出至聊天機器人。
-     * @en Send request to chatbot.
-     */
-    request<T extends Translator<S, O>>(data: T['__schemeType']): Promise<T['__outputType']>;
-}