antigravity-sdk 0.1.9 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 961fa879bafeafa5995f2b1c575aaed95d3cdac823cc0e3141584de6cf2071cb
-  data.tar.gz: 92dc423201619f680ab97f9a0033fc8b1ebc0890e6ff51a6091da1036bfca9ca
+  metadata.gz: 665ab001fa484b0bc5ba97c2e7feb960712b0f8e5070ed521f0caf5ad0f16a58
+  data.tar.gz: 3f556321b1083daadea10792632ac216ba168abb26708fc5cc3a81824301d460
 SHA512:
-  metadata.gz: 5039380738224e6ec92ee23f1e6034f1fa529f690c95e276b581f4764c56292bf33a00272cb73c9fda9797aae21edf4999c23ab75c0a69975032cf10223a1c4d
-  data.tar.gz: b059c8e931c133e4decdaddfb9f197ee9627f53575927c962b659f4ed9aa71a9d26ba0fa58f0176cd98f1deb7195ed0d10dd1ffb324f17835e51e5b007aead6e
+  metadata.gz: d60ff58a66fec4ed17a85ad51015568e743363f313eb1a27b9e6512a767ae5a7abca062796ae0bc19edbe59545202828415b693f6551b77a22d8a1ff5cca99ea
+  data.tar.gz: 9e09ed8b23e59d29f644d10f421b73d70f357c7d9fd601ba67ba429df8c2814768d0264084e5e6439cd9161d2f8c13313f6890bcd11faf6f66f7e43e3ea2578f

data/lib/antigravity/agent.rb

@@ -1,18 +1,11 @@
 # frozen_string_literal: true
 
 module Antigravity
-  class Agent
+  class Agent < Base
+
     attr_accessor :model, :system_instruction, :api_key
     attr_reader :tools, :skills, :hooks, :sidecars, :client, :logger_guard
 
-    def self.emoji
-      Antigravity.emoji(:gem)
-    end
-
-    def emoji
-      self.class.emoji
-    end
-
     def initialize(model: nil, auto_logger: true, &block)
       @model = model || Antigravity.config.default_model
       @api_key = Antigravity.config.api_key

data/lib/antigravity/base.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module Antigravity
+  # Base class for all Antigravity domain objects.
+  # Subclasses automagically get .emoji / #emoji via Emojifiable.
+  #
+  # ⚠️  Keep this class SUPER thin!
+  # Every SDK class inherits from it, so any weight here
+  # is carried by Agent, Tool, Skill, Message, Sidecar::Runner, etc.
+  class Base
+    include Emojifiable
+
+    def self.inherited(subclass)
+      super
+      subclass.include(Emojifiable) unless subclass < Emojifiable
+    end
+  end
+end

data/lib/antigravity/emojis.rb

@@ -3,6 +3,7 @@
 module Antigravity
   EMOJIS = {
     gem: "💎",
+    agent: "🕵️‍♂️",
     prompt: "💬",
     response: "🤖",
     thinking: "🤔",
@@ -12,13 +13,73 @@ module Antigravity
     sidecar: "🚗",
     logger: "🪵",
     skill: "📁",
+    message: "💬",
+    guard: "🛡️",
     test: "🧪",
-    success: "✅"
+    success: "✅",
+    unknown: "🤷"
+  }.freeze
+
+  # Name fragments we recognise in the class hierarchy
+  EMOJI_CLASS_MAP = {
+    "Agent" => :agent,
+    "Tool" => :tool,
+    "Sidecar" => :sidecar,
+    "Runner" => :sidecar,
+    "Skill" => :skill,
+    "Message" => :message,
+    "Chunk" => :message,
+    "Guard" => :guard,
+    "Logger" => :logger
   }.freeze
 
   class << self
     def emoji(key)
-      EMOJIS[key.to_sym] || "💎"
+      EMOJIS[key.to_sym] || EMOJIS[:unknown]
+    end
+
+    def emoji_for(target)
+      key = resolve_emoji_key(target)
+      emoji(key)
+    end
+
+    private
+
+    def resolve_emoji_key(target)
+      # Message/Chunk: role-aware
+      if target.is_a?(Message) || (defined?(Chunk) && target.is_a?(Chunk))
+        return target.respond_to?(:role) && target.role == :user ? :prompt : :response
+      end
+
+      # Direct symbol/string lookup
+      return target if target.is_a?(Symbol) || target.is_a?(String)
+
+      # Class hierarchy reflection
+      klass = target.is_a?(Class) || target.is_a?(Module) ? target : target.class
+      parts = klass.name&.split("::") || []
+      parts.reverse_each do |part|
+        return EMOJI_CLASS_MAP[part] if EMOJI_CLASS_MAP.key?(part)
+      end
+
+      :unknown
+    end
+  end
+
+  # Mixin providing polymorphic .emoji class and #emoji instance methods.
+  # Auto-included by Antigravity::Base via inherited hook.
+  module Emojifiable
+    def emoji
+      Antigravity.emoji_for(self)
+    end
+
+    module ClassMethods
+      def emoji
+        Antigravity.emoji_for(self)
+      end
+    end
+
+    def self.included(base)
+      base.extend(ClassMethods)
     end
   end
 end

data/lib/antigravity/message.rb

@@ -1,16 +1,9 @@
 # frozen_string_literal: true
 
 module Antigravity
-  class Message
-    attr_accessor :role, :content, :thinking, :tool_calls, :model_id, :tokens
-
-    def self.emoji
-      Antigravity.emoji(:response)
-    end
+  class Message < Base
 
-    def emoji
-      role == :user ? Antigravity.emoji(:prompt) : Antigravity.emoji(:response)
-    end
+    attr_accessor :role, :content, :thinking, :tool_calls, :model_id, :tokens
 
     def initialize(role: :assistant, content: "", thinking: "", tool_calls: [], model_id: nil)
       @role = role

data/lib/antigravity/sidecar.rb

@@ -6,17 +6,11 @@ require "fileutils"
 
 module Antigravity
   module Sidecar
-    # Base class for non-blocking agent sidecars
-    class Base
-      attr_reader :name, :queue, :worker_thread
-
-      def self.emoji
-        Antigravity.emoji(:sidecar)
-      end
+    # Abstract runner for non-blocking agent sidecars.
+    # Subclass and implement #process_event.
+    class Runner < Antigravity::Base
 
-      def emoji
-        self.class.emoji
-      end
+      attr_reader :name, :queue, :worker_thread
 
       def initialize(name = "SidecarWorker")
         @name = name
@@ -58,7 +52,7 @@ module Antigravity
     end
 
     # Async Audit Logger Sidecar
-    class AuditLogger < Base
+    class AuditLogger < Runner
       attr_reader :log_file
 
       def initialize(log_file = "log/agent_audit.jsonl")
@@ -78,7 +72,7 @@ module Antigravity
     end
 
     # Async Vulnerability & Code Quality Scanner Sidecar
-    class VulnerabilityScanner < Base
+    class VulnerabilityScanner < Runner
       attr_reader :scanned_events
 
       def initialize
@@ -96,5 +90,7 @@ module Antigravity
         @scanned_events << { tool: tool_name, params: params, verified: true }
       end
     end
+    # Backward-compatible alias
+    Base = Runner
   end
 end

data/lib/antigravity/skill.rb

@@ -3,16 +3,9 @@
 require "yaml"
 
 module Antigravity
-  class Skill
-    attr_reader :name, :description, :instructions, :path
-
-    def self.emoji
-      Antigravity.emoji(:skill)
-    end
+  class Skill < Base
 
-    def emoji
-      self.class.emoji
-    end
+    attr_reader :name, :description, :instructions, :path
 
     def initialize(path)
       @path = File.expand_path(path)

data/lib/antigravity/tool.rb

@@ -1,22 +1,15 @@
 # frozen_string_literal: true
 
 module Antigravity
-  class Tool
+  class Tool < Base
     class << self
       def inherited(subclass)
+        super
         subclass.extend(ClassMethods)
       end
-
-      def emoji
-        Antigravity.emoji(:tool)
-      end
     end
 
     module ClassMethods
-      def emoji
-        Antigravity.emoji(:tool)
-      end
-
       def name(val = nil, desc: nil)
         if val
           @tool_name = val.to_s
@@ -69,10 +62,6 @@ module Antigravity
       end
     end
 
-    def emoji
-      self.class.emoji
-    end
-
     def tool_name
       self.class.name
     end

data/lib/antigravity.rb

@@ -8,6 +8,7 @@ end
 
 require_relative "antigravity/version"
 require_relative "antigravity/emojis"
+require_relative "antigravity/base"
 require_relative "antigravity/config"
 require_relative "antigravity/message"
 require_relative "antigravity/harness"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: antigravity-sdk
 version: !ruby/object:Gem::Version
-  version: 0.1.9
+  version: 0.3.0
 platform: ruby
 authors:
 - Riccardo Carlesso
@@ -61,6 +61,7 @@ extra_rdoc_files: []
 files:
 - lib/antigravity.rb
 - lib/antigravity/agent.rb
+- lib/antigravity/base.rb
 - lib/antigravity/client.rb
 - lib/antigravity/config.rb
 - lib/antigravity/emojis.rb