Class: LLM::Gemini::ResponseParser::CompletionParser

Inherits:
Object
  • Object
show all
Defined in:
lib/llm/providers/gemini/response_parser/completion_parser.rb

Instance Method Summary collapse

Constructor Details

#initialize(body) ⇒ CompletionParser

Returns a new instance of CompletionParser.



5
6
7
 
# File 'lib/llm/providers/gemini/response_parser/completion_parser.rb', line 5

def initialize(body)
  @body = OpenStruct.from_hash(body)
end

Instance Method Details

#format(response) ⇒ Object 



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

def format(response)
  {
    model:,
    prompt_tokens:,
    completion_tokens:,
    total_tokens:,
    choices: format_choices(response)
  }
end