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: about 1 month ago
JSON representation
An implementation of the OmniAI interface for OpenAI.
Host: GitHub
URL: https://github.com/ksylvest/omniai-openai
Owner: ksylvest
License: mit
Created: 2024-06-10T19:15:18.000Z (5 months ago)
Default Branch: main
Last Pushed: 2024-09-12T18:03:21.000Z (2 months ago)
Last Synced: 2024-09-27T06:22:16.185Z (about 2 months ago)
Topics: chatgpt, omniai, openai, ruby, whisper
Language: Ruby
Homepage: https://omniai-openai.ksylvest.com
Size: 136 KB
Stars: 2
Watchers: 2
Forks: 1
Open Issues: 1
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project

README

        # OmniAI::OpenAI

[![CircleCI](https://circleci.com/gh/ksylvest/omniai-openai.svg?style=svg)](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']`)

- `organization` (optional)

- `project` (optional)

- `host` (optional) useful for usage with Ollama or LocalAI

### 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)._

### 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 `gtp-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, ...]

```