Module: Legion::Extensions::Llm::Ledger::Runners::Prompts

Extended by:
Prompts
Includes:
Helpers::Lex
Included in:
Prompts
Defined in:
lib/legion/extensions/llm/ledger/runners/prompts.rb

Instance Method Summary collapse

Instance Method Details

#write_prompt_record(payload = nil, metadata = {}, **message) ⇒ Object 



14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
 
# File 'lib/legion/extensions/llm/ledger/runners/prompts.rb', line 14

def write_prompt_record(payload = nil, metadata = {}, **message)
  payload, metadata = normalize_runner_args(payload, metadata, message)
  headers = Helpers::SubscriptionMessage.extract_headers(payload, metadata)
  props   = metadata[:properties] || {}

  body = payload.is_a?(Hash) ? payload : Helpers::Decryption.decrypt_if_needed(payload, metadata)
  ctx  = body[:message_context] || {}

  expires_at = Helpers::Retention.resolve(
    retention:    headers['x-legion-retention'],
    contains_phi: headers['x-legion-contains-phi'] == 'true'
  )

  Writers::OfficialPromptWriter.write(official_prompt_payload(body, ctx, props, headers, expires_at))
rescue Helpers::DecryptionUnavailable => e
  handle_exception(e, level: :warn, handled: true, operation: 'write_prompt_record.decrypt')
  raise
rescue Helpers::DecryptionFailed => e
  handle_exception(e, level: :error, handled: true, operation: 'write_prompt_record.decrypt')
  raise
rescue StandardError => e
  handle_exception(e, level: :error, handled: true, operation: 'write_prompt_record')
  { result: :error, error: e.message }
end