your_ai_insight 1.0.9 → 1.0.11

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d87c237089e57905bc9d515c0794f85aa8e398bb37d45395fe20b7350193f51d
-  data.tar.gz: 52567467c46c142f6e5e23e27473a0c5dc67679130863513fdff295b77b6fbed
+  metadata.gz: c400f6bb1a051f84900f4ca7c4d229e6f7193aff9bf27b1c6e13772b36935dc6
+  data.tar.gz: a05ff0274c36660ed5a0e0dfde6ec242e8c293384dc835fd1b02ba0c471e10bd
 SHA512:
-  metadata.gz: a5b17824ec83493d6323b9afba879fd95184ceaec6a4928183ff87fe12f42cbea21deec59b5433d5bd360fc0940a502378047b79c95d1eb08bc276637fc94378
-  data.tar.gz: f0fc882bd61d7741b3bc1bbda2504ee975a4bb5e5bd3a207a3af50c6c09f50d7be88060277d1e702d7056efec93fa68f499bcbc3a51b46db11495307b775e531
+  metadata.gz: e0893b4723fc2827ea30eabc856f9a48820bc377a286b017be3d78e486ffa218e5930c6f4dfb1fca77fc5616e87038912bd11fee75304638c25b6ebdb9a9d879
+  data.tar.gz: cc045b166a9f36f017d1f2a90f02412eb213c3d54ee0a739a7958e9a9dd47ec87779a25ac2362c9c460d99178a14cba294a84084efbccd1cc7cd6505f9490dd6

data/app/services/your_ai_insight/claude_service.rb

@@ -3,7 +3,6 @@ require "httparty"
 module YourAiInsight
   class ClaudeService
     include HTTParty
-    base_uri "https://api.anthropic.com"
 
     SYSTEM_PROMPT = <<~PROMPT.freeze
       You are an expert facility management analyst for AllPro IFM.
@@ -32,11 +31,24 @@ module YourAiInsight
     PROMPT
 
     def initialize
-      @api_key = YourAiInsight.config.anthropic_api_key ||
-                 ENV["ANTHROPIC_API_KEY"] ||
-                 raise(ConfigurationError, "Set YourAiInsight.configure { |c| c.anthropic_api_key = ENV['ANTHROPIC_API_KEY'] }")
-      @model      = YourAiInsight.config.claude_model
-      @max_tokens = YourAiInsight.config.max_tokens
+      cfg          = YourAiInsight.config
+      @provider    = (cfg.ai_provider || :claude).to_sym
+      @max_tokens  = cfg.max_tokens
+
+      case @provider
+      when :openrouter
+        @api_key    = cfg.openrouter_api_key ||
+                      ENV["OPENROUTER_API_KEY"] ||
+                      raise(ConfigurationError, "Set config.openrouter_api_key or ENV['OPENROUTER_API_KEY']")
+        @model      = cfg.openrouter_model
+        @site_url   = cfg.openrouter_site_url
+        @site_name  = cfg.openrouter_site_name
+      else
+        @api_key    = cfg.anthropic_api_key ||
+                      ENV["ANTHROPIC_API_KEY"] ||
+                      raise(ConfigurationError, "Set config.anthropic_api_key or ENV['ANTHROPIC_API_KEY']")
+        @model      = cfg.claude_model
+      end
     end
 
     # Free-form chat with optional live data context injected
@@ -57,8 +69,12 @@ module YourAiInsight
     private
 
     def call_api(messages)
+      @provider == :openrouter ? call_openrouter(messages) : call_claude(messages)
+    end
+
+    def call_claude(messages)
       response = self.class.post(
-        "/v1/messages",
+        "https://api.anthropic.com/v1/messages",
         headers: {
           "x-api-key"         => @api_key,
           "anthropic-version" => "2023-06-01",
@@ -74,7 +90,6 @@ module YourAiInsight
       )
 
       body = JSON.parse(response.body)
-
       unless response.success?
         raise ApiError, "Claude API error (HTTP #{response.code}): #{body.dig('error', 'message')}"
       end
@@ -86,6 +101,40 @@ module YourAiInsight
       raise ApiError, "Network error reaching Claude API: #{e.message}"
     end
 
+    def call_openrouter(messages)
+      headers = {
+        "Authorization" => "Bearer #{@api_key}",
+        "Content-Type"  => "application/json"
+      }
+      headers["HTTP-Referer"] = @site_url  if @site_url.present?
+      headers["X-Title"]      = @site_name if @site_name.present?
+
+      # Prepend system prompt as a system message (OpenAI-compatible format)
+      full_messages = [{ role: "system", content: SYSTEM_PROMPT }] + messages
+
+      response = self.class.post(
+        "https://openrouter.ai/api/v1/chat/completions",
+        headers: headers,
+        body: {
+          model:      @model,
+          max_tokens: @max_tokens,
+          messages:   full_messages
+        }.to_json,
+        timeout: 90
+      )
+
+      body = JSON.parse(response.body)
+      unless response.success?
+        raise ApiError, "OpenRouter API error (HTTP #{response.code}): #{body.dig('error', 'message')}"
+      end
+
+      body.dig("choices", 0, "message", "content").presence ||
+        raise(ApiError, "Empty response from OpenRouter API")
+
+    rescue HTTParty::Error, Timeout::Error => e
+      raise ApiError, "Network error reaching OpenRouter API: #{e.message}"
+    end
+
     # Injects a live snapshot into the chat prompt
     def context_block(ctx)
       return nil if ctx.blank?

data/app/services/your_ai_insight/facility_data_service.rb

@@ -80,15 +80,17 @@ module YourAiInsight
     end
 
     def jobs_by_status
-      rows = exec_query("SELECT #{@c.jobs_status_col}, COUNT(*) as cnt FROM #{j} #{job_where} GROUP BY #{@c.jobs_status_col}")
+      col  = @c.jobs_status_col
+      rows = exec_query("SELECT #{col}, COUNT(*) as cnt FROM #{j} #{job_where} GROUP BY #{col}")
       labels = { 0 => "new", 1 => "assigned", 2 => "active", 3 => "work_complete", 4 => "all_tasks_complete", 5 => "finaled" }
-      rows.each_with_object({}) { |r, h| h[labels[r["status"].to_i] || r["status"].to_s] = r["cnt"].to_i }
+      rows.each_with_object({}) { |r, h| h[labels[r[col].to_i] || r[col].to_s] = r["cnt"].to_i }
     rescue StandardError; {}
     end
 
     def jobs_by_priority
-      rows = exec_query("SELECT #{@c.jobs_priority_col}, COUNT(*) as cnt FROM #{j} #{job_where} AND #{@c.jobs_priority_col} IS NOT NULL GROUP BY #{@c.jobs_priority_col}")
-      rows.each_with_object({}) { |r, h| h[r["priority"]] = r["cnt"].to_i }
+      col  = @c.jobs_priority_col
+      rows = exec_query("SELECT #{col}, COUNT(*) as cnt FROM #{j} #{job_where} AND #{col} IS NOT NULL GROUP BY #{col}")
+      rows.each_with_object({}) { |r, h| h[r[col]] = r["cnt"].to_i }
     rescue StandardError; {}
     end
 
@@ -103,28 +105,28 @@ module YourAiInsight
 
     def jobs_on_hold_detail
       exec_query(<<~SQL)
-        SELECT j.id, j.name, j.#{@c.jobs_priority_col}, j.#{@c.jobs_status_col},
-               j.created_at, l.name as location_name, c.name as customer_name
+        SELECT #{j}.id, #{j}.name, #{j}.#{@c.jobs_priority_col}, #{j}.#{@c.jobs_status_col},
+               #{j}.created_at, l.name as location_name, c.name as customer_name
         FROM #{j}
-        LEFT JOIN #{loc} l ON l.id = j.#{@c.jobs_location_id_col}
-        LEFT JOIN #{cust} c ON c.id = j.#{@c.jobs_customer_id_col}
-        #{job_where} AND j.#{@c.jobs_hold_col} = true
-        ORDER BY j.created_at ASC LIMIT 25
+        LEFT JOIN #{loc} l ON l.id = #{j}.#{@c.jobs_location_id_col}
+        LEFT JOIN #{cust} c ON c.id = #{j}.#{@c.jobs_customer_id_col}
+        #{job_where} AND #{j}.#{@c.jobs_hold_col} = true
+        ORDER BY #{j}.created_at ASC LIMIT 25
       SQL
     rescue StandardError; []
     end
 
     def over_budget_jobs_detail
       exec_query(<<~SQL)
-        SELECT j.id, j.name,
+        SELECT #{j}.id, #{j}.name,
                COALESCE(SUM(e.#{@c.expenses_actual_col}),0) as total_actual,
                COALESCE(SUM(e.#{@c.expenses_budget_col}),0) as total_budget,
                l.name as location_name
         FROM #{j}
-        LEFT JOIN #{exp} e ON e.#{@c.expenses_job_id_col} = j.id
-        LEFT JOIN #{loc} l ON l.id = j.#{@c.jobs_location_id_col}
+        LEFT JOIN #{exp} e ON e.#{@c.expenses_job_id_col} = #{j}.id
+        LEFT JOIN #{loc} l ON l.id = #{j}.#{@c.jobs_location_id_col}
         #{job_where}
-        GROUP BY j.id, j.name, l.name
+        GROUP BY #{j}.id, #{j}.name, l.name
         HAVING COALESCE(SUM(e.#{@c.expenses_actual_col}),0) > COALESCE(SUM(e.#{@c.expenses_budget_col}),0)
         ORDER BY (COALESCE(SUM(e.#{@c.expenses_actual_col}),0) - COALESCE(SUM(e.#{@c.expenses_budget_col}),0)) DESC
         LIMIT 15
@@ -140,7 +142,7 @@ module YourAiInsight
       exec_query(<<~SQL)
         SELECT l.name, COUNT(j.id) as job_count
         FROM #{loc} l
-        LEFT JOIN #{j} ON j.#{@c.jobs_location_id_col} = l.id
+        LEFT JOIN #{j} j ON j.#{@c.jobs_location_id_col} = l.id
         #{location_join_where("l")}
         GROUP BY l.name ORDER BY job_count DESC LIMIT 10
       SQL
@@ -151,7 +153,7 @@ module YourAiInsight
       exec_query(<<~SQL)
         SELECT c.name, COUNT(j.id) as job_count
         FROM #{cust} c
-        LEFT JOIN #{j} ON j.#{@c.jobs_customer_id_col} = c.id
+        LEFT JOIN #{j} j ON j.#{@c.jobs_customer_id_col} = c.id
         #{customer_join_where("c")}
         GROUP BY c.name ORDER BY job_count DESC LIMIT 10
       SQL
@@ -191,15 +193,15 @@ module YourAiInsight
 
     def overdue_tasks_detail
       exec_query(<<~SQL)
-        SELECT t.id, t.name, t.#{@c.tasks_complete_col}, t.#{@c.tasks_budget_col},
-               t.#{@c.tasks_actual_col}, j.name as job_name, l.name as location_name
+        SELECT #{t}.id, #{t}.name, #{t}.#{@c.tasks_complete_col}, #{t}.#{@c.tasks_budget_col},
+               #{t}.#{@c.tasks_actual_col}, j.name as job_name, l.name as location_name
         FROM #{t}
-        LEFT JOIN #{j} ON j.id = t.#{@c.tasks_job_id_col}
+        LEFT JOIN #{j} j ON j.id = #{t}.#{@c.tasks_job_id_col}
         LEFT JOIN #{loc} l ON l.id = j.#{@c.jobs_location_id_col}
         #{task_where}
-        AND t.#{@c.tasks_complete_col} < CURRENT_DATE
-        AND t.#{@c.tasks_status_col} != 2
-        ORDER BY t.#{@c.tasks_complete_col} ASC LIMIT 20
+        AND #{t}.#{@c.tasks_complete_col} < CURRENT_DATE
+        AND #{t}.#{@c.tasks_status_col} != 2
+        ORDER BY #{t}.#{@c.tasks_complete_col} ASC LIMIT 20
       SQL
     rescue StandardError; []
     end
@@ -227,9 +229,9 @@ module YourAiInsight
       exec_query(<<~SQL)
         SELECT b.id, b.#{@c.bids_amount_col}, b.received, b.notes,
                t.name as task_name, j.name as job_name, v.name as vendor_name
-        FROM #{b}
-        LEFT JOIN #{t} ON #{t}.id = b.#{@c.bids_task_id_col}
-        LEFT JOIN #{j} ON #{j}.id = #{t}.#{@c.tasks_job_id_col}
+        FROM #{b} b
+        LEFT JOIN #{t} t ON t.id = b.#{@c.bids_task_id_col}
+        LEFT JOIN #{j} j ON j.id = t.#{@c.tasks_job_id_col}
         LEFT JOIN vendors v ON v.id = b.vendor_id
         WHERE (b.#{@c.bids_status_col} IS NULL OR b.#{@c.bids_status_col} NOT IN ('accepted','rejected'))
         AND b.created_at BETWEEN #{range_sql}
@@ -365,7 +367,7 @@ module YourAiInsight
     def br  = @c.budget_requests_table
 
     def range_sql
-      "'#{@date_range.first.to_s(:db)}' AND '#{@date_range.last.to_s(:db)}'"
+      "'#{@date_range.first.strftime('%Y-%m-%d %H:%M:%S')}' AND '#{@date_range.last.strftime('%Y-%m-%d %H:%M:%S')}'"
     end
 
     def range_label

data/lib/your_ai_insight/configuration.rb

@@ -1,12 +1,21 @@
 module YourAiInsight
   class Configuration
-    # ── Required ──────────────────────────────────────────────────────────────
-    attr_accessor :anthropic_api_key
+    # ── AI Provider ───────────────────────────────────────────────────────────
+    # :claude (default) or :openrouter
+    attr_accessor :ai_provider
 
-    # ── Model / Claude settings ───────────────────────────────────────────────
-    attr_accessor :claude_model   # default: claude-sonnet-4-20250514
+    # ── Claude / Anthropic settings ───────────────────────────────────────────
+    attr_accessor :anthropic_api_key
+    attr_accessor :claude_model   # default: claude-opus-4-5
     attr_accessor :max_tokens     # default: 2048
 
+    # ── OpenRouter settings ───────────────────────────────────────────────────
+    attr_accessor :openrouter_api_key
+    # default: meta-llama/llama-3.3-70b-instruct:free  (free, no billing needed)
+    attr_accessor :openrouter_model
+    attr_accessor :openrouter_site_url  # optional — sent as HTTP-Referer
+    attr_accessor :openrouter_site_name # optional — sent as X-Title
+
     # ── Multi-tenancy: scope data to current user's locations/customers ───────
     # Proc receives current_user and returns an Array of location_ids, or nil for all.
     # Example: config.location_scope = ->(user) { user.locations.pluck(:id) }
@@ -87,9 +96,14 @@ module YourAiInsight
     attr_accessor :sub_contractors_table    # "sub_contractors"
 
     def initialize
-      @claude_model   = "claude-sonnet-4-20250514"
+      @ai_provider    = :claude
+      @claude_model   = "claude-opus-4-5"
       @max_tokens     = 2048
       @company_name   = "AllPro IFM"
+
+      @openrouter_model     = "meta-llama/llama-3.3-70b-instruct:free"
+      @openrouter_site_url  = nil
+      @openrouter_site_name = nil
       @logo_url       = nil
       @default_recipients = []
 

data/lib/your_ai_insight/version.rb

@@ -1,3 +1,3 @@
 module YourAiInsight
-  VERSION = "1.0.9"
+  VERSION = "1.0.11"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: your_ai_insight
 version: !ruby/object:Gem::Version
-  version: 1.0.9
+  version: 1.0.11
 platform: ruby
 authors:
 - AllPro IFM