Class: LlmClassifier::Classifier

Inherits:
Object
  • Object
show all
Defined in:
lib/llm_classifier/classifier.rb

Overview

Base classifier class that provides a DSL for defining LLM-powered classifiers

Class Attribute Summary collapse

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(input, **options) ⇒ Classifier

Returns a new instance of Classifier.



87
88
89
90
 
# File 'lib/llm_classifier/classifier.rb', line 87

def initialize(input, **options)
  @input = input
  @options = options
end

Class Attribute Details

.after_classify_callbacks ⇒ Object (readonly)

Returns the value of attribute after_classify_callbacks.



9
10
11
 
# File 'lib/llm_classifier/classifier.rb', line 9

def after_classify_callbacks
  @after_classify_callbacks
end

.before_classify_callbacks ⇒ Object (readonly)

Returns the value of attribute before_classify_callbacks.



9
10
11
 
# File 'lib/llm_classifier/classifier.rb', line 9

def before_classify_callbacks
  @before_classify_callbacks
end

.defined_adapter ⇒ Object (readonly)

Returns the value of attribute defined_adapter.



9
10
11
 
# File 'lib/llm_classifier/classifier.rb', line 9

def defined_adapter
  @defined_adapter
end

.defined_categories ⇒ Object (readonly)

Returns the value of attribute defined_categories.



9
10
11
 
# File 'lib/llm_classifier/classifier.rb', line 9

def defined_categories
  @defined_categories
end

.defined_knowledge ⇒ Object (readonly)

Returns the value of attribute defined_knowledge.



9
10
11
 
# File 'lib/llm_classifier/classifier.rb', line 9

def defined_knowledge
  @defined_knowledge
end

.defined_model ⇒ Object (readonly)

Returns the value of attribute defined_model.



9
10
11
 
# File 'lib/llm_classifier/classifier.rb', line 9

def defined_model
  @defined_model
end

.defined_multi_label ⇒ Object (readonly)

Returns the value of attribute defined_multi_label.



9
10
11
 
# File 'lib/llm_classifier/classifier.rb', line 9

def defined_multi_label
  @defined_multi_label
end

.defined_require_categories ⇒ Object (readonly)

Returns the value of attribute defined_require_categories.



9
10
11
 
# File 'lib/llm_classifier/classifier.rb', line 9

def defined_require_categories
  @defined_require_categories
end

.defined_system_prompt ⇒ Object (readonly)

Returns the value of attribute defined_system_prompt.



9
10
11
 
# File 'lib/llm_classifier/classifier.rb', line 9

def defined_system_prompt
  @defined_system_prompt
end

Instance Attribute Details

#input ⇒ Object (readonly)

Returns the value of attribute input.



85
86
87
 
# File 'lib/llm_classifier/classifier.rb', line 85

def input
  @input
end

#options ⇒ Object (readonly)

Returns the value of attribute options.



85
86
87
 
# File 'lib/llm_classifier/classifier.rb', line 85

def options
  @options
end

Class Method Details

.adapter(adapter_name = nil) ⇒ Object 



38
39
40
41
42
43
44
 
# File 'lib/llm_classifier/classifier.rb', line 38

def adapter(adapter_name = nil)
  if adapter_name.nil?
    @defined_adapter || LlmClassifier.configuration.adapter
  else
    @defined_adapter = adapter_name
  end
end

.after_classify(&block) ⇒ Object 



75
76
77
78
 
# File 'lib/llm_classifier/classifier.rb', line 75

def after_classify(&block)
  @after_classify_callbacks ||= []
  @after_classify_callbacks << block
end

.before_classify(&block) ⇒ Object 



70
71
72
73
 
# File 'lib/llm_classifier/classifier.rb', line 70

def before_classify(&block)
  @before_classify_callbacks ||= []
  @before_classify_callbacks << block
end

.categories(*cats) ⇒ Object 



14
15
16
17
18
19
20
 
# File 'lib/llm_classifier/classifier.rb', line 14

def categories(*cats)
  if cats.empty?
    @defined_categories || []
  else
    @defined_categories = cats.map(&:to_s)
  end
end

.classify(input) ⇒ Object 



80
81
82
 
# File 'lib/llm_classifier/classifier.rb', line 80

def classify(input, **)
  new(input, **).classify
end

.knowledge ⇒ Object 



62
63
64
65
66
67
68
 
# File 'lib/llm_classifier/classifier.rb', line 62

def knowledge(&)
  if block_given?
    @defined_knowledge = Knowledge.new
    @defined_knowledge.instance_eval(&)
  end
  @defined_knowledge
end

.model(model_name = nil) ⇒ Object 



30
31
32
33
34
35
36
 
# File 'lib/llm_classifier/classifier.rb', line 30

def model(model_name = nil)
  if model_name.nil?
    @defined_model || LlmClassifier.configuration.default_model
  else
    @defined_model = model_name
  end
end

.multi_label(value = nil) ⇒ Object 



46
47
48
49
50
51
52
 
# File 'lib/llm_classifier/classifier.rb', line 46

def multi_label(value = nil)
  if value.nil?
    @defined_multi_label || false
  else
    @defined_multi_label = value
  end
end

.require_categories(value = nil) ⇒ Object 



54
55
56
57
58
59
60
 
# File 'lib/llm_classifier/classifier.rb', line 54

def require_categories(value = nil)
  if value.nil?
    @defined_require_categories || false
  else
    @defined_require_categories = value
  end
end

.system_prompt(prompt = nil) ⇒ Object 



22
23
24
25
26
27
28
 
# File 'lib/llm_classifier/classifier.rb', line 22

def system_prompt(prompt = nil)
  if prompt.nil?
    @defined_system_prompt
  else
    @defined_system_prompt = prompt
  end
end

Instance Method Details

#classify ⇒ Object 



92
93
94
95
96
97
98
99
 
# File 'lib/llm_classifier/classifier.rb', line 92

def classify
  processed_input = run_before_callbacks(@input)
  result = perform_classification(processed_input)
  run_after_callbacks(result)
  result
rescue StandardError => e
  Result.failure(error: e.message)
end