ai_sentinel 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 943695a0b61808c8db91070a49ea4fdedc5616a4769f6522eccd0c7bca2a9541
-  data.tar.gz: d8c1d987a8c60bc943c2c156c4642a296e622ff71ab3e43412e35edefcdc9e37
+  metadata.gz: 3fbc0d9833c0562b188f975fa0d32cb8f175bda40149d1e3c3aa0ec64863f4ad
+  data.tar.gz: 4fb375aa24d1dfa447bf9a52a10f06a03cde84610f75dcdf875cc2aff93b3f97
 SHA512:
-  metadata.gz: 2d8b349c120bb241a147509b489843fa7c8a7928660696471c5dbfcccf2a45eaff1e82d6c5630a81a659106a31b5344666c8d2e1e530b0b39832b6fe8c9949b8
-  data.tar.gz: 4d97b9f059722033d67c09049d9c5d66690123b60fd4ca827c3a8301b7ddd32388352b1af49802fd92870d1ac022f9ad9a96579fe182c7c905dfb7f5d023786c
+  metadata.gz: 3516b53aeea1e1175b0d96849a12a3d79ca1ad3c65e91c559f6b746ba2b091f1fd333fe2934abfe14aef82d1da8cc7998a58ca9caba4299332ebd3b226487563
+  data.tar.gz: 9bf82e503bddd8194bf5b492097d507a95e494a48dc9ea4ca2f650638ca48a161e4c50ad922c95495e66d20cd67fb5fd71c23635372c4c5a063223536d34825e

data/lib/ai_sentinel/cli.rb

@@ -1,5 +1,6 @@
 # frozen_string_literal: true
 
+require 'fileutils'
 require 'thor'
 require_relative 'cli/helpers'
 require_relative 'cli/context_display'
@@ -28,6 +29,25 @@ module AiSentinel
       AiSentinel.start(daemonize: options[:daemonize])
     end
 
+    desc 'stop', 'Stop a running daemon'
+    method_option :config, type: :string, aliases: '-c', desc: 'Path to config file'
+    def stop
+      load_config
+      pid_path = AiSentinel.configuration.pid_file
+
+      unless File.exist?(pid_path)
+        say "No PID file found at #{pid_path}. Is the daemon running?"
+        return
+      end
+
+      pid = File.read(pid_path).strip.to_i
+      Process.kill('TERM', pid)
+      say "Sent TERM signal to process #{pid}."
+    rescue Errno::ESRCH
+      say "Process #{pid} not found. Removing stale PID file."
+      FileUtils.rm_f(pid_path)
+    end
+
     desc 'run WORKFLOW', 'Manually trigger a workflow immediately'
     def run_workflow(workflow_name)
       load_config

data/lib/ai_sentinel/config_loader.rb

@@ -20,7 +20,8 @@ module AiSentinel
       'log_files' => ->(config, val) { config.log_files = val },
       'on_prompt_change' => ->(config, val) { config.on_prompt_change = val.to_sym },
       'tool_safety' => ->(config, val) { config.tool_safety = val.transform_keys(&:to_sym) },
-      'max_tool_rounds' => ->(config, val) { config.max_tool_rounds = val }
+      'max_tool_rounds' => ->(config, val) { config.max_tool_rounds = val },
+      'pid_file' => ->(config, val) { config.pid_file = File.expand_path(val) }
     }.freeze
 
     attr_reader :config_path, :raw_config

data/lib/ai_sentinel/configuration.rb

@@ -31,7 +31,7 @@ module AiSentinel
     attr_accessor :provider, :api_key, :database_path, :max_context_messages,
                   :compaction_threshold, :compaction_buffer, :log_file,
                   :log_file_size, :log_files, :on_prompt_change,
-                  :tool_safety, :max_tool_rounds
+                  :tool_safety, :max_tool_rounds, :pid_file
     attr_writer :model, :base_url, :logger
 
     def initialize
@@ -50,6 +50,7 @@ module AiSentinel
       @on_prompt_change = :ask
       @tool_safety = nil
       @max_tool_rounds = DEFAULT_MAX_TOOL_ROUNDS
+      @pid_file = File.join(Dir.home, '.ai_sentinel', 'ai_sentinel.pid')
     end
 
     def logger
@@ -73,7 +74,7 @@ module AiSentinel
         "database_path=#{database_path} log_file=#{log_file || 'STDOUT'} " \
         "max_context_messages=#{max_context_messages} " \
         "compaction_threshold=#{compaction_threshold} compaction_buffer=#{compaction_buffer} " \
-        "max_tool_rounds=#{max_tool_rounds} " \
+        "max_tool_rounds=#{max_tool_rounds} pid_file=#{pid_file} " \
         "api_key=#{api_key ? '[FILTERED]' : 'nil'}>"
     end
     alias to_s inspect

data/lib/ai_sentinel/scheduler.rb

@@ -44,12 +44,13 @@ module AiSentinel
     end
 
     def pid_file
-      @pid_file ||= File.join(Dir.pwd, 'ai_sentinel.pid')
+      @pid_file ||= configuration.pid_file
     end
 
     private
 
     def write_pid_file
+      FileUtils.mkdir_p(File.dirname(pid_file))
       File.write(pid_file, Process.pid.to_s)
     end
 

data/lib/ai_sentinel/version.rb

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

metadata

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