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

export class SparkAPIService extends OpenAIService implements OpenAILikeInterface {

    constructor(apiKey: string, apiSecret: string = '') {
        const key = apiSecret ? `${apiKey}:${apiSecret}` : apiKey;
        super({
            host: "https://spark-api-open.xf-yun.com/v1",
            apiKey: key
        })
    }
    chatComplete(params: ChatCompletionCreateParams): Promise<ChatCompletionStream<ChatCompletionChunk> | ChatCompletion> {
        return super.chatComplete(params)
    }

    embeddings(params: EmbeddingCreateParams): Promise<CreateEmbeddingResponse> {
        throw new Error("Method not implemented.");
    }
}