fix: enums should only coerce matching symbols into strings (#3)

Author: stainless-app[bot]

lib/openai/base_model.rb

@@ -55,11 +55,11 @@ def type_info(spec)
           type_info(spec.slice(:const, :enum, :union).first&.last)
         in Proc
           spec
-        in OpenAI::Converter | Class
+        in OpenAI::Converter | Class | Symbol
           -> { spec }
         in true | false
           -> { OpenAI::BooleanModel }
-        in NilClass | true | false | Symbol | Integer | Float
+        in NilClass | Integer | Float
           -> { spec.class }
         end
       end
@@ -82,6 +82,13 @@ def coerce(target, value)
         case target
         in OpenAI::Converter
           target.coerce(value)
+        in Symbol
+          case value
+          in Symbol | String if (val = value.to_sym) == target
+            val
+          else
+            value
+          end
         in Class
           case target
           in -> { _1 <= NilClass }
@@ -140,6 +147,13 @@ def try_strict_coerce(target, value)
         case target
         in OpenAI::Converter
           target.try_strict_coerce(value)
+        in Symbol
+          case value
+          in Symbol | String if (val = value.to_sym) == target
+            [true, val, 1]
+          else
+            [false, false, 0]
+          end
         in Class
           case [target, value]
           in [-> { _1 <= NilClass }, _]
@@ -367,7 +381,14 @@ class << self
       #
       # @return [Symbol, Object]
       #
-      def coerce(value) = (value.is_a?(String) ? value.to_sym : value)
+      def coerce(value)
+        case value
+        in Symbol | String if values.include?(val = value.to_sym)
+          val
+        else
+          value
+        end
+      end
 
       # @!parse
       #   # @private
@@ -388,7 +409,7 @@ def try_strict_coerce(value)
         return [true, value, 1] if values.include?(value)
 
         case value
-        in String if values.include?(val = value.to_sym)
+        in Symbol | String if values.include?(val = value.to_sym)
           [true, val, 1]
         else
           case [value, values.first]

test/openai/base_model_test.rb

@@ -22,7 +22,7 @@ def test_basic_coerce
     end
 
     assert_pattern do
-      OpenAI::Converter.coerce(A2, %w[a b c]) => [:a, :b, :c]
+      OpenAI::Converter.coerce(A2, %w[a b c]) => [:a, :b, "c"]
     end
   end
 
@@ -338,4 +338,18 @@ def test_basic_eql
     refute_equal(U1, U2)
     assert_equal(U1, U3)
   end
+
+  class U4 < OpenAI::Union
+    variant :a, const: :a
+    variant :b, const: :b
+  end
+
+  def test_basic_const_union
+    assert_pattern do
+      U4.coerce(nil) => nil
+      U4.coerce("") => ""
+      U4.coerce(:a) => :a
+      U4.coerce("a") => :a
+    end
+  end
 end