UNPKG

@llumiverse/drivers

Version:

LLM driver implementations. Currently supported are: openai, huggingface, bedrock, replicate.

33 lines (20 loc) 584 B
import { DriverOptions } from "@llumiverse/core"; import OpenAI from "openai"; import { BaseOpenAIDriver } from "./index.js"; export interface OpenAIDriverOptions extends DriverOptions { /** * The OpenAI api key */ apiKey?: string; //type with azure credentials } export class OpenAIDriver extends BaseOpenAIDriver { service: OpenAI; provider: "openai"; constructor(opts: OpenAIDriverOptions) { super(opts); this.service = new OpenAI({ apiKey: opts.apiKey }); this.provider = "openai"; } }