ai_sentinel 0.2.2 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 426089deb83c2bedf49d58d170b9241d0cd14a0284fa7b23acc7e8b5932f666f
-  data.tar.gz: 7fd0fc6880e2cacc358a2e32bcbed7636e70aade5d688294c5ef79961cb20c92
+  metadata.gz: 54f5dcacdba120c16ed3a4c09384f8b981aef569b9fb66a350400ec8a3f13b4b
+  data.tar.gz: 0b15c9519a2085dcc7b3a3b920ff4b91e89f282da8346429f6152b886334225f
 SHA512:
-  metadata.gz: b19b08eafd1dc89bcb3ca42f29ae483e6751a1bbf50546b6be65a4b2840705cf705d25d340af8d81b0a443267f20227e1f492b5d44403e0a0cd5ef1d5f93f643
-  data.tar.gz: 1eb5109adbf99a1802e393942282f05e7635274e0d658e4799488f9a0526cdf94db249df83cfaf65d0097c155448b8445f05bf6fce330d15fdd7f3bff6c51d1e
+  metadata.gz: 02f23261d5e52c11c03b9c15c7633f5eb12b54daacb49062d3cc8eb5be7168b72c1069e0c1607d3c02d13867af4545455dbd995eb49cc01dfc25007a8ed5d4ff
+  data.tar.gz: 8a9a3159203a5e5e580dfe2801237ff68067a3ef9b30b23389e4e05204851d5135ad6d38093d568bbfd6b328186180d8df56d706363cfc5d5059fb8a110ff067

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/actions/ai_prompt.rb

@@ -32,7 +32,8 @@ module AiSentinel
           prompt_template: step.params[:prompt],
           system_template: step.params[:system],
           tool_executor: tool_executor,
-          max_tool_rounds: step.params.fetch(:max_tool_rounds, configuration.max_tool_rounds)
+          max_tool_rounds: step.params.fetch(:max_tool_rounds, configuration.max_tool_rounds),
+          compaction_prompt: step.params[:compaction_prompt]
         )
       end
 

data/lib/ai_sentinel/context_compactor.rb

@@ -10,11 +10,12 @@ module AiSentinel
       Respond with ONLY the summary, no preamble or explanation.
     PROMPT
 
-    attr_reader :context_key, :configuration
+    attr_reader :context_key, :configuration, :custom_compaction_prompt
 
-    def initialize(context_key:, configuration:)
+    def initialize(context_key:, configuration:, compaction_prompt: nil)
       @context_key = context_key
       @configuration = configuration
+      @custom_compaction_prompt = compaction_prompt
     end
 
     def compact_if_needed
@@ -74,7 +75,7 @@ module AiSentinel
 
       result = provider.chat(
         prompt: prompt,
-        system: SUMMARIZATION_PROMPT,
+        system: custom_compaction_prompt || SUMMARIZATION_PROMPT,
         workflow_name: nil,
         step_name: nil,
         remember: false

data/lib/ai_sentinel/providers/anthropic.rb

@@ -9,7 +9,8 @@ module AiSentinel
       API_VERSION = '2023-06-01'
 
       def chat(prompt:, system: nil, model: nil, workflow_name: nil, step_name: nil, remember: true,
-               prompt_template: nil, system_template: nil, tool_executor: nil, max_tool_rounds: 10)
+               prompt_template: nil, system_template: nil, tool_executor: nil, max_tool_rounds: 10,
+               compaction_prompt: nil)
         model ||= configuration.model
         context_key = "#{workflow_name}:#{step_name}"
 
@@ -26,7 +27,8 @@ module AiSentinel
         assistant_text = extract_text(response_data)
         if remember
           save_context(context_key, prompt, assistant_text,
-                       prompt_template: prompt_template, system_template: system_template)
+                       prompt_template: prompt_template, system_template: system_template,
+                       compaction_prompt: compaction_prompt)
         end
 
         Actions::AiPrompt::Result.new(
@@ -75,8 +77,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

@@ -66,7 +66,8 @@ module AiSentinel
         messages
       end
 
-      def save_context(context_key, user_message, assistant_message, prompt_template: nil, system_template: nil)
+      def save_context(context_key, user_message, assistant_message, prompt_template: nil, system_template: nil,
+                       compaction_prompt: nil)
         return unless Persistence::Database.connected?
 
         ctx = Persistence::Database.find_or_create_context(context_key)
@@ -81,7 +82,7 @@ module AiSentinel
 
         save_prompt_hash(context_key, prompt_template, system_template) if prompt_template
         prune_old_messages(ctx[:id])
-        compact_context(context_key)
+        compact_context(context_key, compaction_prompt: compaction_prompt)
       end
 
       def save_prompt_hash(context_key, prompt_template, system_template)
@@ -91,10 +92,11 @@ module AiSentinel
         )
       end
 
-      def compact_context(context_key)
-        ContextCompactor.new(context_key: context_key, configuration: configuration).compact_if_needed
+      def compact_context(context_key, compaction_prompt: nil)
+        ContextCompactor.new(context_key: context_key, configuration: configuration,
+                             compaction_prompt: compaction_prompt).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

@@ -7,7 +7,8 @@ module AiSentinel
   module Providers
     class Openai < Base
       def chat(prompt:, system: nil, model: nil, workflow_name: nil, step_name: nil, remember: true,
-               prompt_template: nil, system_template: nil, tool_executor: nil, max_tool_rounds: 10)
+               prompt_template: nil, system_template: nil, tool_executor: nil, max_tool_rounds: 10,
+               compaction_prompt: nil)
         model ||= configuration.model
         context_key = "#{workflow_name}:#{step_name}"
 
@@ -24,7 +25,8 @@ module AiSentinel
         assistant_text = extract_text(response_data)
         if remember
           save_context(context_key, prompt, assistant_text,
-                       prompt_template: prompt_template, system_template: system_template)
+                       prompt_template: prompt_template, system_template: system_template,
+                       compaction_prompt: compaction_prompt)
         end
 
         Actions::AiPrompt::Result.new(
@@ -61,19 +63,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.4.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.4.0
 platform: ruby
 authors:
 - Mario Celi