omniai 0.2.0 → 1.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1694fba08fa9ef9bd8dc3496ce1acbc3c321357d082303869df74ee667a83a65
-  data.tar.gz: c0d98caa29e034c610d20654cc175e42eeb69cef414ba79a5fda61d883d2d9f3
+  metadata.gz: a227888b197b576509bec7208b79f125e2c55baac317ce32b3fed8e2101f11cc
+  data.tar.gz: 16fe842b15ef266556c9887b1625c32a62edbbdba21dea0a8b21e418d76cd435
 SHA512:
-  metadata.gz: ddfcd3c15811a63b401290577b36170573e4e170064a8cf64686ceacfc1ddb2c3e5af10ab2a7bb1057db9fe691b8359a94da24c0ce774f71ed5df50063b3925b
-  data.tar.gz: 66dcd111279e7de048473789cfae9020a0c6bb9b812bdd10a1bb2c2c12ee246bd9cebf6b7890322fdcc4a72743033746bd9e5a3f8d283b43ae0bbea5315de05e
+  metadata.gz: a4bc067e862d142036dcc15721fb7f3cf5084112145676b3884977e5d3e7025f34637eab3520ad2a1f2d524909f5e8a55b7a2bee35c139062cb481ffc64b3071
+  data.tar.gz: c95602154261e59d7e42fbc94d7afb42f836f54287c4842c0c982a82ced52271ac8c41afbb6cb620865ca948d55971c789c94967f5f18e2af8305271e1c2885b

data/lib/omniai/chat/request.rb

@@ -69,7 +69,7 @@ module OmniAI
       end
 
       # @param response [HTTP::Response]
-      # @return [OmniAI::Chat::Chunk]
+      # @return [OmniAI::Chat::Stream]
       def stream!(response:)
         raise Error, "#{self.class.name}#stream! unstreamable" unless @stream
 

data/lib/omniai/chat.rb

@@ -17,6 +17,8 @@ module OmniAI
   #
   # @param client [OmniAI::Client] the client
   class Chat
+    JSON_PROMPT = 'Respond with valid JSON. Do not include any non-JSON in the response.'
+
     module Role
       ASSISTANT = 'assistant'
       USER = 'user'

data/lib/omniai/client.rb

@@ -27,6 +27,12 @@ module OmniAI
       @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}>"
+    end
+
     # @return [HTTP::Client]
     def connection
       raise NotImplementedError, "#{self.class.name}#connection undefined"

data/lib/omniai/config.rb

@@ -4,5 +4,11 @@ module OmniAI
   # A configuration for each agent w/ `api_key` / `host` / `logger`.
   class Config
     attr_accessor :api_key, :host, :logger
+
+    # @return [String]
+    def inspect
+      masked_api_key = "#{api_key[..2]}***" if api_key
+      "#<#{self.class.name} api_key=#{masked_api_key.inspect} host=#{host.inspect}>"
+    end
   end
 end

data/lib/omniai/version.rb

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

metadata

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