Class: Roseflow::OpenAI::Config

Inherits:

Anyway::Config

Object
Anyway::Config
Roseflow::OpenAI::Config

show all

Defined in:: lib/roseflow/openai/config.rb

Overview

Configuration class for the OpenAI provider.

Constant Summary collapse

OPENAI_API_URL =

"https://api.openai.com"

CHAT_MODELS =

%w(gpt-4 gpt-4-0314 gpt-4-32k gpt-4-32k-0314 gpt-3.5-turbo gpt-3.5-turbo-0301).freeze

COMPLETION_MODELS =

%w(text-davinci-003 text-davinci-002 text-curie-001 text-babbage-001 text-ada-001 davinci curie babbage ada).freeze

EDIT_MODELS =

%w(text-davinci-edit-001 code-davinci-edit-001).freeze

TRANSCRIPTION_MODELS =

%w(whisper-1).freeze

TRANSLATION_MODELS =

%w(whisper-1).freeze

FINE_TUNE_MODELS =

%w(davinci curie babbage ada).freeze

EMBEDDING_MODELS =

%w(text-embedding-ada-002 text-search-ada-doc-001).freeze

MODERATION_MODELS =

%w(text-moderation-stable text-moderation-latest).freeze

MAX_TOKENS =

{
  "gpt-4": 8192,
  "gpt-4-0314": 8192,
  "gpt-4-32k": 32_768,
  "gpt-4-32k-0314": 32_768,
  "gpt-3.5-turbo": 4096,
  "gpt-3.5-turbo-0301": 4096,
  "text-davinci-003": 4097,
  "text-davinci-002": 4097,
  "code-davinci-002": 8001
}