rager 0.6.0 → 0.7.0

data/lib/rager/utils/http.rb

@@ -13,17 +13,13 @@ module Rager
     module Http
       extend T::Sig
 
-      sig { params(duration: Numeric).void }
-      def self.sleep(duration)
-        if defined?(Async::Task) && (task = Async::Task.current?)
-          task.sleep(duration)
-        else
-          Kernel.sleep(duration)
-        end
-      end
-
-      sig { params(url: String, http_adapter: T.nilable(Rager::Http::Adapters::Abstract)).returns(T.nilable(String)) }
-      def self.download_binary(url, http_adapter: nil)
+      sig {
+        params(
+          url: String,
+          http_adapter: T.nilable(Rager::Http::Adapters::Abstract)
+        ).returns(T.nilable(String))
+      }
+      def self.download(url, http_adapter: nil)
         adapter = http_adapter || Rager.config.http_adapter
         response = adapter.make_request(
           Rager::Http::Request.new(url: url, headers: {})
@@ -42,15 +38,77 @@ module Rager
         end
       end
 
-      sig { params(url: String, path: String, http_adapter: T.nilable(Rager::Http::Adapters::Abstract)).returns(T.nilable(String)) }
-      def self.download_file(url, path, http_adapter: nil)
-        content = download_binary(url, http_adapter: http_adapter)
+      sig {
+        params(
+          url: String,
+          path: String,
+          http_adapter: T.nilable(Rager::Http::Adapters::Abstract)
+        ).returns(T.nilable(String))
+      }
+      def self.download_to_file(url, path, http_adapter: nil)
+        content = download(url, http_adapter: http_adapter)
         return nil if content.nil?
 
         FileUtils.mkdir_p(File.dirname(path))
-        File.write(path, content)
+        File.write(path, content, mode: "wb")
         path
       end
+
+      sig do
+        params(
+          endpoint: String,
+          body: String,
+          error: T::Boolean
+        ).void
+      end
+      def self.log_remote(endpoint, body, error = false)
+        logger = Rager.config.logger
+
+        case Rager.config.log_strategy
+        when Rager::LogStrategy::Stdout
+          error ? logger.error(body) : logger.info(body)
+        when Rager::LogStrategy::Remote
+          http_adapter = Rager.config.http_adapter
+          url = Rager.config.url
+          api_key = Rager.config.api_key
+
+          unless url && api_key
+            raise Rager::Errors::CredentialsError.new("Rager Cloud", details: "Missing url or api_key for remote logging")
+          end
+
+          headers = {
+            "Content-Type" => "application/json",
+            "Authorization" => "Bearer #{api_key}"
+          }
+
+          request = Rager::Http::Request.new(
+            url: "#{url}/#{endpoint}",
+            verb: Rager::Http::Verb::Post,
+            headers: headers,
+            body: body
+          )
+
+          begin
+            response = http_adapter.make_request(request)
+
+            unless response.success?
+              raise Rager::Errors::HttpError.new(
+                http_adapter,
+                request.url,
+                response.status,
+                body: T.cast(response.body, String),
+                details: "HTTP request failed during remote logging"
+              )
+            end
+          rescue Rager::Error => e
+            if Rager.config.log_raise
+              raise e
+            else
+              Rager.config.logger.error(e.message)
+            end
+          end
+        end
+      end
     end
   end
 end

data/lib/rager/utils/replicate.rb

@@ -10,20 +10,33 @@ module Rager
     module Replicate
       extend T::Sig
 
-      sig { params(prediction_url: String, key: T.nilable(String), http_adapter: T.nilable(Rager::Http::Adapters::Abstract)).returns(T.nilable(String)) }
+      sig {
+        params(
+          prediction_url: String,
+          key: T.nilable(String),
+          http_adapter: T.nilable(Rager::Http::Adapters::Abstract)
+        ).returns(T.nilable(String))
+      }
       def self.download_prediction(prediction_url, key = nil, http_adapter: nil)
         download_url = get_download_url(prediction_url, key, http_adapter: http_adapter)
         return nil if download_url.nil?
 
-        Rager::Utils::Http.download_binary(download_url)
+        Rager::Utils::Http.download(download_url)
       end
 
-      sig { params(prediction_url: String, key: T.nilable(String), path: String, http_adapter: T.nilable(Rager::Http::Adapters::Abstract)).returns(T.nilable(String)) }
+      sig {
+        params(
+          prediction_url: String,
+          key: T.nilable(String),
+          path: String,
+          http_adapter: T.nilable(Rager::Http::Adapters::Abstract)
+        ).returns(T.nilable(String))
+      }
       def self.download_prediction_to_file(prediction_url, key = nil, path:, http_adapter: nil)
         download_url = get_download_url(prediction_url, key, http_adapter: http_adapter)
         return nil if download_url.nil?
 
-        Rager::Utils::Http.download_file(download_url, path)
+        Rager::Utils::Http.download_to_file(download_url, path)
       end
 
       sig { params(prediction_url: String, key: T.nilable(String), http_adapter: T.nilable(Rager::Http::Adapters::Abstract)).returns(T.nilable(String)) }
@@ -70,7 +83,16 @@ module Rager
         raise Rager::Errors::ParseError.new("Failed to parse prediction response", details: e.message)
       end
 
-      sig { params(prediction_url: String, key: T.nilable(String), path: T.nilable(String), max_attempts: Integer, sleep_interval: Integer, http_adapter: T.nilable(Rager::Http::Adapters::Abstract)).returns(T.nilable(String)) }
+      sig {
+        params(
+          prediction_url: String,
+          key: T.nilable(String),
+          path: T.nilable(String),
+          max_attempts: Integer,
+          sleep_interval: Integer,
+          http_adapter: T.nilable(Rager::Http::Adapters::Abstract)
+        ).returns(T.nilable(String))
+      }
       def self.poll_prediction(prediction_url, key: nil, path: nil, max_attempts: 30, sleep_interval: 10, http_adapter: nil)
         max_attempts.times do
           result = if path
@@ -79,7 +101,7 @@ module Rager
             download_prediction(prediction_url, key, http_adapter: http_adapter)
           end
           return result unless result.nil?
-          Rager::Utils::Http.sleep(sleep_interval)
+          Rager::Utils::Runtime.sleep(sleep_interval)
         end
 
         raise Rager::Errors::TimeoutError.new(

data/lib/rager/utils/runtime.rb

@@ -0,0 +1,21 @@
+# typed: strict
+# frozen_string_literal: true
+
+require "sorbet-runtime"
+
+module Rager
+  module Utils
+    module Runtime
+      extend T::Sig
+
+      sig { params(duration: Numeric).void }
+      def self.sleep(duration)
+        if defined?(Async::Task) && (task = Async::Task.current?)
+          task.sleep(duration)
+        else
+          Kernel.sleep(duration)
+        end
+      end
+    end
+  end
+end

data/lib/rager/version.rb

@@ -2,5 +2,5 @@
 # frozen_string_literal: true
 
 module Rager
-  VERSION = "0.6.0"
+  VERSION = "0.7.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rager
 version: !ruby/object:Gem::Version
-  version: 0.6.0
+  version: 0.7.0
 platform: ruby
 authors:
 - mvkvc
@@ -72,6 +72,7 @@ files:
 - lib/rager/chat/schema.rb
 - lib/rager/config.rb
 - lib/rager/context.rb
+- lib/rager/context_options.rb
 - lib/rager/embed/options.rb
 - lib/rager/embed/providers/abstract.rb
 - lib/rager/embed/providers/openai.rb
@@ -90,32 +91,37 @@ files:
 - lib/rager/http/request.rb
 - lib/rager/http/response.rb
 - lib/rager/http/verb.rb
-- lib/rager/image_gen/options.rb
-- lib/rager/image_gen/output_format.rb
-- lib/rager/image_gen/providers/abstract.rb
-- lib/rager/image_gen/providers/replicate.rb
-- lib/rager/logger.rb
-- lib/rager/mesh_gen/options.rb
-- lib/rager/mesh_gen/providers/abstract.rb
-- lib/rager/mesh_gen/providers/replicate.rb
+- lib/rager/image/options.rb
+- lib/rager/image/output_format.rb
+- lib/rager/image/providers/abstract.rb
+- lib/rager/image/providers/replicate.rb
+- lib/rager/log_strategy.rb
+- lib/rager/mesh/options.rb
+- lib/rager/mesh/providers/abstract.rb
+- lib/rager/mesh/providers/replicate.rb
 - lib/rager/operation.rb
 - lib/rager/options.rb
+- lib/rager/outcome.rb
 - lib/rager/providers.rb
+- lib/rager/rerank/input.rb
 - lib/rager/rerank/options.rb
-- lib/rager/rerank/output.rb
 - lib/rager/rerank/providers/abstract.rb
 - lib/rager/rerank/providers/cohere.rb
+- lib/rager/rerank/result.rb
 - lib/rager/result.rb
 - lib/rager/search/options.rb
-- lib/rager/search/output.rb
 - lib/rager/search/providers/abstract.rb
 - lib/rager/search/providers/jina.rb
+- lib/rager/search/result.rb
+- lib/rager/template/input.rb
 - lib/rager/template/options.rb
 - lib/rager/template/providers/abstract.rb
 - lib/rager/template/providers/erb.rb
+- lib/rager/template/providers/mustache.rb
 - lib/rager/types.rb
 - lib/rager/utils/http.rb
 - lib/rager/utils/replicate.rb
+- lib/rager/utils/runtime.rb
 - lib/rager/version.rb
 homepage: https://github.com/mvkvc/rager_rb
 licenses:
@@ -135,7 +141,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.6.7
+rubygems_version: 3.6.9
 specification_version: 4
 summary: Build continuously improving generative workflows.
 test_files: []

data/lib/rager/rerank/output.rb

@@ -1,13 +0,0 @@
-# typed: strict
-# frozen_string_literal: true
-
-require "sorbet-runtime"
-
-module Rager
-  module Rerank
-    class Output < T::Struct
-      const :scores, T::Array[Float]
-      const :documents, T::Array[String]
-    end
-  end
-end

data/lib/rager/search/output.rb

@@ -1,14 +0,0 @@
-# typed: strict
-# frozen_string_literal: true
-
-require "sorbet-runtime"
-
-module Rager
-  module Search
-    class Output < T::Struct
-      const :urls, T::Array[String]
-      const :titles, T::Array[String]
-      const :contents, T::Array[String]
-    end
-  end
-end