aicli 0.2.9 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f0ecf3b2378a31b6986dd9758fa346b3bd1aeef95547e60f57a16ae990358c3b
-  data.tar.gz: baaf14692eaba18db2ecf54dd77912f5d138153d4698500e519b62496f1455bb
+  metadata.gz: 3502a13ce44daa62a15e1e9bfe0bb20d372bca46b782f35dae8be05dac9dba1f
+  data.tar.gz: 7d4462f9b8bd94263aaf5da5ec87d35406f2b814842ad6217b0db0434ff027b1
 SHA512:
-  metadata.gz: 49266f502dca2be1094fcaa23cf084b8863e2cecfcfe1e45c281b0e1935c86f0ed597ecfe461deee228a7c97036a5d141ec0505bfa245249f987908318b98779
-  data.tar.gz: fc2e6d276064e9ac15a4cb44e708556a21cb1a6d02f4ac8be6d14fa81b29d81b0d6b45ba5dddac10d51f9a8e0708612c0c975ba45893d6228331b6068ac3090b
+  metadata.gz: 3f6c788b8aa265951ac9dd7ad859659c8c695f57c0760e68b4f2d8683bd11070c78c714836a077e8c9cc161790e77899f0fdc83823b4a2c9657725a6340d4536
+  data.tar.gz: 0ac353b46b92efc2a4fc887e0732d767017c313bd54c0ff8dddfdf8300ffa2eba386c3c15843aa4bbfd99b0c5c17d94616cdf36c162e02838bf118711023ae62

data/lib/aicli/helpers/completion.rb

@@ -4,7 +4,9 @@ module AiCli
   module Helpers
     module Completion
       EXPLAIN_IN_SECOND_REQUEST = true
-      SHELL_CODE_EXCLUSIONS = [/```[a-zA-Z]*\n?/i, /```[a-zA-Z]*/i, "\n"].freeze
+      # Opening fence must include a newline after optional language tag (e.g. ```zsh\n).
+      # Using \n? allowed ``` alone to match while streaming, leaking "zsh" into "zshfind".
+      SHELL_CODE_EXCLUSIONS = [/```[a-zA-Z]*\n/i, /```[a-zA-Z]*/i, "\n"].freeze
 
       module_function
 
@@ -184,7 +186,15 @@ module AiCli
           handle_ruby_llm_error(e)
         end
 
-        data
+        sanitize_script(data)
+      end
+
+      def sanitize_script(script)
+        text = script.to_s.strip
+        return text if text.empty?
+
+        shell = Regexp.escape(OsDetect.detect_shell)
+        text.sub(/\A#{shell}/i, '')
       end
 
       def handle_ruby_llm_error(error)
@@ -225,7 +235,7 @@ module AiCli
           #{message}
         MSG
       end
-      private_class_method :stream_to_writer, :handle_ruby_llm_error
+      private_class_method :stream_to_writer, :handle_ruby_llm_error, :sanitize_script
     end
   end
 end

data/lib/aicli/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: aicli
 version: !ruby/object:Gem::Version
-  version: 0.2.9
+  version: 0.3.0
 platform: ruby
 authors:
 - Antonio Molinari