rager 0.5.0 → 0.7.0

data/lib/rager/chat.rb

@@ -1,35 +0,0 @@
-# typed: strict
-# frozen_string_literal: true
-
-require "sorbet-runtime"
-
-module Rager
-  module Chat
-    extend T::Sig
-
-    sig do
-      params(
-        messages: Rager::Types::ChatInput,
-        options: Rager::Chat::Options
-      ).returns(Rager::Types::ChatOutput)
-    end
-    def self.chat(messages, options = Rager::Chat::Options.new)
-      provider = get_provider(options.provider)
-      provider.chat(messages, options)
-    end
-
-    sig do
-      params(
-        key: String
-      ).returns(Rager::Chat::Providers::Abstract)
-    end
-    def self.get_provider(key)
-      case key.downcase
-      when "openai"
-        Rager::Chat::Providers::Openai.new
-      else
-        raise Rager::Errors::UnknownProviderError.new(Rager::Operation::Chat, key)
-      end
-    end
-  end
-end

data/lib/rager/embed.rb

@@ -1,35 +0,0 @@
-# typed: strict
-# frozen_string_literal: true
-
-require "sorbet-runtime"
-
-module Rager
-  module Embed
-    extend T::Sig
-
-    sig do
-      params(
-        text: Rager::Types::EmbedInput,
-        options: Rager::Embed::Options
-      ).returns(Rager::Types::EmbedOutput)
-    end
-    def self.embed(text, options = Rager::Embed::Options.new)
-      provider = get_provider(options.provider)
-      provider.embed(text, options)
-    end
-
-    sig do
-      params(
-        key: String
-      ).returns(Rager::Embed::Providers::Abstract)
-    end
-    def self.get_provider(key)
-      case key.downcase
-      when "openai"
-        Rager::Embed::Providers::Openai.new
-      else
-        raise Rager::Errors::UnknownProviderError.new(Rager::Operation::Embed, key)
-      end
-    end
-  end
-end

data/lib/rager/errors/missing_credentials_error.rb

@@ -1,19 +0,0 @@
-# typed: strict
-# frozen_string_literal: true
-
-require "sorbet-runtime"
-
-module Rager
-  module Errors
-    class MissingCredentialsError < Rager::Error
-      extend T::Sig
-
-      sig { params(provider_name: String, env_var: T.nilable(String)).void }
-      def initialize(provider_name, env_var = nil)
-        message = "Missing credentials for provider #{provider_name}"
-        message += " -- attempted lookup with #{env_var}" if env_var
-        super(message)
-      end
-    end
-  end
-end

data/lib/rager/errors/unknown_provider_error.rb

@@ -1,17 +0,0 @@
-# typed: strict
-# frozen_string_literal: true
-
-require "sorbet-runtime"
-
-module Rager
-  module Errors
-    class UnknownProviderError < Rager::Error
-      extend T::Sig
-
-      sig { params(operation: Rager::Operation, key: String).void }
-      def initialize(operation, key)
-        super("Unknown provider #{key} for operation #{operation.serialize}")
-      end
-    end
-  end
-end

data/lib/rager/image_gen.rb

@@ -1,31 +0,0 @@
-# typed: strict
-# frozen_string_literal: true
-
-require "sorbet-runtime"
-
-module Rager
-  module ImageGen
-    extend T::Sig
-
-    sig do
-      params(
-        prompt: Rager::Types::ImageGenInput,
-        options: Rager::ImageGen::Options
-      ).returns(Rager::Types::ImageGenOutput)
-    end
-    def self.image_gen(prompt, options = Rager::ImageGen::Options.new)
-      provider = get_provider(options.provider)
-      provider.image_gen(prompt, options)
-    end
-
-    sig { params(key: String).returns(Rager::ImageGen::Providers::Abstract) }
-    def self.get_provider(key)
-      case key.downcase
-      when "replicate"
-        Rager::ImageGen::Providers::Replicate.new
-      else
-        raise Rager::Errors::UnknownProviderError.new(Rager::Operation::ImageGen, key)
-      end
-    end
-  end
-end

data/lib/rager/mesh_gen.rb

@@ -1,31 +0,0 @@
-# typed: strict
-# frozen_string_literal: true
-
-require "sorbet-runtime"
-
-module Rager
-  module MeshGen
-    extend T::Sig
-
-    sig do
-      params(
-        image_url: Rager::Types::MeshGenInput,
-        options: Rager::MeshGen::Options
-      ).returns(Rager::Types::MeshGenOutput)
-    end
-    def self.mesh_gen(image_url, options = Rager::MeshGen::Options.new)
-      provider = get_provider(options.provider)
-      provider.mesh_gen(image_url, options)
-    end
-
-    sig { params(key: String).returns(Rager::MeshGen::Providers::Abstract) }
-    def self.get_provider(key)
-      case key.downcase
-      when "replicate"
-        Rager::MeshGen::Providers::Replicate.new
-      else
-        raise Rager::Errors::UnknownProviderError.new(Rager::Operation::MeshGen, key)
-      end
-    end
-  end
-end

data/lib/rager/rerank.rb

@@ -1,35 +0,0 @@
-# typed: strict
-# frozen_string_literal: true
-
-require "sorbet-runtime"
-
-module Rager
-  module Rerank
-    extend T::Sig
-
-    sig do
-      params(
-        query: Rager::Types::RerankInput,
-        options: Rager::Rerank::Options
-      ).returns(Rager::Types::RerankOutput)
-    end
-    def self.rerank(query, options = Rager::Rerank::Options.new)
-      provider = get_provider(options.provider)
-      provider.rerank(query, options)
-    end
-
-    sig do
-      params(
-        key: String
-      ).returns(Rager::Rerank::Providers::Abstract)
-    end
-    def self.get_provider(key)
-      case key.downcase
-      when "cohere"
-        Rager::Rerank::Providers::Cohere.new
-      else
-        raise Rager::Errors::UnknownProviderError.new(Rager::Operation::Rerank, key)
-      end
-    end
-  end
-end

data/lib/rager/search/providers/brave.rb

@@ -1,59 +0,0 @@
-# typed: strict
-# frozen_string_literal: true
-
-require "json"
-require "uri"
-require "sorbet-runtime"
-
-module Rager
-  module Search
-    module Providers
-      class Brave < Rager::Search::Providers::Abstract
-        extend T::Sig
-
-        sig do
-          override.params(
-            query: String,
-            options: Rager::Search::Options
-          ).returns(Rager::Types::SearchOutput)
-        end
-        def search(query, options)
-          api_key = options.api_key || ENV["BRAVE_API_KEY"]
-          raise Rager::Errors::MissingCredentialsError.new("Brave", "BRAVE_API_KEY") if api_key.nil?
-
-          params = {"q" => query}.tap do |p|
-            p["count"] = options.n.to_s if options.n
-          end
-
-          request = Rager::Http::Request.new(
-            verb: Rager::Http::Verb::Get,
-            url: "https://api.search.brave.com/res/v1/web/search?#{URI.encode_www_form(params)}",
-            headers: {
-              "Accept" => "application/json",
-              "Accept-Encoding" => "gzip",
-              "x-subscription-token" => api_key
-            }
-          )
-
-          response = Rager.config.http_adapter.make_request(request)
-          response_body = T.cast(T.must(response.body), String)
-
-          raise Rager::Errors::HttpError.new(Rager.config.http_adapter, response.status, response_body) if response.status != 200
-
-          parsed_response = JSON.parse(response_body)
-          web_results = parsed_response.dig("web", "results") || []
-
-          web_results.filter_map do |result|
-            next unless result["title"] && result["url"] && result["description"]
-
-            Rager::Search::Result.new(
-              title: result["title"],
-              url: result["url"],
-              description: result["description"]
-            )
-          end
-        end
-      end
-    end
-  end
-end

data/lib/rager/search.rb

@@ -1,35 +0,0 @@
-# typed: strict
-# frozen_string_literal: true
-
-require "sorbet-runtime"
-
-module Rager
-  module Search
-    extend T::Sig
-
-    sig do
-      params(
-        query: Rager::Types::SearchInput,
-        options: Rager::Search::Options
-      ).returns(Rager::Types::SearchOutput)
-    end
-    def self.search(query, options = Rager::Search::Options.new)
-      provider = get_provider(options.provider)
-      provider.search(query, options)
-    end
-
-    sig do
-      params(
-        key: String
-      ).returns(Rager::Search::Providers::Abstract)
-    end
-    def self.get_provider(key)
-      case key.downcase
-      when "brave"
-        Rager::Search::Providers::Brave.new
-      else
-        raise Rager::Errors::UnknownProviderError.new(Rager::Operation::Search, key)
-      end
-    end
-  end
-end

data/lib/rager/template.rb

@@ -1,35 +0,0 @@
-# typed: strict
-# frozen_string_literal: true
-
-require "sorbet-runtime"
-
-module Rager
-  module Template
-    extend T::Sig
-
-    sig do
-      params(
-        input: Rager::Types::TemplateInput,
-        options: Rager::Template::Options
-      ).returns(Rager::Types::TemplateOutput)
-    end
-    def self.template(input, options = Rager::Template::Options.new)
-      provider = get_provider(options.provider)
-      provider.template(input, options)
-    end
-
-    sig do
-      params(
-        key: String
-      ).returns(Rager::Template::Providers::Abstract)
-    end
-    def self.get_provider(key)
-      case key.downcase
-      when "erb"
-        Rager::Template::Providers::Erb.new
-      else
-        raise Rager::Errors::UnknownProviderError.new(Rager::Operation::Template, key)
-      end
-    end
-  end
-end