Class: Rack::AI::Features::Classification

Inherits:
Object
  • Object
show all
Defined in:
lib/rack/ai/features/classification.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(provider, config) ⇒ Classification 



9
10
11
12
13
 
# File 'lib/rack/ai/features/classification.rb', line 9

def initialize(provider, config)
  @name = :classification
  @provider = provider
  @config = config
end

Instance Attribute Details

#config ⇒ Object (readonly)

Returns the value of attribute config.



7
8
9
 
# File 'lib/rack/ai/features/classification.rb', line 7

def config
  @config
end

#name ⇒ Object (readonly)

Returns the value of attribute name.



7
8
9
 
# File 'lib/rack/ai/features/classification.rb', line 7

def name
  @name
end

#provider ⇒ Object (readonly)

Returns the value of attribute provider.



7
8
9
 
# File 'lib/rack/ai/features/classification.rb', line 7

def provider
  @provider
end

Instance Method Details

#enabled? ⇒ Boolean 



15
16
17
 
# File 'lib/rack/ai/features/classification.rb', line 15

def enabled?
  @config.feature_enabled?(:classification)
end

#process_request(env) ⇒ Object 



23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
 
# File 'lib/rack/ai/features/classification.rb', line 23

def process_request(env)
  request_data = @provider.build_request_data(env)
  result = @provider.classify_request(request_data)
  
  # Apply confidence threshold
  threshold = @config.classification.confidence_threshold
  if result[:confidence] < threshold
    result[:classification] = :uncertain
    result[:action] = :allow
  else
    result[:action] = determine_action(result[:classification])
  end

  # Add metadata
  result[:feature] = @name
  result[:timestamp] = Time.now.iso8601
  result[:request_id] = generate_request_id(env)

  result
end

#process_response? ⇒ Boolean 



19
20
21
 
# File 'lib/rack/ai/features/classification.rb', line 19

def process_response?
  false
end