google_palm_api 0.1.0 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 07dac69e13dd3173be4f180d964976dd23fa6fc77b36af0742507d853cf17638
-  data.tar.gz: a495c0b2f59377c57c3d90f4d8bbedbc4835a75b194e745c7d99d0322e0fc8f4
+  metadata.gz: 0b076d778e0a488d4b456cc10cc3243850ef90288c040dcdb94edf7dcd91a121
+  data.tar.gz: 6c40d661c77996cb8c72b865b0b85771e3afde4298db9100ee93a4488cfea198
 SHA512:
-  metadata.gz: 7732a91b0c60d747cb48c1f7c5f9d939d07708a518745ca2c0112b91b52343ef3c1e0b7359a39bcb33b4b5fb68e12b08858c5e5c607f8a9e8e26105111781da1
-  data.tar.gz: 428afa1b2b46347acec9826fb8ca9f92eaa68418ba69761170b5e82809f120a4b86b09b47210e41fb405473a94c36bc0e9eba41cf5db4a8cdda9006ac9432608
+  metadata.gz: 3f35f5be9055fbf1a7141b3ffd372fc18adff924a89c2dcca58d68c5783fd13b486ab53bc7eb859b1f9f40bb6f52071d384bb66f338ba684ada848c3d932a5d7
+  data.tar.gz: 41e126d177942335e3161c7691cbeef9dfb0d43b0158f420ff35fb7865d76b88761242758a17b66264c9a6dff437e91e3c35b0dca840e52a431ba2ca959d62ff

data/CHANGELOG.md

@@ -1,5 +1,11 @@
 ## [Unreleased]
 
+## [0.1.2] - 2023-06-20
+- Accept the `model:` param
+
+## [0.1.1] - 2023-06-09
+- Fix the `generate_chat_message()`
+
 ## [0.1.0] - 2023-05-23
 
 - Initial release

data/Gemfile.lock

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

data/README.md

@@ -1,8 +1,9 @@
 # GooglePalmApi
 
-TODO: Delete this and the text below, and describe your gem
+Ruby API client for the Google PaLM APIs.
+
+![Tests status](https://github.com/andreibondarev/google_palm_api/actions/workflows/ci.yml/badge.svg) [![Gem Version](https://badge.fury.io/rb/google_palm_api.svg)](https://badge.fury.io/rb/google_palm_api)
 
-Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/google_palm_api`. To experiment with that code, run `bin/console` for an interactive prompt.
 
 ## Installation
 

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}}
@@ -85,24 +86,27 @@ module GooglePalmApi
     # @return [Hash]
     #
     def generate_chat_message(
-      prompt:,
+      messages:,
+      prompt: nil,
       context: nil,
       examples: nil,
-      messages: nil,
       temperature: nil,
       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: {messages: [{content: prompt}]}}
-        req.body[:context] = context if context
-        req.body[:examples] = examples if examples
-        req.body[:messages] = messages if messages
+        req.body = {prompt: {}}
+
+        req.body[:prompt][:messages] = messages if messages
+        req.body[:prompt][:context] = context if context
+        req.body[:prompt][:examples] = examples if examples
+
         req.body[:temperature] = temperature || DEFAULTS[:temperature]
         req.body[:candidate_count] = candidate_count if candidate_count
         req.body[:top_p] = top_p if top_p
@@ -163,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.0"
+  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.0
+  version: 0.1.2
 platform: ruby
 authors:
 - Andrei Bondarev
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-05-24 00:00:00.000000000 Z
+date: 2023-06-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pry-byebug