Module: LLM::Ollama::ResponseParser

Defined in:
lib/llm/providers/ollama/response_parser.rb

Instance Method Summary collapse

Instance Method Details

#parse_completion(body) ⇒ Hash

Parameters:

  • body (Hash) —

    The response body from the LLM provider

Returns:

  • (Hash) 


9
10
11
12
13
14
15
16
 
# File 'lib/llm/providers/ollama/response_parser.rb', line 9

def parse_completion(body)
  {
    model: body["model"],
    choices: [LLM::Message.new(*body["message"].values_at("role", "content"), {completion: self})],
    prompt_tokens: body.dig("prompt_eval_count"),
    completion_tokens: body.dig("eval_count")
  }
end