durable-llm 0.1.4 → 0.1.6

data/sig/durable/llm.rbs

@@ -1,6 +1,308 @@
 module Durable
   module Llm
     VERSION: String
-    # See the writing guide of rbs: https://github.com/ruby/rbs#guides
+
+    def self.configure: () { (Configuration) -> void } -> void
+    def self.configuration: () -> Configuration
+    def self.config: () -> Configuration
+
+    class Error < StandardError
+    end
+
+    class APIError < Error
+    end
+
+    class RateLimitError < Error
+    end
+
+    class AuthenticationError < Error
+    end
+
+    class InvalidRequestError < Error
+    end
+
+    class ResourceNotFoundError < Error
+    end
+
+    class TimeoutError < Error
+    end
+
+    class ServerError < Error
+    end
+
+    class UnsupportedProviderError < Error
+    end
+
+    class ConfigurationError < Error
+    end
+
+    class ModelNotFoundError < Error
+    end
+
+    class InsufficientQuotaError < Error
+    end
+
+    class InvalidResponseError < Error
+    end
+
+    class NetworkError < Error
+    end
+
+    class StreamingError < Error
+    end
+
+    class Configuration
+      attr_accessor default_provider: String
+      attr_reader providers: Hash[Symbol, untyped]
+
+      def initialize: () -> void
+      def clear: () -> void
+      def load_from_datasette: () -> void
+      def load_from_env: () -> void
+      def method_missing: (Symbol method_name, *untyped args) -> untyped
+      def respond_to_missing?: (Symbol method_name, ?bool include_private) -> bool
+    end
+
+    class Client
+      attr_reader provider: untyped
+      attr_accessor model: String?
+
+      def initialize: (Symbol | String provider_name, ?Hash[Symbol | String, untyped] options) -> void
+      def default_params: () -> Hash[Symbol, String?]
+      def complete: (String text, ?Hash[Symbol, untyped] _opts) -> String
+      alias quick_complete complete
+      def completion: (?Hash[Symbol, untyped] params) -> untyped
+      def chat: (?Hash[Symbol, untyped] params) -> untyped
+      def embed: (?Hash[Symbol, untyped] params) -> untyped
+      def stream: (?Hash[Symbol, untyped] params) { (untyped) -> void } -> untyped
+      def stream?: () -> bool
+
+      private
+      def process_params: (?Hash[Symbol, untyped] opts) -> Hash[Symbol, untyped]
+    end
+
+    module Providers
+      def self.load_all: () -> void
+      def self.providers: () -> Array[Symbol]
+      def self.model_ids: () -> Array[String]
+      def self.model_id_to_provider: (String model_id) -> Class?
+
+      class Base
+        attr_accessor api_key: String?
+
+        def initialize: (?api_key: String?) -> void
+        def default_api_key: () -> String?
+        def completion: (Hash[Symbol | String, untyped] options) -> untyped
+        def self.models: () -> Array[String]
+        def models: () -> Array[String]
+        def self.stream?: () -> bool
+        def stream?: () -> bool
+        def stream: (Hash[Symbol | String, untyped] options) { (untyped) -> void } -> untyped
+        def embedding: (model: String, input: String | Array[String], **untyped options) -> untyped
+
+        private
+        def handle_response: (untyped response) -> untyped
+      end
+
+      class OpenAI < Base
+        BASE_URL: String
+        attr_accessor organization: String?
+
+        def initialize: (?api_key: String?, ?organization: String?) -> void
+        def completion: (Hash[Symbol | String, untyped] options) -> OpenAIResponse
+        def embedding: (model: String, input: String | Array[String], **untyped options) -> OpenAIEmbeddingResponse
+        def models: () -> Array[String]
+        def self.stream?: () -> true
+        def stream: (Hash[Symbol | String, untyped] options) { (OpenAIStreamResponse) -> void } -> void
+
+        class OpenAIResponse
+          attr_reader raw_response: Hash[String, untyped]
+          def initialize: (Hash[String, untyped] response) -> void
+          def choices: () -> Array[OpenAIChoice]
+          def data: () -> untyped
+          def embedding: () -> untyped
+          def to_s: () -> String
+        end
+
+        class OpenAIChoice
+          attr_reader message: OpenAIMessage
+          attr_reader finish_reason: String?
+          def initialize: (Hash[String, untyped] choice) -> void
+          def to_s: () -> String
+        end
+
+        class OpenAIMessage
+          attr_reader role: String
+          attr_reader content: String
+          def initialize: (Hash[String, untyped] message) -> void
+          def to_s: () -> String
+        end
+
+        class OpenAIStreamResponse
+          attr_reader choices: OpenAIStreamChoice
+          def initialize: (Hash[String, untyped] parsed) -> void
+          def to_s: () -> String
+        end
+
+        class OpenAIStreamChoice
+          attr_reader delta: OpenAIStreamDelta
+          attr_reader finish_reason: String?
+          def initialize: (Array[Hash[String, untyped]] | Hash[String, untyped] choice) -> void
+          def to_s: () -> String
+        end
+
+        class OpenAIStreamDelta
+          attr_reader role: String?
+          attr_reader content: String?
+          def initialize: (Hash[String, untyped] delta) -> void
+          def to_s: () -> String
+        end
+
+        class OpenAIEmbeddingResponse
+          attr_reader embedding: Array[Float]
+          def initialize: (Hash[String, untyped] data) -> void
+          def to_a: () -> Array[Float]
+        end
+      end
+
+      class Anthropic < Base
+        BASE_URL: String
+
+        def initialize: (?api_key: String?) -> void
+        def completion: (Hash[Symbol | String, untyped] options) -> AnthropicResponse
+        def models: () -> Array[String]
+        def self.models: () -> Array[String]
+        def self.stream?: () -> true
+        def stream: (Hash[Symbol | String, untyped] options) { (AnthropicStreamResponse) -> void } -> void
+
+        class AnthropicResponse
+          attr_reader raw_response: Hash[String, untyped]
+          def initialize: (Hash[String, untyped] response) -> void
+          def choices: () -> Array[AnthropicChoice]
+          def to_s: () -> String
+        end
+
+        class AnthropicChoice
+          attr_reader message: AnthropicMessage
+          def initialize: (untyped content) -> void
+          def to_s: () -> String
+        end
+
+        class AnthropicMessage
+          attr_reader role: String
+          attr_reader content: String
+          def initialize: (untyped content) -> void
+          def to_s: () -> String
+        end
+
+        class AnthropicStreamResponse
+          attr_reader choices: Array[AnthropicStreamChoice]
+          def initialize: (String fragment) -> void
+          def to_s: () -> String
+        end
+
+        class AnthropicStreamChoice
+          attr_reader delta: AnthropicStreamDelta
+          def initialize: (Hash[String, untyped] delta) -> void
+          def to_s: () -> String
+        end
+
+        class AnthropicStreamDelta
+          attr_reader type: String?
+          attr_reader text: String?
+          def initialize: (Hash[String, untyped] delta) -> void
+          def to_s: () -> String
+        end
+      end
+
+      class Groq < Base
+        BASE_URL: String
+
+        def self.conn: () -> untyped
+        def conn: () -> untyped
+        def initialize: (?api_key: String?) -> void
+        def completion: (Hash[Symbol | String, untyped] options) -> GroqResponse
+        def embedding: (model: String, input: String | Array[String], **untyped options) -> GroqResponse
+        def models: () -> Array[String]
+        def self.stream?: () -> false
+
+        class GroqResponse
+          attr_reader raw_response: Hash[String, untyped]
+          def initialize: (Hash[String, untyped] response) -> void
+          def choices: () -> Array[GroqChoice]
+          def to_s: () -> String
+          def to_h: () -> Hash[String, untyped]
+        end
+
+        class GroqChoice
+          attr_reader message: GroqMessage
+          attr_reader finish_reason: String?
+          def initialize: (Hash[String, untyped] choice) -> void
+          def to_s: () -> String
+        end
+
+        class GroqMessage
+          attr_reader role: String
+          attr_reader content: String
+          def initialize: (Hash[String, untyped] message) -> void
+          def to_s: () -> String
+        end
+      end
+
+      class Huggingface < Base
+        BASE_URL: String
+
+        def initialize: (?api_key: String?) -> void
+        def completion: (Hash[Symbol | String, untyped] options) -> HuggingfaceResponse
+        def models: () -> Array[String]
+        def self.models: () -> Array[String]
+
+        class HuggingfaceResponse
+          attr_reader raw_response: untyped
+          def initialize: (untyped response) -> void
+          def choices: () -> Array[HuggingfaceChoice]
+          def to_s: () -> String
+        end
+
+        class HuggingfaceChoice
+          attr_reader text: String
+          def initialize: (Hash[String, untyped] choice) -> void
+          def to_s: () -> String
+        end
+      end
+
+      class Cohere < Base
+        BASE_URL: String
+
+        def initialize: (?api_key: String?) -> void
+        def completion: (Hash[Symbol | String, untyped] options) -> CohereResponse
+        def models: () -> Array[String]
+        def self.stream?: () -> false
+
+        class CohereResponse
+          attr_reader raw_response: Hash[String, untyped]
+          def initialize: (Hash[String, untyped] response) -> void
+          def choices: () -> Array[CohereChoice]
+          def to_s: () -> String
+        end
+
+        class CohereChoice
+          attr_reader text: String
+          def initialize: (Hash[String, untyped] generation) -> void
+          def to_s: () -> String
+        end
+      end
+
+      Openai: singleton(OpenAI)
+      Claude: singleton(Anthropic)
+      Claude3: singleton(Anthropic)
+    end
+
+    class CLI < Thor
+      def self.exit_on_failure?: () -> true
+      def prompt: (*String prompt) -> void
+      def chat: () -> void
+      def models: () -> void
+    end
   end
 end

metadata

@@ -1,29 +1,74 @@
 --- !ruby/object:Gem::Specification
 name: durable-llm
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.6
 platform: ruby
 authors:
 - Durable Programming Team
-autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-10-10 00:00:00.000000000 Z
+date: 1980-01-02 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: event_stream_parser
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.0.0
 - !ruby/object:Gem::Dependency
   name: faraday
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">"
+    - - ">="
       - !ruby/object:Gem::Version
         version: '1.0'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '3.0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">"
+    - - ">="
       - !ruby/object:Gem::Version
         version: '1.0'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: highline
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.1'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 3.1.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.1'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 3.1.0
 - !ruby/object:Gem::Dependency
   name: json
   requirement: !ruby/object:Gem::Requirement
@@ -31,6 +76,9 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '2.6'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.6.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
@@ -38,62 +86,77 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '2.6'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.6.0
 - !ruby/object:Gem::Dependency
-  name: thor
+  name: ostruct
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.3'
+        version: 0.6.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.3'
+        version: 0.6.0
 - !ruby/object:Gem::Dependency
-  name: zeitwerk
+  name: thor
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '2.6'
+        version: '1.3'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.3.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '2.6'
+        version: '1.3'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.3.0
 - !ruby/object:Gem::Dependency
-  name: highline
+  name: zeitwerk
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '3.1'
+        version: '2.6'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.6.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '3.1'
+        version: '2.6'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.6.0
 - !ruby/object:Gem::Dependency
-  name: event_stream_parser
+  name: dotenv
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.0'
-  type: :runtime
+        version: '2.8'
+  type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.0'
+        version: '2.8'
 - !ruby/object:Gem::Dependency
   name: minitest
   requirement: !ruby/object:Gem::Requirement
@@ -137,33 +200,33 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '1.0'
 - !ruby/object:Gem::Dependency
-  name: dotenv
+  name: vcr
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '2.8'
+        version: '6.0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '2.8'
+        version: '6.0'
 - !ruby/object:Gem::Dependency
-  name: vcr
+  name: yard
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '6.0'
+        version: '0.9'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '6.0'
+        version: '0.9'
 description: Durable-LLM is a unified interface for interacting with multiple Large
   Language Model APIs, simplifying integration of AI capabilities into Ruby applications.
 email:
@@ -172,14 +235,18 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- ".envrc"
 - CHANGELOG.md
 - CLI.md
 - CONFIGURE.md
 - Gemfile
-- Gemfile.lock
 - LICENSE.txt
 - README.md
 - Rakefile
+- devenv.lock
+- devenv.nix
+- devenv.yaml
+- durable-llm.gemspec
 - enterprise/GET_ENTERPRISE.md
 - examples/anthropic.md
 - examples/openai.md
@@ -189,14 +256,27 @@ files:
 - lib/durable/llm/cli.rb
 - lib/durable/llm/client.rb
 - lib/durable/llm/configuration.rb
+- lib/durable/llm/convenience.rb
 - lib/durable/llm/errors.rb
+- lib/durable/llm/provider_utilities.rb
 - lib/durable/llm/providers.rb
 - lib/durable/llm/providers/anthropic.rb
+- lib/durable/llm/providers/azure_openai.rb
 - lib/durable/llm/providers/base.rb
 - lib/durable/llm/providers/cohere.rb
+- lib/durable/llm/providers/deepseek.rb
+- lib/durable/llm/providers/fireworks.rb
+- lib/durable/llm/providers/google.rb
 - lib/durable/llm/providers/groq.rb
 - lib/durable/llm/providers/huggingface.rb
+- lib/durable/llm/providers/mistral.rb
 - lib/durable/llm/providers/openai.rb
+- lib/durable/llm/providers/opencode.rb
+- lib/durable/llm/providers/openrouter.rb
+- lib/durable/llm/providers/perplexity.rb
+- lib/durable/llm/providers/together.rb
+- lib/durable/llm/providers/xai.rb
+- lib/durable/llm/response_helpers.rb
 - lib/durable/llm/version.rb
 - sig/durable/llm.rbs
 homepage: https://github.com/durableprogramming/durable-llm
@@ -207,7 +287,6 @@ metadata:
   homepage_uri: https://github.com/durableprogramming/durable-llm
   source_code_uri: https://github.com/durableprogramming/durable-llm
   changelog_uri: https://github.com/durableprogramming/durable-llm/blob/main/CHANGELOG.md
-post_install_message: 
 rdoc_options: []
 require_paths:
 - lib
@@ -222,8 +301,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.4.10
-signing_key: 
+rubygems_version: 3.7.1
 specification_version: 4
 summary: A Ruby gem providing access to LLM APIs from various vendors
 test_files: []

data/Gemfile.lock

@@ -1,103 +0,0 @@
-PATH
-  remote: .
-  specs:
-    durable-llm (0.1.3)
-      event_stream_parser (~> 1.0)
-      faraday (> 1.0)
-      highline (~> 3.1)
-      json (~> 2.6)
-      thor (~> 1.3)
-      zeitwerk (~> 2.6)
-
-GEM
-  remote: https://rubygems.org/
-  specs:
-    addressable (2.8.7)
-      public_suffix (>= 2.0.2, < 7.0)
-    ast (2.4.2)
-    base64 (0.2.0)
-    bigdecimal (3.1.8)
-    crack (1.0.0)
-      bigdecimal
-      rexml
-    dotenv (2.8.1)
-    event_stream_parser (1.0.0)
-    faraday (2.12.0)
-      faraday-net_http (>= 2.0, < 3.4)
-      json
-      logger
-    faraday-multipart (1.0.4)
-      multipart-post (~> 2)
-    faraday-net_http (3.3.0)
-      net-http
-    hashdiff (1.1.1)
-    highline (3.1.1)
-      reline
-    io-console (0.7.2)
-    json (2.7.2)
-    language_server-protocol (3.17.0.3)
-    logger (1.6.1)
-    minitest (5.25.1)
-    mocha (2.4.5)
-      ruby2_keywords (>= 0.0.5)
-    multipart-post (2.4.1)
-    net-http (0.4.1)
-      uri
-    parallel (1.26.3)
-    parser (3.3.5.0)
-      ast (~> 2.4.1)
-      racc
-    public_suffix (6.0.1)
-    racc (1.8.1)
-    rainbow (3.1.1)
-    rake (13.2.1)
-    regexp_parser (2.9.2)
-    reline (0.5.10)
-      io-console (~> 0.5)
-    rexml (3.3.8)
-    rubocop (1.66.1)
-      json (~> 2.3)
-      language_server-protocol (>= 3.17.0)
-      parallel (~> 1.10)
-      parser (>= 3.3.0.2)
-      rainbow (>= 2.2.2, < 4.0)
-      regexp_parser (>= 2.4, < 3.0)
-      rubocop-ast (>= 1.32.2, < 2.0)
-      ruby-progressbar (~> 1.7)
-      unicode-display_width (>= 2.4.0, < 3.0)
-    rubocop-ast (1.32.3)
-      parser (>= 3.3.1.0)
-    ruby-openai (7.1.0)
-      event_stream_parser (>= 0.3.0, < 2.0.0)
-      faraday (>= 1)
-      faraday-multipart (>= 1)
-    ruby-progressbar (1.13.0)
-    ruby2_keywords (0.0.5)
-    thor (1.3.2)
-    unicode-display_width (2.6.0)
-    uri (0.13.1)
-    vcr (6.3.1)
-      base64
-    webmock (3.24.0)
-      addressable (>= 2.8.0)
-      crack (>= 0.3.2)
-      hashdiff (>= 0.4.0, < 2.0.0)
-    zeitwerk (2.6.18)
-
-PLATFORMS
-  x86_64-linux
-
-DEPENDENCIES
-  dotenv (~> 2.8)
-  durable-llm!
-  minitest (~> 5.0)
-  mocha (~> 2.1)
-  rake (~> 13.0)
-  rubocop (~> 1.21)
-  ruby-openai (~> 7.1)
-  thor (~> 1.3)
-  vcr (~> 6.0)
-  webmock (~> 3.24)
-
-BUNDLED WITH
-   2.4.10