rails_code_auditor 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 83cd23ea4027538587a1814cbf2a3472bd818fe0cb15bf492a3ab54b565c2e63
-  data.tar.gz: 40438e254bed8d07966559ac52df8a3063f427167fbc236a8eac0fba6a591c1d
+  metadata.gz: 598189dda2fba4a0b37869531aae8cb254a7effbcdff35b5be9638892c79eafb
+  data.tar.gz: d90fc145ba60a605012b332a8c74a2400ba7607f0278387ec74eed58a0211677
 SHA512:
-  metadata.gz: e1cdd901ff82b3f2c2d5f0be65175a03b534a6d3a401c34a9c85861401ea2cf416162a1045fe34b2e8cf0ef79a3a220a1c1e9f87b81fff05d06b973735974227
-  data.tar.gz: 721d41c939a3e00321508a4c36033221e58b6f67a6fd1f0ee9df090e1ae289fafe0b65c94e0f07b7e20c6b3c4bad375463b763f00d440c7a96c29af529e6cde2
+  metadata.gz: 506c2a77006ff53ac87a5e58e86d322f79323d10f5a0714d1c0a9a75fa3b27ec32082894af8bd314146be709a464cce296558c5b01e3496100fc862fd2f5938e
+  data.tar.gz: 97a84b00a918e79e7a7ec9ab4e5b1c6100b6f1c719267649120b13ce9d4c3279d6b119cc5f91c3da7ced0cb4832309efbe987ed20b13f76fd8f74d9047c93b90

data/lib/rails_code_auditor/llm_client.rb

@@ -9,12 +9,13 @@ module RailsCodeAuditor
         tool.dup.tap do |entry|
           if entry.is_a?(Hash) && entry[:details].is_a?(String)
             entry[:details] = entry[:details].slice(0, 500) # Trim details to avoid LLM overload
-            entry[:details].gsub!(/\e\[[\d;]*m/, '')         # Remove ANSI color codes
+            entry[:details].gsub!(/\e\[[\d;]*m/, "") # Remove ANSI color codes
           end
         end
       end
     end
-    def self.score_with_llm(json_results)
+
+    def self.score_with_llm(json_results, endpoint: "http://localhost:11434/api/generate", model: "llama3")
       sanitized = sanitize_results(json_results)
       puts "[*] Scoring with LLM (LLaMA3)..."
 
@@ -34,9 +35,9 @@ module RailsCodeAuditor
         #{JSON.pretty_generate(sanitized)}
       PROMPT
 
-      uri = URI("http://localhost:11434/api/generate")
+      uri = URI(endpoint)
       body = {
-        model: "llama3",
+        model: model,
         prompt: prompt,
         stream: false
       }
@@ -48,24 +49,21 @@ module RailsCodeAuditor
       # Try to extract JSON from any surrounding text
       json_match = raw_output.match(/\{.*\}/m)
 
-      if json_match
-        parsed_scores = JSON.parse(json_match[0])
-        scored_results = parsed_scores.transform_values do |score|
-          remark = case score
-                  when 90..100 then "Excellent"
-                  when 75..89  then "Good"
-                  when 60..74  then "Average"
-                  else              "Needs Improvement"
-                  end
-          { score: score, remark: remark }
-        end
-        scored_results
-      else
-        raise "Response did not contain valid JSON"
+      raise "Response did not contain valid JSON" unless json_match
+
+      parsed_scores = JSON.parse(json_match[0])
+      parsed_scores.transform_values do |score|
+        remark = case score
+                 when 90..100 then "Excellent"
+                 when 75..89  then "Good"
+                 when 60..74  then "Average"
+                 else "Needs Improvement"
+                 end
+        { score: score, remark: remark }
       end
-    rescue => e
+    rescue StandardError => e
       puts "[!] LLM scoring failed: #{e.message}"
       nil
     end
   end
-end
+end

data/lib/rails_code_auditor/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RailsCodeAuditor
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

data/lib/rails_code_auditor.rb

@@ -15,11 +15,17 @@ module RailsCodeAuditor
   def self.run(args)
     puts "[*] Running Rails Code Auditor..."
 
+    # Parse CLI options
+    use_llm     = args.include?("--use-llm")
+    model_arg   = args[args.index("--llm-model") + 1] if args.include?("--llm-model")
+    endpoint_arg = args[args.index("--llm-endpoint") + 1] if args.include?("--llm-endpoint")
+
     raw_results = Analyzer.run_all
     results = ReportGenerator.normalize(raw_results)
     results[:simplecov] = SimpleCovRunner.run
-    scores = if args.include?("--use-llm")
-               LlmClient.score_with_llm(results) || Scorer.score(results)
+    scores = if use_llm
+               LlmClient.score_with_llm(results, model: model_arg || "llama3",
+                                                 endpoint: endpoint_arg || "http://localhost:11434/api/generate") || Scorer.score(results)
              else
                Scorer.score(results)
              end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: rails_code_auditor
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - sivamanikandan
 bindir: exe
 cert_chain: []
-date: 2025-07-24 00:00:00.000000000 Z
+date: 2025-07-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: brakeman