google_palm_api 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ac7103e077c1ab4551ab2c296f2eb3d99eb3b7359e59eea1a3b1b4bbdcfea815
-  data.tar.gz: ce2f516c03358ec6e1b320a333e2b630fd716565ed8c53e48528681a0c74eece
+  metadata.gz: 0b076d778e0a488d4b456cc10cc3243850ef90288c040dcdb94edf7dcd91a121
+  data.tar.gz: 6c40d661c77996cb8c72b865b0b85771e3afde4298db9100ee93a4488cfea198
 SHA512:
-  metadata.gz: eef61e7eb7981f6f12f5568deb359e5a71c2339d0696e4c6595b5521df9a5bb16649587ba0c773f2c563b9aa9b4d6a44403cf1f2634854f388ce40cf05644486
-  data.tar.gz: 963774e47464929a32a2dc92247a8ed171eb79efb1b911bd0be25a28bb7396490e627a155f198e45354e42b5e96c15a57974befe4935d1836f2f7d42609048c0
+  metadata.gz: 3f35f5be9055fbf1a7141b3ffd372fc18adff924a89c2dcca58d68c5783fd13b486ab53bc7eb859b1f9f40bb6f52071d384bb66f338ba684ada848c3d932a5d7
+  data.tar.gz: 41e126d177942335e3161c7691cbeef9dfb0d43b0158f420ff35fb7865d76b88761242758a17b66264c9a6dff437e91e3c35b0dca840e52a431ba2ca959d62ff

data/CHANGELOG.md

@@ -1,5 +1,8 @@
 ## [Unreleased]
 
+## [0.1.2] - 2023-06-20
+- Accept the `model:` param
+
 ## [0.1.1] - 2023-06-09
 - Fix the `generate_chat_message()`
 

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    google_palm_api (0.1.1)
+    google_palm_api (0.1.2)
       faraday (>= 1.0.0)
       faraday_middleware (>= 1.0.0)
 

data/lib/google_palm_api/client.rb

@@ -47,9 +47,10 @@ module GooglePalmApi
       top_k: nil,
       safety_settings: nil,
       stop_sequences: nil,
-      client: nil
+      client: nil,
+      model: nil
     )
-      response = connection.post("/v1beta2/models/#{DEFAULTS[:completion_model_name]}:generateText") do |req|
+      response = connection.post("/v1beta2/models/#{model || DEFAULTS[:completion_model_name]}:generateText") do |req|
         req.params = {key: api_key}
 
         req.body = {prompt: {text: prompt}}
@@ -93,10 +94,11 @@ module GooglePalmApi
       candidate_count: nil,
       top_p: nil,
       top_k: nil,
-      client: nil
+      client: nil,
+      model: nil
     )
       # Overwrite the default ENDPOINT_URL for this method.
-      response = connection.post("/v1beta2/models/#{DEFAULTS[:chat_completion_model_name]}:generateMessage") do |req|
+      response = connection.post("/v1beta2/models/#{model || DEFAULTS[:chat_completion_model_name]}:generateMessage") do |req|
         req.params = {key: api_key}
 
         req.body = {prompt: {}}
@@ -165,8 +167,8 @@ module GooglePalmApi
     # @param [String] prompt
     # @return [Hash]
     #
-    def count_message_tokens(model:, prompt:)
-      response = connection.post("/v1beta2/models/#{model}:countMessageTokens") do |req|
+    def count_message_tokens(prompt:, model: nil)
+      response = connection.post("/v1beta2/models/#{model || DEFAULTS[:chat_completion_model_name]}:countMessageTokens") do |req|
         req.params = {key: api_key}
 
         req.body = {prompt: {messages: [{content: prompt}]}}

data/lib/google_palm_api/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module GooglePalmApi
-  VERSION = "0.1.1"
+  VERSION = "0.1.2"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: google_palm_api
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Andrei Bondarev
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-06-09 00:00:00.000000000 Z
+date: 2023-06-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pry-byebug