omniai 1.2.1 → 1.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5b37e667b2b004a012bf7d7d76fd80c08aa8948d46b8670651daa3a426c591d2
-  data.tar.gz: ded01ccbbf34486159fd965953a9d5cec9a62da16404960e5b0c8bca8d5384ad
+  metadata.gz: 239eb058ae53afd3f28fe00a18af282a856ca842a4f81bcdf22413a02d8d44fc
+  data.tar.gz: e74ac69ab6435fd9da3d1602c9cce5e052f53bebf609d8815868b58be0897884
 SHA512:
-  metadata.gz: c6c26067f0b0b3c7ecefbdc00ed1746f0f597f729a41ff1e0a253bb723cf89ef1b82ccc2c411a8b43b93d5f54c83a9f3e58ab7074be324e9b3bf24e099e7ca2a
-  data.tar.gz: fe75154f85d6a68b1698c923548570695c724293462225bdb9d29db8ccf884ab2965bce3090a9f9c64851c5547ab4a655e59e7079c0923bff9ee18f46320517c
+  metadata.gz: a6b18cae8b5aab610bf255860ccd514b794eda3021b2f9b3aa86020968e0f449b683b4db74291e71583698b61d7612a7e49b3ee0310c806a826f7b27c50c7d92
+  data.tar.gz: bd12b38f3c9a150fe09644aaf187aa05aa1063ee136196e11fcaa356f4fe5bdc2c93f6292a79025f301c6d35a1202186f6ad787787828dfd5af91f229a34b550

data/README.md

@@ -60,6 +60,18 @@ require 'omniai/openai'
 client = OmniAI::OpenAI::Client.new
 ```
 
+#### Usage with LocalAI
+
+LocalAI support is offered through [OmniAI::OpenAI](https://github.com/ksylvest/omniai-openai):
+
+[Usage with LocalAI](https://github.com/ksylvest/omniai-openai#usage-with-localai)
+
+#### Usage with Ollama
+
+Ollama support is offered through [OmniAI::OpenAI](https://github.com/ksylvest/omniai-openai):
+
+[Usage with Ollama](https://github.com/ksylvest/omniai-openai#usage-with-ollama)
+
 ### Chat
 
 Clients that support chat (e.g. Anthropic w/ "Claude", Google w/ "Gemini", Mistral w/ "LeChat", OpenAI w/ "ChatGPT", etc) generate completions using the following calls:

data/lib/omniai/client.rb

@@ -13,19 +13,28 @@ module OmniAI
   class Client
     class Error < StandardError; end
 
-    attr_accessor :api_key
+    attr_accessor :api_key, :logger, :host
 
-    # @param api_key [String]
-    # @param logger [Logger]
-    def initialize(api_key:, logger: nil)
+    # @param api_key [String] optional
+    # @param host [String] optional - supports for customzing the host of the client (e.g. 'http://localhost:8080')
+    # @param logger [Logger] optional
+    def initialize(api_key: nil, logger: nil, host: nil)
       @api_key = api_key
+      @host = host
       @logger = logger
     end
 
     # @return [String]
     def inspect
-      masked_api_key = "#{api_key[..2]}***" if api_key
-      "#<#{self.class.name} api_key=#{masked_api_key.inspect}>"
+      props = []
+      props << "api_key=#{masked_api_key.inspect}" if @api_key
+      props << "host=#{@host.inspect}" if @host
+      "#<#{self.class.name} #{props.join(' ')}>"
+    end
+
+    # @return [String]
+    def masked_api_key
+      "#{api_key[..2]}***" if api_key
     end
 
     # @return [HTTP::Client]

data/lib/omniai/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module OmniAI
-  VERSION = '1.2.1'
+  VERSION = '1.2.2'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: omniai
 version: !ruby/object:Gem::Version
-  version: 1.2.1
+  version: 1.2.2
 platform: ruby
 authors:
 - Kevin Sylvestre