OpenAI provider update. New types.

roseflow-ai · Nov 21, 2023 · 67d5cf4 · 67d5cf4
1 parent 7d5a009
commit 67d5cf4
Show file tree

Hide file tree

Showing 4 changed files with 32 additions and 0 deletions.
diff --git a/lib/roseflow/ai/provider.rb b/lib/roseflow/ai/provider.rb
@@ -18,6 +18,10 @@ def models
         @models ||= instance.models
       end
 
+      def get_file_content(file_id)
+        @instance.get_file_content(file_id)
+      end
+
       private
 
       attr_reader :instance

diff --git a/lib/roseflow/ai/providers/openai_adapter.rb b/lib/roseflow/ai/providers/openai_adapter.rb
@@ -9,6 +9,10 @@ class OpenAIAdapter < BaseAdapter
         def models
           @provider.models
         end
+
+        def get_file_content(file_id)
+          @provider.get_file_content(file_id)
+        end
       end
     end
   end

diff --git a/lib/roseflow/chat/message.rb b/lib/roseflow/chat/message.rb
@@ -30,10 +30,24 @@ class ModelMessage < Message
 
     class UserMessage < Message
       attribute :role, Types::String.constrained(included_in: %w(user))
+
+      def self.from(input)
+        new(
+          role: "user",
+          content: input
+        )
+      end
     end
 
     class SystemMessage < Message
       attribute :role, Types::String.constrained(included_in: %w(system))
+
+      def self.from(input)
+        new(
+          role: "system",
+          content: input
+        )
+      end
     end
   end
 end
diff --git a/lib/roseflow/types.rb b/lib/roseflow/types.rb
@@ -12,5 +12,15 @@ module OpenAI
     FunctionCallObject = Types::Hash
     StringOrObject = Types::String | FunctionCallObject
     StringOrArray = Types::String | Types::Array
+
+    VisionChatMessageContent = Types::Hash.schema(
+      text: Types::String,
+      type: Types::String.default("text"),
+    )
+
+    VisionImageMessageContent = Types::Hash.schema(
+      image_url: Types::Hash.schema(image_url: Types::String),
+      type: Types::String.default("image_url"),
+    )
   end
 end