better complex parameter support - still temporary untill full support comes to RubyLLM

Author: patvice

lib/ruby_llm/mcp.rb

@@ -19,6 +19,7 @@ def client(*args, **kwargs)
     def support_complex_parameters!
       require_relative "mcp/providers/open_ai/complex_parameter_support"
       require_relative "mcp/providers/anthropic/complex_parameter_support"
+      require_relative "mcp/providers/gemini/complex_parameter_support"
     end
   end
 end

lib/ruby_llm/mcp/providers/anthropic/complex_parameter_support.rb

@@ -5,20 +5,36 @@ module MCP
     module Providers
       module Anthropic
         module ComplexParameterSupport
+          module_function
+
           def clean_parameters(parameters)
             parameters.transform_values do |param|
-              format = {
-                type: param.type,
-                description: param.description
-              }.compact
+              build_properties(param).compact
+            end
+          end
 
-              if param.type == "array"
-                format[:items] = param.items
-              elsif param.type == "object"
-                format[:properties] = clean_parameters(param.properties)
-              end
+          def required_parameters(parameters)
+            parameters.select { |_, param| param.required }.keys
+          end
 
-              format
+          def build_properties(param)
+            case param.type
+            when :array
+              {
+                type: param.type,
+                items: { type: param.item_type }
+              }
+            when :object
+              {
+                type: param.type,
+                properties: clean_parameters(param.properties),
+                required: required_parameters(param.properties)
+              }
+            else
+              {
+                type: param.type,
+                description: param.description
+              }
             end
           end
         end

lib/ruby_llm/mcp/providers/open_ai/complex_parameter_support.rb

@@ -5,18 +5,29 @@ module MCP
     module Providers
       module OpenAI
         module ComplexParameterSupport
+          module_function
+
           def param_schema(param)
-            format = {
-              type: param.type,
-              description: param.description
-            }.compact
+            properties = case param.type
+                         when :array
+                           {
+                             type: param.type,
+                             items: { type: param.item_type }
+                           }
+                         when :object
+                           {
+                             type: param.type,
+                             properties: param.properties.transform_values { |value| param_schema(value) },
+                             required: param.properties.select { |_, p| p.required }.keys
+                           }
+                         else
+                           {
+                             type: param.type,
+                             description: param.description
+                           }
+                         end
 
-            if param.type == "array"
-              format[:items] = param.items
-            elsif param.type == "object"
-              format[:properties] = param.properties.transform_values { |value| param_schema(value) }
-            end
-            format
+            properties.compact
           end
         end
       end