llm_gateway 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 98a42a5943a3b9797fa7ea4d782785e4932c65f7183eab6a452a86b2ca57e585
-  data.tar.gz: cbe9e03dc5d76ebfdc8ba911b6c077b803113922f3c470c67fdadcac1b779ff8
+  metadata.gz: 21b3998df57de474c78626d8267db572418f80426ddaf82730cf8738e181d96c
+  data.tar.gz: 9a37ff5a3907a8b0d48dbd0fc83be2e50cd5757a60cacd051e93ff1dc63d734e
 SHA512:
-  metadata.gz: f58cb7936e3dcc9d643c1747f2d7404b2c1ca7474ad444f31178ce403351fba5ba1c981f91fabd1971097e7cb639b17709c9b62be9f784444d47d516801404b4
-  data.tar.gz: 45de5da177dbdd30d1490d1f5f2115a5b5d4e7adeb213d5a4a9c874bf99ed84d557e0e9988a77194bdcac2c491eb5903e4b66659d985284032bb6bfccba6cd1f
+  metadata.gz: 1be591a45a6fbee0b89846c679e0a9709e3a5493757eac3196be8194f8e592615b177c15f20d6763cf6a444bf8dbf3d899cf6e9fd9b11de7e2f02845f53ab1ff
+  data.tar.gz: ed5a981d3e7ff311d26fe4924760e286e7370da598464e4b00ad004c940015432d873be73afae7067d5fa659adbe7af701b5b6f4163432c10cdf7f950532d690

data/CHANGELOG.md

@@ -6,6 +6,7 @@
 
 **Merged pull requests:**
 
+- feat: add tool base class [\#4](https://github.com/Hyper-Unearthing/llm_gateway/pull/4) ([billybonks](https://github.com/billybonks))
 - feat: add prompt base class [\#3](https://github.com/Hyper-Unearthing/llm_gateway/pull/3) ([billybonks](https://github.com/billybonks))
 - lint files and add coverage [\#2](https://github.com/Hyper-Unearthing/llm_gateway/pull/2) ([billybonks](https://github.com/billybonks))
 - test: vcr lookup was not working when using different commands [\#1](https://github.com/Hyper-Unearthing/llm_gateway/pull/1) ([billybonks](https://github.com/billybonks))

data/README.md

@@ -90,6 +90,24 @@ result = teacher_prompt.run
 You can combine the Prompt class with tools for more complex interactions:
 
 ```ruby
+# Define a tool class
+class GetWeatherTool < LlmGateway::Tool
+  name 'get_weather'
+  description 'Get current weather for a location'
+  input_schema({
+    type: 'object',
+    properties: {
+      location: { type: 'string', description: 'City name' }
+    },
+    required: ['location']
+  })
+
+  def execute(input, login = nil)
+    # Your weather API implementation here
+    "The weather in #{input['location']} is sunny and 25°C"
+  end
+end
+
 class WeatherAssistantPrompt < LlmGateway::Prompt
   def initialize(model, location)
     super(model)
@@ -105,17 +123,7 @@ class WeatherAssistantPrompt < LlmGateway::Prompt
   end
 
   def tools
-    [{
-      name: 'get_weather',
-      description: 'Get current weather for a location',
-      input_schema: {
-        type: 'object',
-        properties: {
-          location: { type: 'string', description: 'City name' }
-        },
-        required: ['location']
-      }
-    }]
+    [GetWeatherTool]
   end
 end
 
@@ -127,7 +135,25 @@ result = weather_prompt.run
 ### Tool Usage (Function Calling)
 
 ```ruby
-# Define a tool
+# Define a tool class
+class GetWeatherTool < LlmGateway::Tool
+  name 'get_weather'
+  description 'Get current weather for a location'
+  input_schema({
+    type: 'object',
+    properties: {
+      location: { type: 'string', description: 'City name' }
+    },
+    required: ['location']
+  })
+
+  def execute(input, login = nil)
+    # Your weather API implementation here
+    "The weather in #{input['location']} is sunny and 25°C"
+  end
+end
+
+# Use the tool
 weather_tool = {
   name: 'get_weather',
   description: 'Get current weather for a location',
@@ -140,7 +166,6 @@ weather_tool = {
   }
 }
 
-# Use the tool
 result = LlmGateway::Client.chat(
   'claude-sonnet-4-20250514',
   'What\'s the weather in Singapore?',
@@ -155,8 +180,21 @@ result = LlmGateway::Client.chat(
 class WeatherAssistant
   def initialize
     @transcript = []
+    @weather_tool = {
+      name: 'get_weather',
+      description: 'Get current weather for a location',
+      input_schema: {
+        type: 'object',
+        properties: {
+          location: { type: 'string', description: 'City name' }
+        },
+        required: ['location']
+      }
+    }
   end
 
+  attr_reader :weather_tool
+
   def process_message(content)
     # Add user message to transcript
     @transcript << { role: 'user', content: [{ type: 'text', text: content }] }
@@ -164,7 +202,7 @@ class WeatherAssistant
     result = LlmGateway::Client.chat(
       'claude-sonnet-4-20250514',
       @transcript,
-      tools: [weather_tool],
+      tools: [@weather_tool],
       system: 'You are a helpful weather assistant.'
     )
 
@@ -195,7 +233,7 @@ class WeatherAssistant
         follow_up = LlmGateway::Client.chat(
           'claude-sonnet-4-20250514',
           @transcript,
-          tools: [weather_tool],
+          tools: [@weather_tool],
           system: 'You are a helpful weather assistant.'
         )
 

data/lib/llm_gateway/tool.rb

@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+module LlmGateway
+  class Tool
+      def initialize(*args)
+        # Empty constructor to allow subclasses to call super
+      end
+
+      def self.name(value = nil)
+        @name = value if value
+        @name
+      end
+
+      def self.description(value = nil)
+        @description = value if value
+        @description
+      end
+
+      def self.input_schema(value = nil)
+        @input_schema = value if value
+        @input_schema
+      end
+
+      def self.cache(value = nil)
+        @cache = value if value
+        @cache
+      end
+
+      def self.definition
+        {
+          name: @name,
+          description: @description,
+          input_schema: @input_schema,
+          cache_control: @cache ? { type: "ephemeral" } : nil
+        }.compact
+      end
+
+      def self.tool_name
+        definition[:name]
+      end
+
+      def execute(input, login)
+        raise NotImplementedError, "Subclasses must implement execute"
+      end
+  end
+end

data/lib/llm_gateway/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module LlmGateway
-  VERSION = "0.1.2"
+  VERSION = "0.1.3"
 end

data/lib/llm_gateway.rb

@@ -7,6 +7,7 @@ require_relative "llm_gateway/fluent_mapper"
 require_relative "llm_gateway/base_client"
 require_relative "llm_gateway/client"
 require_relative "llm_gateway/prompt"
+require_relative "llm_gateway/tool"
 
 # Load adapters - order matters for inheritance
 require_relative "llm_gateway/adapters/claude/client"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: llm_gateway
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - billybonks
@@ -40,6 +40,7 @@ files:
 - lib/llm_gateway/errors.rb
 - lib/llm_gateway/fluent_mapper.rb
 - lib/llm_gateway/prompt.rb
+- lib/llm_gateway/tool.rb
 - lib/llm_gateway/utils.rb
 - lib/llm_gateway/version.rb
 - sig/llm_gateway.rbs