ai_sentinel 0.2.2 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 426089deb83c2bedf49d58d170b9241d0cd14a0284fa7b23acc7e8b5932f666f
-  data.tar.gz: 7fd0fc6880e2cacc358a2e32bcbed7636e70aade5d688294c5ef79961cb20c92
+  metadata.gz: 57b7b717300a40c7538e4a7da84bd574677c9bd38fa55c094ad44725354e07d8
+  data.tar.gz: 1897eb521d5a9b2c71a1b1d54e7aa015172466accb65ef6c0bfa6a3b9ac5879a
 SHA512:
-  metadata.gz: b19b08eafd1dc89bcb3ca42f29ae483e6751a1bbf50546b6be65a4b2840705cf705d25d340af8d81b0a443267f20227e1f492b5d44403e0a0cd5ef1d5f93f643
-  data.tar.gz: 1eb5109adbf99a1802e393942282f05e7635274e0d658e4799488f9a0526cdf94db249df83cfaf65d0097c155448b8445f05bf6fce330d15fdd7f3bff6c51d1e
+  metadata.gz: 46a5467ba869e34a6b2e4f942c23cdcb719a8b2d8d6583ba29237873bb0145bed9ef736d6949bdf0434bf0f05fce8529b790b89f691594f40545667be12b574d
+  data.tar.gz: 8d05fe5f7477260054a4d9083fa2cbdcecb0effc5c2c701b1768fae3882604e9a8840e0cb5ceba0fb11b61a2fc5a488f1e3eef61d298f3d1e06c6d6e60327e64

data/README.md

@@ -1,5 +1,7 @@
 # AiSentinel
 
+[![Gem Version](https://badge.fury.io/rb/ai_sentinel.svg?icon=si%3Arubygems)](https://badge.fury.io/rb/ai_sentinel)
+
 A lightweight Ruby gem for scheduling AI-driven tasks. Define workflows in a YAML config file that run on a cron schedule, process data through LLMs, and take conditional actions based on the results. Designed to be self-hostable on minimal hardware -- just Ruby and SQLite.
 
 ## Table of contents

data/exe/ai_sentinel

@@ -2,4 +2,10 @@
 # frozen_string_literal: true
 
 require 'ai_sentinel'
-AiSentinel::CLI.start(ARGV)
+
+begin
+  AiSentinel::CLI.start(ARGV)
+rescue StandardError => e
+  AiSentinel.log_error(e, context: 'Fatal error')
+  exit 1
+end

data/lib/ai_sentinel/providers/anthropic.rb

@@ -75,8 +75,8 @@ module AiSentinel
           result = tool_executor.execute(tool_name, tool_input)
           AiSentinel.logger.info("    Tool result: #{result.to_s[0..200]}")
           { 'type' => 'tool_result', 'tool_use_id' => tool_id, 'content' => result.to_s }
-        rescue Error => e
-          AiSentinel.logger.warn("    Tool error: #{e.message}")
+        rescue StandardError => e
+          AiSentinel.log_error(e, context: "Tool '#{tool_name}' error")
           { 'type' => 'tool_result', 'tool_use_id' => tool_id, 'content' => "Error: #{e.message}",
             'is_error' => true }
         end

data/lib/ai_sentinel/providers/base.rb

@@ -94,7 +94,7 @@ module AiSentinel
       def compact_context(context_key)
         ContextCompactor.new(context_key: context_key, configuration: configuration).compact_if_needed
       rescue StandardError => e
-        AiSentinel.logger.warn("Context compaction failed for '#{context_key}': #{e.message}")
+        AiSentinel.log_error(e, context: "Context compaction failed for '#{context_key}'")
       end
 
       def prune_old_messages(context_id)

data/lib/ai_sentinel/providers/openai.rb

@@ -61,19 +61,17 @@ module AiSentinel
       def execute_tool_call(tool_executor, tool_call, round)
         function = tool_call['function']
         tool_name = function['name']
-        tool_input = JSON.parse(function['arguments'])
         tool_id = tool_call['id']
 
         AiSentinel.logger.info("    Tool call [round #{round + 1}]: #{tool_name}(#{function['arguments']})")
 
-        begin
-          result = tool_executor.execute(tool_name, tool_input)
-          AiSentinel.logger.info("    Tool result: #{result.to_s[0..200]}")
-          { 'role' => 'tool', 'tool_call_id' => tool_id, 'content' => result.to_s }
-        rescue Error => e
-          AiSentinel.logger.warn("    Tool error: #{e.message}")
-          { 'role' => 'tool', 'tool_call_id' => tool_id, 'content' => "Error: #{e.message}" }
-        end
+        tool_input = JSON.parse(function['arguments'])
+        result = tool_executor.execute(tool_name, tool_input)
+        AiSentinel.logger.info("    Tool result: #{result.to_s[0..200]}")
+        { 'role' => 'tool', 'tool_call_id' => tool_id, 'content' => result.to_s }
+      rescue StandardError => e
+        AiSentinel.log_error(e, context: "Tool '#{tool_name}' error")
+        { 'role' => 'tool', 'tool_call_id' => tool_id, 'content' => "Error: #{e.message}" }
       end
 
       def build_messages(prompt, system, context_key, remember, limit: nil)

data/lib/ai_sentinel/runner.rb

@@ -31,7 +31,7 @@ module AiSentinel
       context
     rescue StandardError => e
       Persistence::ExecutionLog.fail(execution_id, e.message)
-      AiSentinel.logger.error("Workflow '#{workflow.name}' failed: #{e.message}")
+      AiSentinel.log_error(e, context: "Workflow '#{workflow.name}' failed")
       raise
     end
 

data/lib/ai_sentinel/scheduler.rb

@@ -17,8 +17,11 @@ module AiSentinel
       apply_working_directory
 
       if daemonize
+        Persistence::Database.disconnect
         Process.daemon(true, true)
+        Persistence::Database.setup(configuration.database_path)
         write_pid_file
+        setup_crash_cleanup
         AiSentinel.logger.info("AiSentinel started in background (PID #{Process.pid}, #{registry.size} workflow(s))")
       else
         AiSentinel.logger.info("AiSentinel started (#{registry.size} workflow(s)). Press Ctrl+C to stop.")
@@ -32,10 +35,16 @@ module AiSentinel
       @rufus.join
       cleanup_pid_file
       AiSentinel.logger.info('AiSentinel stopped')
+    rescue StandardError => e
+      AiSentinel.log_error(e, context: 'Scheduler crashed')
+      cleanup_pid_file
+      raise
     end
 
     def stop
-      @rufus.shutdown
+      @rufus&.shutdown
+    rescue StandardError => e
+      AiSentinel.log_error(e, context: 'Error during shutdown')
     end
 
     def trigger(workflow_name)
@@ -72,13 +81,17 @@ module AiSentinel
       FileUtils.rm_f(pid_file)
     end
 
+    def setup_crash_cleanup
+      at_exit { cleanup_pid_file }
+    end
+
     def register_workflows
       registry.each do |name, workflow|
         @rufus.cron(workflow.schedule_expression) do
           runner = Runner.new(workflow: workflow, configuration: configuration)
           runner.execute
         rescue StandardError => e
-          AiSentinel.logger.error("Workflow '#{name}' failed: #{e.message}")
+          AiSentinel.log_error(e, context: "Workflow '#{name}' failed")
         end
 
         AiSentinel.logger.info("Registered workflow '#{name}' with schedule '#{workflow.schedule_expression}'")

data/lib/ai_sentinel/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module AiSentinel
-  VERSION = '0.2.2'
+  VERSION = '0.3.0'
 end

data/lib/ai_sentinel.rb

@@ -50,6 +50,14 @@ module AiSentinel
       configuration.logger
     end
 
+    def log_error(error, context: nil)
+      message = context ? "#{context}: #{error.message}" : error.message
+      logger.error(message)
+      error.backtrace&.first(10)&.each { |line| logger.error("  #{line}") }
+    rescue StandardError
+      nil
+    end
+
     def resolve_api_key
       configuration.api_key ||= ENV.fetch(configuration.env_key_name, nil)
     end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ai_sentinel
 version: !ruby/object:Gem::Version
-  version: 0.2.2
+  version: 0.3.0
 platform: ruby
 authors:
 - Mario Celi