geminai 1.0.0 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 182008113410d8953e5d9dbb32d0b3d31e400f0b64dc6de753bd2b0a0ecd0e3b
-  data.tar.gz: 5e4f6e4c304a109ec19e1bb00e98d7c60cd27b3f9c36d88fd51721aef9f424a2
+  metadata.gz: 87a7d8da785e91c6871f9f108570fb01f3165530ec932629776b9e61e7bc994f
+  data.tar.gz: 90235d17777b4c869e5523171c8deeaf961b9727a9204f76caefe594f93ff24e
 SHA512:
-  metadata.gz: f1c57d71925d784ff071a79f207b2f5fff3615774929b39a1dced79408c05dbb2bebbc561edc1a6b7b725d394bd9082c0590184ffe131b118d8d6c8edd02143d
-  data.tar.gz: bb3cf7288bfba9404ea343f75f931ba31b75154a17db76d435703d09acc989eaf7b586d39a8e22f7c5458ac2ebbcde3ad9144701e5a2074100366849b6001e29
+  metadata.gz: 3baba4a0640a4e1d30300e502b01bf2edfa1797230c39c72615b706b107eb1f22d4018b7a15ad766c85b6bce852e92beaa84cd2df43d577a05c5791cf6704fb3
+  data.tar.gz: f0659f0c2a44b6b6115b7114020c82b72560ce0882382a5bb6b8ad5efc2f2229ff2f26b4f183bcb06827411f859771f270d91ff330214cfc8fe280f9b7d4b26f

data/lib/geminai/client.rb

@@ -42,6 +42,16 @@ module Geminai
         body[:response_format][:image_size] = options.delete(:image_size) if options.key?(:image_size)
       end
 
+      # Handle schema shortcut option
+      schema = options.delete(:schema)
+      if schema
+        body[:response_format] = {
+          type: "text",
+          mime_type: "application/json",
+          schema: SchemaBuilder.build(schema)
+        }
+      end
+
       if (voice = options.delete(:voice))
         body[:generation_config] = {speech_config: [{voice: voice}]}
       end

data/lib/geminai/schema_builder.rb

@@ -0,0 +1,49 @@
+module Geminai
+  module SchemaBuilder
+    def self.build(schema)
+      case schema
+      when Symbol, String
+        { type: normalize_type(schema) }
+      when Array
+        { type: "array", items: build(schema.first) }
+      when Hash
+        if schema.key?(:type) || schema.key?("type")
+          normalized = schema.transform_keys(&:to_sym)
+          normalized[:properties] = normalized[:properties].transform_values { |v| build(v) } if normalized[:properties]
+          normalized[:items] = build(normalized[:items]) if normalized[:items]
+          normalized
+        else
+          properties = {}
+          required = []
+          schema.each do |key, val|
+            properties[key] = build(val)
+            required << key.to_s
+          end
+          {
+            type: "object",
+            properties: properties,
+            required: required
+          }
+        end
+      else
+        raise ArgumentError, "Unsupported schema format: #{schema.inspect}"
+      end
+    end
+
+    def self.normalize_type(type)
+      type = type.to_s.downcase
+      case type
+      when "str", "string"
+        "string"
+      when "int", "integer"
+        "integer"
+      when "float", "number"
+        "number"
+      when "bool", "boolean"
+        "boolean"
+      else
+        type
+      end
+    end
+  end
+end

data/lib/geminai/version.rb

@@ -1,3 +1,3 @@
 module Geminai
-  VERSION = "1.0.0"
+  VERSION = "1.1.0"
 end

data/lib/geminai.rb

@@ -3,6 +3,7 @@ require_relative "geminai/step"
 require_relative "geminai/grounding_metadata"
 require_relative "geminai/interaction"
 require_relative "geminai/client"
+require_relative "geminai/schema_builder"
 
 module Geminai
   # Helper to construct a client

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: geminai
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.1.0
 platform: ruby
 authors:
 - swlkr
@@ -20,6 +20,7 @@ files:
 - lib/geminai/client.rb
 - lib/geminai/grounding_metadata.rb
 - lib/geminai/interaction.rb
+- lib/geminai/schema_builder.rb
 - lib/geminai/step.rb
 - lib/geminai/version.rb
 homepage: https://github.com/swlkr/geminai