fixes

Author: andreibondarev

README.md

@@ -6,20 +6,37 @@ Welcome to your new gem! In this directory, you'll find the files you need to be
 
 ## Installation
 
-TODO: Replace `UPDATE_WITH_YOUR_GEM_NAME_PRIOR_TO_RELEASE_TO_RUBYGEMS_ORG` with your gem name right after releasing it to RubyGems.org. Please do not do it earlier due to security reasons. Alternatively, replace this section with instructions to install your gem from git if you don't plan to release to RubyGems.org.
-
 Install the gem and add to the application's Gemfile by executing:
 
-    $ bundle add UPDATE_WITH_YOUR_GEM_NAME_PRIOR_TO_RELEASE_TO_RUBYGEMS_ORG
+    $ bundle add google_palm_api
 
 If bundler is not being used to manage dependencies, install the gem by executing:
 
-    $ gem install UPDATE_WITH_YOUR_GEM_NAME_PRIOR_TO_RELEASE_TO_RUBYGEMS_ORG
+    $ gem install google_palm_api
 
 ## Usage
 
-TODO: Write usage instructions here
-
+```ruby
+require 'google_palm_api'
+```
+```ruby
+client.generate_text(prompt:)
+```
+```ruby
+client.generate_chat_message(prompt:)
+```
+```ruby
+client.embed(text:)
+```
+```ruby
+client.get_model(model:)
+```
+```ruby
+client.list_models()
+```
+```ruby
+client.count_message_tokens(message:)
+```
 ## Development
 
 After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.

lib/google_palm_api/client.rb

@@ -40,7 +40,6 @@ def initialize(api_key:)
     #
     def generate_text(
       prompt:,
-      model: nil,
       temperature: nil,
       candidate_count: nil,
       max_output_tokens: nil,
@@ -50,7 +49,7 @@ def generate_text(
       stop_sequences: nil,
       client: nil
     )
-      response = connection.post("/v1beta2/models/#{model || DEFAULTS[:completion_model_name]}:generateText") do |req|
+      response = connection.post("/v1beta2/models/#{DEFAULTS[:completion_model_name]}:generateText") do |req|
         req.params = {key: api_key}
 
         req.body = {prompt: { text: prompt }}
@@ -87,7 +86,6 @@ def generate_text(
     #
     def generate_chat_message(
       prompt:,
-      model: nil,
       context: nil,
       examples: nil,
       messages: nil,
@@ -98,11 +96,10 @@ def generate_chat_message(
       client: nil
     )
       # Overwrite the default ENDPOINT_URL for this method.
-      response = connection.post("/v1beta2/models/#{model || DEFAULTS[:chat_completion_model_name]}:generateMessage") do |req|
+      response = connection.post("/v1beta2/models/#{DEFAULTS[:chat_completion_model_name]}:generateMessage") do |req|
         req.params = {key: api_key}
 
         req.body = {prompt: { messages: [{content: prompt}] }}
-        req.body[:model] = model if model
         req.body[:context] = context if context
         req.body[:examples] = examples if examples
         req.body[:messages] = messages if messages