import type * as Openai from "#Source/openai/index.ts"

import type { BaseEmbeddingOptions, EmbeddingOptions, EmbeddingResult } from "./embedding.ts"
import { BaseEmbedding } from "./embedding.ts"

export interface OpenaiNextEmbeddingOptions extends BaseEmbeddingOptions {
  openai: Openai.Openai
}
export class OpenaiNextEmbedding extends BaseEmbedding {
  protected readonly openai: Openai.Openai

  constructor(options: OpenaiNextEmbeddingOptions) {
    super(options)
    this.logger.setDefaultName("OpenaiNextEmbedding")

    this.openai = options.openai
  }

  async embedding(options: EmbeddingOptions): Promise<EmbeddingResult> {
    const result = await this.openai.customEmbedding(options)
    return result
  }
}