Class: Legion::CLI::Chat::Tools::EntityExtract

Inherits:
RubyLLM::Tool
  • Object
show all
Defined in:
lib/legion/cli/chat/tools/entity_extract.rb

Instance Method Summary collapse

Instance Method Details

#execute(text:, entity_types: nil, min_confidence: 0.7) ⇒ Object 



25
26
27
28
29
30
31
 
# File 'lib/legion/cli/chat/tools/entity_extract.rb', line 25

def execute(text:, entity_types: nil, min_confidence: 0.7)
  return 'Apollo entity extractor not available.' unless extractor_available?

  types = parse_types(entity_types)
  result = run_extraction(text, types, min_confidence.to_f)
  format_result(result)
end