import { ChatCompletion, ChatCompletionChunk, ChatCompletionCreateParams, CreateEmbeddingResponse, EmbeddingCreateParams } from "openai/resources";
import { OpenAILikeInterface } from "../interfaces/OpenAILike";
import { OpenAIService } from "./openai";
import { ChatCompletionStream } from "../utils/stream";

export class DouBaoAPIService extends OpenAIService implements OpenAILikeInterface {

    constructor(apiKey: string, isBot?: boolean) {
        super({
            host: isBot ? "https://ark.cn-beijing.volces.com/api/v3/bots" : "https://ark.cn-beijing.volces.com/api/v3",
            apiKey: `${apiKey}`
        })
    }
    chatComplete(params: ChatCompletionCreateParams): Promise<ChatCompletionStream<ChatCompletionChunk> | ChatCompletion> {
        return super.chatComplete(params)
    }

    embeddings(params: EmbeddingCreateParams): Promise<CreateEmbeddingResponse> {
        return super.embeddings(params)
    }
}