smart_agent 0.2.6 → 0.2.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 352e9af56a4c21484309f623b893ce166561458624d9f9a6d9cfcabaff0a4766
-  data.tar.gz: 0ff30d3602f0e205c4ca335249203a41c95a6c6d4033847494bf64b7abfb5ff4
+  metadata.gz: ff8fd3192792e6b8cb6f4763f443d72525ddcec5ac8f3ea8b48dd8b1b46271ac
+  data.tar.gz: af335195cf3dfa97419c9a9b3eee058b9167c4b3e288b813283721efd97172ea
 SHA512:
-  metadata.gz: db49b780b5efc00c7a1a731338cbf5aca96dff969ea2740f44f4cea81926e772db957319ee222078de39011355e55ebf8c151323a4441f7e8f36e11c36ec769f
-  data.tar.gz: ac700a1c932f24d2be2f12fe43fe5ecf2707fbee821ce77a1caf7f081bb6a9fab7d6129f51a2e8753f4eea13988406be654b4d7c35b48ba752aa98a8be5d62cf
+  metadata.gz: 1287d8e23a9535e82f286c10209b5f01bf4d3a560c61e77aa0f64f3926332907c484e07f83a1986ba942580a92deeee19db9ed203d2896e5b0d1c32e97b02bec
+  data.tar.gz: a5afa33689f0bb2792915ff9414e3cd4fae54ce13254ba8490f58e1ec4910d1287edfcfd77bdb4b54f8f170279cb2720ba4be28d177d4f5abfefab2d4170e34b

data/lib/smart_agent/result.rb

@@ -1,3 +1,5 @@
+require "json"
+
 module SmartAgent
   class Result
     def initialize(response)
@@ -9,16 +11,10 @@ module SmartAgent
       if @response.class == String
         return false
       else
-        tool_calls = @response.dig("choices", 0, "message", "tool_calls")
-        if tool_calls
-          unless tool_calls.empty?
-            return tool_calls
-          else
-            return false
-          end            
-        else
-          return false
-        end
+        tool_calls = normalized_tool_calls
+        return false if tool_calls.empty?
+
+        return tool_calls
       end
     end
 
@@ -33,5 +29,59 @@ module SmartAgent
     def response
       @response
     end
+
+    private
+
+    def normalized_tool_calls
+      tool_calls = openai_tool_calls
+      return tool_calls if tool_calls && !tool_calls.empty?
+
+      tool_use_blocks.map { |block| normalize_tool_use(block) }.compact
+    end
+
+    def openai_tool_calls
+      @response.dig("choices", 0, "message", "tool_calls")
+    end
+
+    def tool_use_blocks
+      content_blocks.select { |block| block["type"] == "tool_use" }
+    end
+
+    def content_blocks
+      blocks = []
+      collect_content_blocks(@response, blocks)
+      blocks.select { |block| block.is_a?(Hash) }
+    end
+
+    def collect_content_blocks(value, blocks)
+      case value
+      when Hash
+        content = value["content"]
+        blocks.concat(content) if content.is_a?(Array)
+        value.each_value { |child| collect_content_blocks(child, blocks) }
+      when Array
+        value.each { |child| collect_content_blocks(child, blocks) }
+      end
+    end
+
+    def normalize_tool_use(block)
+      name = block["name"]
+      return nil unless name
+
+      {
+        "id" => block["id"],
+        "type" => "function",
+        "function" => {
+          "name" => name,
+          "arguments" => tool_arguments(block["input"]),
+        },
+      }
+    end
+
+    def tool_arguments(input)
+      return input if input.is_a?(String)
+
+      JSON.generate(input || {})
+    end
   end
 end

data/lib/smart_agent/version.rb

@@ -1,3 +1,3 @@
 module SmartAgent
-  VERSION = "0.2.6"
+  VERSION = "0.2.7"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: smart_agent
 version: !ruby/object:Gem::Version
-  version: 0.2.6
+  version: 0.2.7
 platform: ruby
 authors:
 - Zhuang Biaowei
@@ -15,14 +15,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.4.1
+        version: 0.4.2
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.4.1
+        version: 0.4.2
 - !ruby/object:Gem::Dependency
   name: mcp-sdk.rb
   requirement: !ruby/object:Gem::Requirement