llm.rb 0.7.1 → 0.7.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7d5d93a645b666da3d6947c2076189063aec26e7bc3381cfb84d6a6aea4ce8fa
-  data.tar.gz: db764cd8e9180a3c21ca5bf2b35d8a5fa3f525cb63101381aca09f9e92cb5d37
+  metadata.gz: 2c8c175d5b640f4d04e114569781bb2cad10b17f76489f4eaecd2ea7963c2baf
+  data.tar.gz: b00e4c4b7bf9211157e66f1ac015540ecec86b1f4d6d8b3bcaa46068ac97fbaf
 SHA512:
-  metadata.gz: c17999419e02e8c2d9d689299d149dd76077b1f52154557253dd9fe3876ff1109eaaa29d7d7aceda8147f38aa572261b455eaa253fbede212b874af42c8e03e0
-  data.tar.gz: 1061af3f752a1e8cbc37c8929a23f6bfa97328f66f3adf030e435f468785ffa6a3aae32fe5120936f872b60d8a48a31fcbf35cee0de66b8ee12ddedf17228beb
+  metadata.gz: e1d27b0fa9a59ec4baf55e7863675c71827c66af091f4b4d2db268e08d83709bbb87c23175c0dda8e2583ced40a2c76bafcba95f118ec64e6f44245878e668dd
+  data.tar.gz: 0201f576e4180dfef2ace9b699711eb013ac4556baf2bc9a584060000ec94fb7f764d1bb1e0d9d8187f76deeea6761e2323f661974e5e93fb50a29e0a62207d8

data/lib/llm/provider.rb

@@ -193,6 +193,20 @@ class LLM::Provider
     end
   end
 
+  ##
+  # Add one or more headers to all requests
+  # @example
+  #   llm = LLM.openai(key: ENV["KEY"])
+  #   llm.with(headers: {"OpenAI-Organization" => ENV["ORG"]})
+  #   llm.with(headers: {"OpenAI-Project" => ENV["PROJECT"]})
+  # @param [Hash<String,String>] headers
+  #  One or more headers
+  # @return [LLM::Provider]
+  #  Returns self
+  def with(headers:)
+    tap { (@headers ||= {}).merge!(headers) }
+  end
+
   private
 
   ##

data/lib/llm/providers/anthropic.rb

@@ -84,11 +84,11 @@ module LLM
     private
 
     def headers
-      {
+      (@headers || {}).merge(
         "Content-Type" => "application/json",
         "x-api-key" => @key,
         "anthropic-version" => "2023-06-01"
-      }
+      )
     end
 
     def response_parser

data/lib/llm/providers/gemini.rb

@@ -131,9 +131,9 @@ module LLM
     private
 
     def headers
-      {
+      (@headers || {}).merge(
         "Content-Type" => "application/json"
-      }
+      )
     end
 
     def response_parser

data/lib/llm/providers/ollama.rb

@@ -95,10 +95,10 @@ module LLM
     private
 
     def headers
-      {
+      (@headers || {}).merge(
         "Content-Type" => "application/json",
         "Authorization" => "Bearer #{@key}"
-      }
+      )
     end
 
     def response_parser

data/lib/llm/providers/openai.rb

@@ -119,10 +119,10 @@ module LLM
     private
 
     def headers
-      {
+      (@headers || {}).merge(
         "Content-Type" => "application/json",
         "Authorization" => "Bearer #{@key}"
-      }
+      )
     end
 
     def response_parser

data/lib/llm/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module LLM
-  VERSION = "0.7.1"
+  VERSION = "0.7.2"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: llm.rb
 version: !ruby/object:Gem::Version
-  version: 0.7.1
+  version: 0.7.2
 platform: ruby
 authors:
 - Antar Azri