antigravity-sdk 0.3.0 → 0.5.0

data/lib/antigravity/tool.rb

@@ -7,6 +7,12 @@ module Antigravity
         super
         subclass.extend(ClassMethods)
       end
+
+      # Factory method: define a tool inline with a block.
+      #   Tool.define(:name, desc: '...', params: { city: { type: :string } }) { |city:| ... }
+      def define(name, desc: '', params: {}, &block)
+        Dynamic.new(name, description: desc, params: params, &block)
+      end
     end
 
     module ClassMethods
@@ -72,12 +78,13 @@ module Antigravity
 
     # Factory for dynamic Proc-based tools
     class Dynamic < Tool
-      attr_reader :block
+      attr_reader :block, :params
 
-      def initialize(name, description: "", &block)
+      def initialize(name, description: '', params: {}, &block)
         super()
         @dynamic_name = name.to_s
         @dynamic_description = description
+        @params = params
         @block = block
       end
 
@@ -85,17 +92,37 @@ module Antigravity
         @dynamic_name
       end
 
-      def call(params)
-        @block.call(params)
+      def call(params = nil, **kwargs)
+        return nil unless @block
+
+        if params && kwargs.empty?
+          # Old-style: tool.call({key: value})
+          @block.call(params)
+        else
+          # New-style: tool.call(key: value)
+          @block.call(**kwargs)
+        end
       end
 
       def to_json_schema
+        properties = {}
+        required_params = []
+
+        @params.each do |param_name, spec|
+          properties[param_name] = {
+            type: (spec[:type] || :string).to_s,
+            description: spec[:description] || spec[:desc] || ''
+          }
+          required_params << param_name.to_s unless spec[:required] == false
+        end
+
         {
           name: @dynamic_name,
           description: @dynamic_description,
           parameters: {
-            type: "object",
-            properties: {}
+            type: 'object',
+            properties: properties,
+            required: required_params
           }
         }
       end

data/lib/antigravity/tool_runner.rb

@@ -0,0 +1,59 @@
+# frozen_string_literal: true
+
+require 'json'
+
+module Antigravity
+  # Registers, resolves, and executes custom tools for agent callbacks.
+  # Tools are invoked by the harness when the model decides to call them.
+  class ToolRunner < Base
+    def initialize
+      @tools = {}
+    end
+
+    # Register a tool (Tool::Dynamic or any Tool subclass)
+    def register(tool)
+      name = tool.respond_to?(:tool_name) ? tool.tool_name : tool.to_s
+      raise ToolError, "Tool '#{name}' is already registered" if @tools.key?(name)
+
+      @tools[name] = tool
+    end
+
+    # Check if a tool is registered
+    def registered?(name)
+      @tools.key?(name.to_s)
+    end
+
+    # Execute a tool by name with keyword args.
+    # Returns the result or a { error: "..." } hash on failure.
+    def execute(name, **kwargs)
+      tool = @tools[name.to_s]
+      raise ToolNotFoundError, "Tool '#{name}' not found" unless tool
+
+      tool.call(**kwargs)
+    rescue ToolNotFoundError
+      raise
+    rescue => e
+      { error: e.message }
+    end
+
+    # Generate harness-compatible tool definitions for HarnessConfig
+    def to_harness_tools
+      @tools.values.map do |tool|
+        schema = tool.to_json_schema
+        {
+          name: schema[:name],
+          description: schema[:description],
+          parametersJsonSchema: JSON.generate(schema[:parameters])
+        }
+      end
+    end
+
+    def size
+      @tools.size
+    end
+
+    def empty?
+      @tools.empty?
+    end
+  end
+end

data/lib/antigravity.rb

@@ -9,15 +9,26 @@ end
 require_relative "antigravity/version"
 require_relative "antigravity/emojis"
 require_relative "antigravity/base"
+require_relative "antigravity/errors"
 require_relative "antigravity/config"
 require_relative "antigravity/message"
+require_relative "antigravity/protocol"
 require_relative "antigravity/harness"
+require_relative "antigravity/policy"
 require_relative "antigravity/hooks"
 require_relative "antigravity/guards"
 require_relative "antigravity/sidecar"
 require_relative "antigravity/tool"
+require_relative "antigravity/tool_runner"
+require_relative "antigravity/skill_resolver"
 require_relative "antigravity/skill"
 require_relative "antigravity/client"
+require_relative "antigravity/connection/binary_fetcher"
+require_relative "antigravity/connection/websocket_client"
+require_relative "antigravity/connection/local_connection"
+require_relative "antigravity/conversation"
+require_relative "antigravity/colors"
+require_relative "antigravity/lifecycle_logger"
 require_relative "antigravity/agent"
 
 module Antigravity

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: antigravity-sdk
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.5.0
 platform: ruby
 authors:
 - Riccardo Carlesso
@@ -10,19 +10,19 @@ cert_chain: []
 date: 1980-01-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: faye-websocket
+  name: websocket
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.11'
+        version: '1.2'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.11'
+        version: '1.2'
 - !ruby/object:Gem::Dependency
   name: json
   requirement: !ruby/object:Gem::Requirement
@@ -63,15 +63,27 @@ files:
 - lib/antigravity/agent.rb
 - lib/antigravity/base.rb
 - lib/antigravity/client.rb
+- lib/antigravity/colors.rb
 - lib/antigravity/config.rb
+- lib/antigravity/connection/binary_fetcher.rb
+- lib/antigravity/connection/local_connection.rb
+- lib/antigravity/connection/websocket_client.rb
+- lib/antigravity/conversation.rb
 - lib/antigravity/emojis.rb
+- lib/antigravity/errors.rb
 - lib/antigravity/guards.rb
 - lib/antigravity/harness.rb
 - lib/antigravity/hooks.rb
+- lib/antigravity/lifecycle_logger.rb
 - lib/antigravity/message.rb
+- lib/antigravity/policy.rb
+- lib/antigravity/policy/constants.rb
+- lib/antigravity/protocol.rb
 - lib/antigravity/sidecar.rb
 - lib/antigravity/skill.rb
+- lib/antigravity/skill_resolver.rb
 - lib/antigravity/tool.rb
+- lib/antigravity/tool_runner.rb
 - lib/antigravity/version.rb
 homepage: https://github.com/palladius/antigravity-ruby-sdk
 licenses: