antigravity-sdk 0.1.4 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: afe9c665bf383ac5f2b9e0cf07ea94f085373c0c2a0d0b10c135a56a639e0eac
-  data.tar.gz: d2cf2720c77a9d6d854c8415c2113c13f3ca91c5aabd86d413cb39422e47cf31
+  metadata.gz: 6e2f8d1b432e31aa733976b69c28c01db1c0a8f7d109eb920e489d22ee0fa001
+  data.tar.gz: c1dec66ab20e10063913499f9811a8baf1aaf63bf1113d8008b4410e4a68b92a
 SHA512:
-  metadata.gz: b730d5312a05e57473a765d8398cc3e2b80edcbcf931bf802541ca72f0962b7600c6dbeac7761b4815da34caae823e1b1f6dacd993eb9550d156173aac797a39
-  data.tar.gz: 7cedb00e25e9e8d17d00a6e94588301b75f126c107e4a06fd145db7c87c08fb5d2ea8cc0f78aa3a39577263b5067e0bc7e84e0f4176ec35931a95d7bdbba36d0
+  metadata.gz: 3601db0d85ee97d1607f61b08091560382bb6ff648042c1bf5340c72c58bd5a3ba026bb79d152281d7a80bd1103542f6de19f86fa4a6a20076884db2cf718374
+  data.tar.gz: 457ba9d83a710200dfefa3d07b14a83a2349efcd19f46c2033addf550070bdf927e37f21a93109d36eeeb6dc84e536b829397315683049f00ef52b67f638961d

data/lib/antigravity/agent.rb

@@ -3,9 +3,9 @@
 module Antigravity
   class Agent
     attr_accessor :model, :system_instruction, :api_key
-    attr_reader :tools, :skills, :hooks, :sidecars, :client
+    attr_reader :tools, :skills, :hooks, :sidecars, :client, :logger_guard
 
-    def initialize(model: nil, &block)
+    def initialize(model: nil, auto_logger: true, &block)
       @model = model || Antigravity.config.default_model
       @api_key = Antigravity.config.api_key
       @system_instruction = nil
@@ -14,6 +14,12 @@ module Antigravity
       @sidecars = []
       @hooks = Hooks.new
       @client = Client.new
+      @logger_guard = nil
+
+      # Automagic Logger attachment unless disabled via ENV["ANTIGRAVITY_LOGGER"]=false or auto_logger: false
+      if auto_logger && logger_enabled?
+        attach_logger
+      end
 
       yield(self) if block_given?
     end
@@ -35,10 +41,10 @@ module Antigravity
       sidecar
     end
 
-    def attach_logger(log_target = "log/antigravity.log", level: ::Logger::INFO)
-      logger_guard = Guards::AgentLogger.new(log_target, level: level)
-      logger_guard.attach_to(self)
-      logger_guard
+    def attach_logger(log_target = nil, level: ::Logger::INFO, silent_notice: false)
+      @logger_guard = Guards::AgentLogger.new(log_target, level: level, silent_notice: silent_notice)
+      @logger_guard.attach_to(self)
+      @logger_guard
     end
 
     def load_skill(skill_path)
@@ -80,5 +86,14 @@ module Antigravity
       response
     end
     alias ask prompt
+
+    private
+
+    def logger_enabled?
+      env_val = ENV["ANTIGRAVITY_LOGGER"]&.downcase
+      return false if %w[false 0 none no].include?(env_val)
+
+      true
+    end
   end
 end

data/lib/antigravity/guards.rb

@@ -7,18 +7,29 @@ module Antigravity
   module Guards
     # Configurable opt-in logger guard supporting file loggers and Rails.logger
     class AgentLogger
-      attr_reader :logger
-
-      def initialize(log_target = "log/antigravity.log", level: Logger::INFO)
-        if log_target.is_a?(String)
-          FileUtils.mkdir_p(File.dirname(log_target)) rescue nil
-          @logger = ::Logger.new(log_target)
-        elsif log_target.respond_to?(:info)
-          @logger = log_target
+      attr_reader :logger, :target_description
+
+      def initialize(log_target = nil, level: Logger::INFO, silent_notice: false)
+        resolved_target = resolve_log_target(log_target)
+
+        if resolved_target.is_a?(String)
+          dir = File.dirname(resolved_target)
+          FileUtils.mkdir_p(dir) unless Dir.exist?(dir)
+          @logger = ::Logger.new(resolved_target)
+          @target_description = resolved_target
+        elsif resolved_target.respond_to?(:info)
+          @logger = resolved_target
+          @target_description = "Rails.logger"
         else
           @logger = ::Logger.new($stdout)
+          @target_description = "$stdout"
         end
+
         @logger.level = level if @logger.respond_to?(:level=)
+
+        unless silent_notice
+          puts "🪵 Logging to #{@target_description}"
+        end
       end
 
       def before_prompt(prompt_text)
@@ -45,6 +56,22 @@ module Antigravity
         agent.before_tool_call { |t, p| logger_guard.before_tool_call(t, p) }
         agent.after_tool_call { |t, p, r| logger_guard.after_tool_call(t, p, r) }
       end
+
+      private
+
+      def resolve_log_target(target)
+        return target if target
+
+        if defined?(Rails) && Rails.respond_to?(:logger) && Rails.logger
+          Rails.logger
+        elsif ENV["RAILS_ENV"] && !ENV["RAILS_ENV"].empty?
+          "log/#{ENV['RAILS_ENV']}.log"
+        elsif ENV["RACK_ENV"] && !ENV["RACK_ENV"].empty?
+          "log/#{ENV['RACK_ENV']}.log"
+        else
+          "log/antigravity.log"
+        end
+      end
     end
 
     # Configurable opt-in file protection guard

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: antigravity-sdk
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.5
 platform: ruby
 authors:
 - Riccardo Carlesso