Class: LLM::Ollama::Format::CompletionFormat

Inherits:

Object

• Object
• LLM::Ollama::Format::CompletionFormat

Defined in:

lib/llm/shell/internal/llm.rb/lib/llm/providers/ollama/format/completion_format.rb

Instance Method Summary

• #format ⇒ Hash

  Returns the message for the Ollama chat completions API.

• #initialize(message) ⇒ CompletionFormat constructor

  A new instance of CompletionFormat.

Constructor Details

#initialize(message) ⇒ CompletionFormat

Returns a new instance of CompletionFormat.

Parameters:

• message (LLM::Message) —

  The message to format

# File 'lib/llm/shell/internal/llm.rb/lib/llm/providers/ollama/format/completion_format.rb', line 10

def initialize(message)
  @message = message
end

Instance Method Details

#format ⇒ Hash

Returns the message for the Ollama chat completions API

Returns:

• (Hash)

# File 'lib/llm/shell/internal/llm.rb/lib/llm/providers/ollama/format/completion_format.rb', line 17

def format
  catch(:abort) do
    if Hash === message
      {role: message[:role]}.merge(format_content(message[:content]))
    else
      format_message
    end
  end
end