Merge pull request #3 from andreibondarev/version-0.1.2

version 0.1.2

Author: andreibondarev

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()`
 

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)
 

lib/google_palm_api/client.rb

@@ -47,9 +47,10 @@ def generate_text(
       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 @@ def generate_chat_message(
       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 @@ def list_models(page_size: nil, page_token: nil)
     # @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}]}}

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