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 GeminiAPIService extends OpenAIService implements OpenAILikeInterface {

    constructor(apiKey: string) {
        super({
            host: "https://generativelanguage.googleapis.com/v1beta/openai/",
            apiKey: `${apiKey}`
        })
    }

    
    chatComplete(params: ChatCompletionCreateParams): Promise<ChatCompletionStream<ChatCompletionChunk> | ChatCompletion> {
        return super.chatComplete(params)
    }

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