Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/ksylvest/omniai-openai

An implementation of the OmniAI interface for OpenAI.
https://github.com/ksylvest/omniai-openai

chatgpt omniai openai ruby whisper

Last synced: 17 days ago
JSON representation

An implementation of the OmniAI interface for OpenAI.

Awesome Lists containing this project

README

        

# OmniAI::OpenAI

[![LICENSE](https://img.shields.io/badge/license-MIT-blue.svg)](https://github.com/ksylvest/omniai-openai/blob/main/LICENSE)
[![RubyGems](https://img.shields.io/gem/v/omniai-openai)](https://rubygems.org/gems/omniai-openai)
[![GitHub](https://img.shields.io/badge/github-repo-blue.svg)](https://github.com/ksylvest/omniai-openai)
[![Yard](https://img.shields.io/badge/docs-site-blue.svg)](https://omniai-openai.ksylvest.com)
[![CircleCI](https://img.shields.io/circleci/build/github/ksylvest/omniai-openai)](https://circleci.com/gh/ksylvest/omniai-openai)

An OpenAI implementation of the [OmniAI](https://github.com/ksylvest/omniai) interface supporting ChatGPT, Whisper, Text-to-Voice, Voice-to-Text, and more. This library is community maintained.

## Installation

```sh
gem install omniai-openai
```

## Usage

### Client

A client is setup as follows if `ENV['OPENAI_API_KEY']` exists:

```ruby
client = OmniAI::OpenAI::Client.new
```

A client may also be passed the following options:

- `api_key` (required - default is `ENV['OPENAI_API_KEY']`)
- `api_prefix` (optional) - used with a host when necessary
- `organization` (optional)
- `project` (optional)
- `host` (optional) useful for usage with Ollama, LocalAI or other OpenAI API compatible services

### Configuration

Global configuration is supported for the following options:

```ruby
OmniAI::OpenAI.configure do |config|
config.api_key = 'sk-...' # default: ENV['OPENAI_API_KEY']
config.organization = '...' # default: ENV['OPENAI_ORGANIZATION']
config.project = '...' # default: ENV['OPENAI_PROJECT']
config.host = '...' # default: 'https://api.openai.com' - override for usage with LocalAI / Ollama
end
```

#### Usage with [LocalAI](https://localai.io/)

LocalAI offers [built in compatability with the OpenAI specification](https://localai.io/). To initialize a client that points to a Ollama change the host accordingly:

```ruby
client = OmniAI::OpenAI::Client.new(host: 'http://localhost:8080', api_key: nil)
```

_For details on installation or running LocalAI see the [getting started tutorial](https://localai.io/basics/getting_started/)._

#### Usage with [Ollama](https://www.ollama.com/)

Ollama offers [built in compatability with the OpenAI specification](https://ollama.com/blog/openai-compatibility). To initialize a client that points to a Ollama change the host accordingly:

```ruby
client = OmniAI::OpenAI::Client.new(host: 'http://localhost:11434', api_key: nil)
```

_For details on installation or running Ollama checkout [the project README](https://github.com/ollama/ollama)._

#### Usage with [OpenRouter](https://open_router.ai/)

Other fee-based systems/services have adopted all or some of the OpenAI API. For example [open_router.ai](https://open_router.ai) is a web-services that provides access to many models and providers using their own as well as an OpenAI API.

```ruby
client = OmniAI::OpenAI::Client.new(
host: 'https://open_router.ai',
api_key: ENV['OPENROUTER_API_KEY'],
api_prefix: '/api')
```

### Chat

A chat completion is generated by passing in a simple text prompt:

```ruby
completion = client.chat('Tell me a joke!')
completion.content # 'Why did the chicken cross the road? To get to the other side.'
```

A chat completion may also be generated by using a prompt builder:

```ruby
completion = client.chat do |prompt|
prompt.system('Your are an expert in geography.')
prompt.user('What is the capital of Canada?')
end
completion.content # 'The capital of Canada is Ottawa.'
```

#### Model

`model` takes an optional string (default is `gpt-4o`):

```ruby
completion = client.chat('How fast is a cheetah?', model: OmniAI::OpenAI::Chat::Model::GPT_3_5_TURBO)
completion.content # 'A cheetah can reach speeds over 100 km/h.'
```

[OpenAI API Reference `model`](https://platform.openai.com/docs/api-reference/chat/create#chat-create-model)

#### Temperature

`temperature` takes an optional float between `0.0` and `2.0` (defaults is `0.7`):

```ruby
completion = client.chat('Pick a number between 1 and 5', temperature: 2.0)
completion.content # '3'
```

[OpenAI API Reference `temperature`](https://platform.openai.com/docs/api-reference/chat/create#chat-create-temperature)

#### Stream

`stream` takes an optional a proc to stream responses in real-time chunks instead of waiting for a complete response:

```ruby
stream = proc do |chunk|
print(chunk.content) # 'Better', 'three', 'hours', ...
end
client.chat('Be poetic.', stream:)
```

[OpenAI API Reference `stream`](https://platform.openai.com/docs/api-reference/chat/create#chat-create-stream)

#### Format

`format` takes an optional symbol (`:json`) and that setes the `response_format` to `json_object`:

```ruby
completion = client.chat(format: :json) do |prompt|
prompt.system(OmniAI::Chat::JSON_PROMPT)
prompt.user('What is the name of the drummer for the Beatles?')
end
JSON.parse(completion.content) # { "name": "Ringo" }
```

[OpenAI API Reference `response_format`](https://platform.openai.com/docs/api-reference/chat/create#chat-create-stream)

> When using JSON mode, you must also instruct the model to produce JSON yourself via a system or user message.

### Transcribe

A transcription is generated by passing in a path to a file:

```ruby
transcription = client.transcribe(file.path)
transcription.text # '...'
```

#### Prompt

`prompt` is optional and can provide additional context for transcribing:

```ruby
transcription = client.transcribe(file.path, prompt: '')
transcription.text # '...'
```

[OpenAI API Reference `prompt`](https://platform.openai.com/docs/api-reference/audio/createTranscription#audio-createtranscription-prompt)

#### Format

`format` is optional and supports `json`, `text`, `srt` or `vtt`:

```ruby
transcription = client.transcribe(file.path, format: OmniAI::Transcribe::Format::TEXT)
transcription.text # '...'
```

[OpenAI API Reference `response_format`](https://platform.openai.com/docs/api-reference/audio/createTranscription#audio-createtranscription-response_format)

#### Language

`language` is optional and may improve accuracy and latency:

```ruby
transcription = client.transcribe(file.path, language: OmniAI::Transcribe::Language::SPANISH)
transcription.text
```

[OpenAI API Reference `language`](https://platform.openai.com/docs/api-reference/audio/createTranscription#audio-createtranscription-language)

#### Temperature

`temperature` is optional and must be between 0.0 (more deterministic) and 1.0 (less deterministic):

```ruby
transcription = client.transcribe(file.path, temperature: 0.2)
transcription.text
```

[OpenAI API Reference `temperature`](https://platform.openai.com/docs/api-reference/audio/createTranscription#audio-createtranscription-temperature)

### Speak

Speech can be generated by passing text with a block:

```ruby
File.open('example.ogg', 'wb') do |file|
client.speak('How can a clam cram in a clean cream can?') do |chunk|
file << chunk
end
end
```

If a block is not provided then a tempfile is returned:

```ruby
tempfile = client.speak('Can you can a can as a canner can can a can?')
tempfile.close
tempfile.unlink
```

#### Voice

`voice` is optional and must be one of the supported voices:

```ruby
client.speak('She sells seashells by the seashore.', voice: OmniAI::OpenAI::Speak::Voice::SHIMMER)
```

[OpenAI API Reference `voice`](https://platform.openai.com/docs/api-reference/audio/createSpeech#audio-createspeech-voice)

#### Model

`model` is optional and must be either `tts-1` or `tts-1-hd` (default):

```ruby
client.speak('I saw a kitten eating chicken in the kitchen.', format: OmniAI::OpenAI::Speak::Model::TTS_1)
```

[OpenAI API Refernce `model`](https://platform.openai.com/docs/api-reference/audio/createSpeech#audio-createspeech-model)

#### Speed

`speed` is optional and must be between 0.25 and 0.40:

```ruby
client.speak('How much wood would a woodchuck chuck if a woodchuck could chuck wood?', speed: 4.0)
```

[OmniAI API Reference `speed`](https://platform.openai.com/docs/api-reference/audio/createSpeech#audio-createspeech-speed)

#### Format

`format` is optional and supports `MP3` (default), `OPUS`, `AAC`, `FLAC`, `WAV` or `PCM`:

```ruby
client.speak('A pessemistic pest exists amidst us.', format: OmniAI::OpenAI::Speak::Format::FLAC)
```

[OpenAI API Reference `format`](https://platform.openai.com/docs/api-reference/audio/createSpeech#audio-createspeech-response_format)

## Files

### Finding an File

```ruby
client.files.find(id: 'file_...')
```

### Listing all Files

```ruby
client.files.all
```

### Uploading a File

#### Using a File

```ruby
file = client.files.build(io: File.open('demo.pdf', 'wb'))
file.save!
```

#### Using a Path

```ruby
file = client.files.build(io: 'demo.pdf'))
file.save!
```

### Downloading a File

```ruby
file = client.files.find(id: 'file_...')
File.open('...', 'wb') do |file|
file.content do |chunk|
file << chunk
end
end
```

### Destroying a File

```ruby
client.files.destroy!('file_...')
```

## Assistants

### Finding an Assistant

```ruby
client.assistants.find(id: 'asst_...')
```

### Listing all Assistants

```ruby
client.assistants.all
```

### Creating an Assistant

```ruby
assistant = client.assistants.build
assistant.name = 'Ringo'
assistant.model = OmniAI::OpenAI::Chat::Model::GPT_4
assistant.description = 'The drummer for the Beatles.'
assistant.save!
```

### Updating an Assistant

```ruby
assistant = client.assistants.find(id: 'asst_...')
assistant.name = 'George'
assistant.model = OmniAI::OpenAI::Chat::Model::GPT_4
assistant.description = 'A guitarist for the Beatles.'
assistant.save!
```

### Destroying an Assistant

```ruby
client.assistants.destroy!('asst_...')
```

## Threads

### Finding a Thread

```ruby
client.threads.find(id: 'thread_...')
```

### Creating a Thread

```ruby
thread = client.threads.build
thread.metadata = { user: 'Ringo' }
thread.save!
```

### Updating a Thread

```ruby
thread = client.threads.find(id: 'thread_...')
thread.metadata = { user: 'Ringo' }
thread.save!
```

### Destroying a Threads

```ruby
client.threads.destroy!('thread_...')
```

### Messages

#### Finding a Message

```ruby
thread = client.threads.find(id: 'thread_...')
message = thread.messages.find(id: 'msg_...')
message.save!
```

#### Listing all Messages

```ruby
thread = client.threads.find(id: 'thread_...')
thread.messages.all
```

#### Creating a Message

```ruby
thread = client.threads.find(id: 'thread_...')
message = thread.messages.build(role: 'user', content: 'Hello?')
message.save!
```

#### Updating a Message

```ruby
thread = client.threads.find(id: 'thread_...')
message = thread.messages.build(role: 'user', content: 'Hello?')
message.save!
```

### Runs

#### Finding a Run

```ruby
thread = client.threads.find(id: 'thread_...')
run = thread.runs.find(id: 'run_...')
run.save!
```

#### Listing all Runs

```ruby
thread = client.threads.find(id: 'thread_...')
thread.runs.all
```

#### Creating a Run

```ruby
run = client.runs.find(id: 'thread_...')
run = thread.runs.build
run.metadata = { user: 'Ringo' }
run.save!
```

#### Updating a Run

```ruby
thread = client.threads.find(id: 'thread_...')
run = thread.messages.find(id: 'run_...')
run.metadata = { user: 'Ringo' }
run.save!
```

#### Polling a Run

```ruby
run.terminated? # false
run.poll!
run.terminated? # true
run.status # 'cancelled' / 'failed' / 'completed' / 'expired'
```

#### Cancelling a Run

```ruby
thread = client.threads.find(id: 'thread_...')
run = thread.runs.cancel!(id: 'run_...')
```

### Embed

Text can be converted into a vector embedding for similarity comparison usage via:

```ruby
response = client.embed('The quick brown fox jumps over a lazy dog.')
response.embedding # [0.0, ...]
```