guardrails-ruby 0.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +7 -0
- data/CLAUDE.md +507 -0
- data/Gemfile +2 -0
- data/LICENSE +21 -0
- data/README.md +243 -0
- data/Rakefile +9 -0
- data/examples/basic.rb +64 -0
- data/examples/custom_check.rb +103 -0
- data/examples/rails_controller.rb +73 -0
- data/guardrails-ruby.gemspec +30 -0
- data/lib/guardrails_ruby/check.rb +64 -0
- data/lib/guardrails_ruby/checks/competitor_mention.rb +36 -0
- data/lib/guardrails_ruby/checks/encoding.rb +33 -0
- data/lib/guardrails_ruby/checks/format.rb +35 -0
- data/lib/guardrails_ruby/checks/hallucinated_emails.rb +30 -0
- data/lib/guardrails_ruby/checks/hallucinated_urls.rb +38 -0
- data/lib/guardrails_ruby/checks/keyword_filter.rb +33 -0
- data/lib/guardrails_ruby/checks/max_length.rb +30 -0
- data/lib/guardrails_ruby/checks/pii.rb +54 -0
- data/lib/guardrails_ruby/checks/prompt_injection.rb +36 -0
- data/lib/guardrails_ruby/checks/relevance.rb +43 -0
- data/lib/guardrails_ruby/checks/topic.rb +25 -0
- data/lib/guardrails_ruby/checks/toxic_language.rb +28 -0
- data/lib/guardrails_ruby/configuration.rb +15 -0
- data/lib/guardrails_ruby/guard.rb +129 -0
- data/lib/guardrails_ruby/middleware.rb +30 -0
- data/lib/guardrails_ruby/rails/controller.rb +57 -0
- data/lib/guardrails_ruby/rails/railtie.rb +20 -0
- data/lib/guardrails_ruby/redactors/keyword_redactor.rb +33 -0
- data/lib/guardrails_ruby/redactors/pii_redactor.rb +59 -0
- data/lib/guardrails_ruby/result.rb +53 -0
- data/lib/guardrails_ruby/version.rb +5 -0
- data/lib/guardrails_ruby/violation.rb +41 -0
- data/lib/guardrails_ruby.rb +38 -0
- metadata +115 -0
|
@@ -0,0 +1,53 @@
|
|
|
1
|
+
# frozen_string_literal: true
|
|
2
|
+
|
|
3
|
+
module GuardrailsRuby
|
|
4
|
+
class Result
|
|
5
|
+
attr_reader :violations, :original_text
|
|
6
|
+
|
|
7
|
+
def initialize(original_text: nil, violations: [])
|
|
8
|
+
@original_text = original_text
|
|
9
|
+
@violations = violations
|
|
10
|
+
end
|
|
11
|
+
|
|
12
|
+
def passed?
|
|
13
|
+
@violations.empty?
|
|
14
|
+
end
|
|
15
|
+
|
|
16
|
+
def sanitized
|
|
17
|
+
redact_violation = @violations.find { |v| v.action == :redact && v.sanitized }
|
|
18
|
+
redact_violation ? redact_violation.sanitized : @original_text
|
|
19
|
+
end
|
|
20
|
+
|
|
21
|
+
def merge(other)
|
|
22
|
+
merged_violations = @violations + other.violations
|
|
23
|
+
|
|
24
|
+
# Use the most recent sanitized text as the base
|
|
25
|
+
merged_text = other.original_text || @original_text
|
|
26
|
+
|
|
27
|
+
self.class.new(
|
|
28
|
+
original_text: merged_text,
|
|
29
|
+
violations: merged_violations
|
|
30
|
+
)
|
|
31
|
+
end
|
|
32
|
+
|
|
33
|
+
def blocked?
|
|
34
|
+
@violations.any? { |v| v.action == :block }
|
|
35
|
+
end
|
|
36
|
+
|
|
37
|
+
def warnings
|
|
38
|
+
@violations.select { |v| v.action == :warn }
|
|
39
|
+
end
|
|
40
|
+
|
|
41
|
+
def redactions
|
|
42
|
+
@violations.select { |v| v.action == :redact }
|
|
43
|
+
end
|
|
44
|
+
|
|
45
|
+
def to_s
|
|
46
|
+
if passed?
|
|
47
|
+
"Result: passed"
|
|
48
|
+
else
|
|
49
|
+
"Result: #{violations.size} violation(s) - #{violations.map(&:to_s).join('; ')}"
|
|
50
|
+
end
|
|
51
|
+
end
|
|
52
|
+
end
|
|
53
|
+
end
|
|
@@ -0,0 +1,41 @@
|
|
|
1
|
+
# frozen_string_literal: true
|
|
2
|
+
|
|
3
|
+
module GuardrailsRuby
|
|
4
|
+
class Violation
|
|
5
|
+
ACTIONS = %i[block warn redact log].freeze
|
|
6
|
+
|
|
7
|
+
attr_reader :type, :detail, :action, :severity, :matches, :sanitized
|
|
8
|
+
|
|
9
|
+
def initialize(type:, detail:, action: :block, severity: nil, matches: nil, sanitized: nil)
|
|
10
|
+
unless ACTIONS.include?(action)
|
|
11
|
+
raise ArgumentError, "Invalid action #{action.inspect}. Must be one of: #{ACTIONS.join(', ')}"
|
|
12
|
+
end
|
|
13
|
+
|
|
14
|
+
@type = type
|
|
15
|
+
@detail = detail
|
|
16
|
+
@action = action
|
|
17
|
+
@severity = severity || default_severity(action)
|
|
18
|
+
@matches = matches
|
|
19
|
+
@sanitized = sanitized
|
|
20
|
+
end
|
|
21
|
+
|
|
22
|
+
def to_s
|
|
23
|
+
"#{type}: #{detail} (action=#{action})"
|
|
24
|
+
end
|
|
25
|
+
|
|
26
|
+
def inspect
|
|
27
|
+
"#<#{self.class} type=#{type.inspect}, detail=#{detail.inspect}, action=#{action.inspect}>"
|
|
28
|
+
end
|
|
29
|
+
|
|
30
|
+
private
|
|
31
|
+
|
|
32
|
+
def default_severity(action)
|
|
33
|
+
case action
|
|
34
|
+
when :block then :high
|
|
35
|
+
when :redact then :medium
|
|
36
|
+
when :warn then :low
|
|
37
|
+
when :log then :info
|
|
38
|
+
end
|
|
39
|
+
end
|
|
40
|
+
end
|
|
41
|
+
end
|
|
@@ -0,0 +1,38 @@
|
|
|
1
|
+
# frozen_string_literal: true
|
|
2
|
+
|
|
3
|
+
require_relative "guardrails_ruby/version"
|
|
4
|
+
require_relative "guardrails_ruby/configuration"
|
|
5
|
+
require_relative "guardrails_ruby/violation"
|
|
6
|
+
require_relative "guardrails_ruby/result"
|
|
7
|
+
require_relative "guardrails_ruby/check"
|
|
8
|
+
require_relative "guardrails_ruby/guard"
|
|
9
|
+
require_relative "guardrails_ruby/middleware"
|
|
10
|
+
|
|
11
|
+
# Require built-in checks
|
|
12
|
+
checks_dir = File.join(__dir__, "guardrails_ruby", "checks")
|
|
13
|
+
Dir[File.join(checks_dir, "*.rb")].sort.each { |f| require f }
|
|
14
|
+
|
|
15
|
+
# Require redactors
|
|
16
|
+
require_relative "guardrails_ruby/redactors/pii_redactor"
|
|
17
|
+
require_relative "guardrails_ruby/redactors/keyword_redactor"
|
|
18
|
+
|
|
19
|
+
# Require Rails integration (controller concern is always available;
|
|
20
|
+
# railtie is only loaded when Rails is defined)
|
|
21
|
+
require_relative "guardrails_ruby/rails/controller"
|
|
22
|
+
require_relative "guardrails_ruby/rails/railtie" if defined?(::Rails::Railtie)
|
|
23
|
+
|
|
24
|
+
module GuardrailsRuby
|
|
25
|
+
class << self
|
|
26
|
+
def configuration
|
|
27
|
+
@configuration ||= Configuration.new
|
|
28
|
+
end
|
|
29
|
+
|
|
30
|
+
def configure
|
|
31
|
+
yield(configuration)
|
|
32
|
+
end
|
|
33
|
+
|
|
34
|
+
def reset_configuration!
|
|
35
|
+
@configuration = Configuration.new
|
|
36
|
+
end
|
|
37
|
+
end
|
|
38
|
+
end
|
metadata
ADDED
|
@@ -0,0 +1,115 @@
|
|
|
1
|
+
--- !ruby/object:Gem::Specification
|
|
2
|
+
name: guardrails-ruby
|
|
3
|
+
version: !ruby/object:Gem::Version
|
|
4
|
+
version: 0.1.0
|
|
5
|
+
platform: ruby
|
|
6
|
+
authors:
|
|
7
|
+
- Johannes Dwi Cahyo
|
|
8
|
+
bindir: bin
|
|
9
|
+
cert_chain: []
|
|
10
|
+
date: 1980-01-02 00:00:00.000000000 Z
|
|
11
|
+
dependencies:
|
|
12
|
+
- !ruby/object:Gem::Dependency
|
|
13
|
+
name: minitest
|
|
14
|
+
requirement: !ruby/object:Gem::Requirement
|
|
15
|
+
requirements:
|
|
16
|
+
- - "~>"
|
|
17
|
+
- !ruby/object:Gem::Version
|
|
18
|
+
version: '5.0'
|
|
19
|
+
type: :development
|
|
20
|
+
prerelease: false
|
|
21
|
+
version_requirements: !ruby/object:Gem::Requirement
|
|
22
|
+
requirements:
|
|
23
|
+
- - "~>"
|
|
24
|
+
- !ruby/object:Gem::Version
|
|
25
|
+
version: '5.0'
|
|
26
|
+
- !ruby/object:Gem::Dependency
|
|
27
|
+
name: rake
|
|
28
|
+
requirement: !ruby/object:Gem::Requirement
|
|
29
|
+
requirements:
|
|
30
|
+
- - "~>"
|
|
31
|
+
- !ruby/object:Gem::Version
|
|
32
|
+
version: '13.0'
|
|
33
|
+
type: :development
|
|
34
|
+
prerelease: false
|
|
35
|
+
version_requirements: !ruby/object:Gem::Requirement
|
|
36
|
+
requirements:
|
|
37
|
+
- - "~>"
|
|
38
|
+
- !ruby/object:Gem::Version
|
|
39
|
+
version: '13.0'
|
|
40
|
+
- !ruby/object:Gem::Dependency
|
|
41
|
+
name: webmock
|
|
42
|
+
requirement: !ruby/object:Gem::Requirement
|
|
43
|
+
requirements:
|
|
44
|
+
- - "~>"
|
|
45
|
+
- !ruby/object:Gem::Version
|
|
46
|
+
version: '3.0'
|
|
47
|
+
type: :development
|
|
48
|
+
prerelease: false
|
|
49
|
+
version_requirements: !ruby/object:Gem::Requirement
|
|
50
|
+
requirements:
|
|
51
|
+
- - "~>"
|
|
52
|
+
- !ruby/object:Gem::Version
|
|
53
|
+
version: '3.0'
|
|
54
|
+
executables: []
|
|
55
|
+
extensions: []
|
|
56
|
+
extra_rdoc_files: []
|
|
57
|
+
files:
|
|
58
|
+
- CLAUDE.md
|
|
59
|
+
- Gemfile
|
|
60
|
+
- LICENSE
|
|
61
|
+
- README.md
|
|
62
|
+
- Rakefile
|
|
63
|
+
- examples/basic.rb
|
|
64
|
+
- examples/custom_check.rb
|
|
65
|
+
- examples/rails_controller.rb
|
|
66
|
+
- guardrails-ruby.gemspec
|
|
67
|
+
- lib/guardrails_ruby.rb
|
|
68
|
+
- lib/guardrails_ruby/check.rb
|
|
69
|
+
- lib/guardrails_ruby/checks/competitor_mention.rb
|
|
70
|
+
- lib/guardrails_ruby/checks/encoding.rb
|
|
71
|
+
- lib/guardrails_ruby/checks/format.rb
|
|
72
|
+
- lib/guardrails_ruby/checks/hallucinated_emails.rb
|
|
73
|
+
- lib/guardrails_ruby/checks/hallucinated_urls.rb
|
|
74
|
+
- lib/guardrails_ruby/checks/keyword_filter.rb
|
|
75
|
+
- lib/guardrails_ruby/checks/max_length.rb
|
|
76
|
+
- lib/guardrails_ruby/checks/pii.rb
|
|
77
|
+
- lib/guardrails_ruby/checks/prompt_injection.rb
|
|
78
|
+
- lib/guardrails_ruby/checks/relevance.rb
|
|
79
|
+
- lib/guardrails_ruby/checks/topic.rb
|
|
80
|
+
- lib/guardrails_ruby/checks/toxic_language.rb
|
|
81
|
+
- lib/guardrails_ruby/configuration.rb
|
|
82
|
+
- lib/guardrails_ruby/guard.rb
|
|
83
|
+
- lib/guardrails_ruby/middleware.rb
|
|
84
|
+
- lib/guardrails_ruby/rails/controller.rb
|
|
85
|
+
- lib/guardrails_ruby/rails/railtie.rb
|
|
86
|
+
- lib/guardrails_ruby/redactors/keyword_redactor.rb
|
|
87
|
+
- lib/guardrails_ruby/redactors/pii_redactor.rb
|
|
88
|
+
- lib/guardrails_ruby/result.rb
|
|
89
|
+
- lib/guardrails_ruby/version.rb
|
|
90
|
+
- lib/guardrails_ruby/violation.rb
|
|
91
|
+
homepage: https://github.com/johannesdwicahyo/guardrails-ruby
|
|
92
|
+
licenses:
|
|
93
|
+
- MIT
|
|
94
|
+
metadata:
|
|
95
|
+
homepage_uri: https://github.com/johannesdwicahyo/guardrails-ruby
|
|
96
|
+
source_code_uri: https://github.com/johannesdwicahyo/guardrails-ruby
|
|
97
|
+
changelog_uri: https://github.com/johannesdwicahyo/guardrails-ruby/blob/main/CHANGELOG.md
|
|
98
|
+
rdoc_options: []
|
|
99
|
+
require_paths:
|
|
100
|
+
- lib
|
|
101
|
+
required_ruby_version: !ruby/object:Gem::Requirement
|
|
102
|
+
requirements:
|
|
103
|
+
- - ">="
|
|
104
|
+
- !ruby/object:Gem::Version
|
|
105
|
+
version: '3.0'
|
|
106
|
+
required_rubygems_version: !ruby/object:Gem::Requirement
|
|
107
|
+
requirements:
|
|
108
|
+
- - ">="
|
|
109
|
+
- !ruby/object:Gem::Version
|
|
110
|
+
version: '0'
|
|
111
|
+
requirements: []
|
|
112
|
+
rubygems_version: 3.6.9
|
|
113
|
+
specification_version: 4
|
|
114
|
+
summary: Input/output validation and safety framework for LLM applications in Ruby
|
|
115
|
+
test_files: []
|