Completion method to OpenRouter adapter.

roseflow-ai · Aug 3, 2023 · e5770f7 · e5770f7
1 parent a278995
commit e5770f7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/lib/roseflow/ai/models/openrouter_adapter.rb b/lib/roseflow/ai/models/openrouter_adapter.rb
@@ -16,6 +16,10 @@ def call(operation, options, &block)
         def chat(options, &block)
           @model.chat(options.delete(:messages), options, &block)
         end
+
+        def completion(options, &block)
+          @model.completion(options, &block)
+        end
       end
     end
   end

diff --git a/lib/roseflow/embeddings/embedding.rb b/lib/roseflow/embeddings/embedding.rb
@@ -19,7 +19,6 @@ def initialize(**kwargs)
       def call
         raise ArgumentError, "An input must be provided" unless @input
         raise EmbeddingModelNotSpecifiedError, "An embedding model must be specified" unless @model
-        # response = @model.call(:embedding, model: @model.name, input: @input)
         response = @model.embed(input: @input)
         embedding = response.embedding
         @vector = Primitives::Vector.new(values: embedding.vector, dimensions: embedding.length)