lib/roseflow/openai/response.rb in roseflow-openai-0.1.0 vs lib/roseflow/openai/response.rb in roseflow-openai-0.2.0

@@ -2,15 +2,10 @@
 
 require "dry-struct"
 require "roseflow/types"
 require "roseflow/openai/embedding"
 
-module Types
-  include Dry.Types()
-  Number = Types::Integer | Types::Float
-end
-
 module Roseflow
   module OpenAI
     FailedToCreateEmbeddingError = Class.new(StandardError)
 
     class ApiResponse
@@ -140,10 +135,10 @@
 
     class OpenAIEmbedding < Dry::Struct
       transform_keys(&:to_sym)
 
       attribute :object, Types::String.default("embedding")
-      attribute :embedding, Types::Array(Types::Number)
+      attribute :embedding, Types::Array(::Types::Number)
       attribute :index, Types::Integer
     end # OpenAIEmbedding
 
     class EmbeddingApiResponseBody < Dry::Struct
       transform_keys(&:to_sym)