cohere-ruby 0.9.8 → 0.9.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 83208ee6cbdf496180ac2e6ed258388a74a266d86188b81ceb011bf5f2b61f44
-  data.tar.gz: 9f0ff2d20ea3c988663bc1c104f46f0c94a6d8d833d4a0ebf1cb81f765d8502d
+  metadata.gz: 35acc85a1ec589048eae603675fd23afeec9c66c0ce8c14954c20916c099b93b
+  data.tar.gz: e48ea97c36afafa8ca134c240a19ff159af1086643e99efba0aaf5aaed225cca
 SHA512:
-  metadata.gz: a0425fe338748ba829c6d15b3fe319c23cf6fe47ab18c885476dcccffe969888c215162d9139768e53ed569b5e6799c9abc17ecb028551f3cd48c2f92cec49e6
-  data.tar.gz: 48add01bf4d9ee6a43ee27809efe2eb352eee73bceac26b6fcfff29ae060063b8e3f4dc5803e317d80a612d0f6890ac18ca1d90cf5fd62254e3d1bc6f1dce8de
+  metadata.gz: eb6700a2088e5b37842fb6a601e160bbc6e8a6e8c15a7dd52b42f74cc0ce5fc71c77d789d9eaa20a440b1a3c4b4bab46088a5117b84d0a5027531acbfaf28067
+  data.tar.gz: 4e759bd938560031412e4602ee86321e3953e61dfa9c368c843b0d1b71e0676b061007dda48514b8ef5e9df072f7df3b5c817452f06a72553864bf77a679b492

data/CHANGELOG.md

@@ -1,5 +1,8 @@
 ## [Unreleased]
 
+## [0.9.9] - 2024-04-05
+- Adding missing parameters to endpoints
+
 ## [0.9.8] - 2024-02-10
 - Adding method for /chat endpoint
 

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    cohere-ruby (0.9.8)
+    cohere-ruby (0.9.9)
       faraday (>= 2.0.1, < 3.0)
 
 GEM

data/README.md

@@ -29,6 +29,7 @@ If bundler is not being used to manage dependencies, install the gem by executin
 ## Usage
 
 ### Instantiating API client
+
 ```ruby
 require "cohere"
 
@@ -36,7 +37,9 @@ client = Cohere::Client.new(
   ENV['COHERE_API_KEY']
 )
 ```
+
 ### Generate
+
 ```ruby
 client.generate(
     prompt: "Once upon a time in a magical land called"
@@ -44,13 +47,53 @@ client.generate(
 ```
 
 ### Chat
+
 ```ruby
 client.chat(
     message: "Hey! How are you?"
 )
 ```
 
+`chat` supports a streaming option. You can pass a block to the `chat` method and it will yield a new chunk as soon as it is received.
+
+```ruby
+client.chat(message: "Hey! How are you?", stream: true) do |chunk, overall_received_bytes|
+  puts "Received #{overall_received_bytes} bytes: #{chunk.force_encoding(Encoding::UTF_8)}"
+end
+```
+
+`force_encoding` is preferred to avoid JSON parsing issue when Cohere returns emoticon.
+
+`chat` supports Tool use (function calling).
+
+```ruby
+tools = [
+   {
+     name: "query_daily_sales_report",
+     description: "Connects to a database to retrieve overall sales volumes and sales information for a given day.",
+     parameter_definitions: {
+       day: {
+         description: "Retrieves sales data for this day, formatted as YYYY-MM-DD.",
+         type: "str",
+         required: true
+       }
+     }
+   }
+]
+
+message = "Can you provide a sales summary for 29th September 2023, and also give me some details about the products in the 'Electronics' category, for example their prices and stock levels?"
+
+client.chat(
+  model: model,
+  message: message,
+  tools: tools,
+)
+```
+
+
+
 ### Embed
+
 ```ruby
 client.embed(
     texts: ["hello!"]
@@ -58,6 +101,7 @@ client.embed(
 ```
 
 ### Classify
+
 ```ruby
 examples = [
     { text: "Dermatologists don't like her!", label: "Spam" },
@@ -84,6 +128,7 @@ client.classify(
 ```
 
 ### Tokenize
+
 ```ruby
 client.tokenize(
     text: "hello world!"
@@ -91,6 +136,7 @@ client.tokenize(
 ```
 
 ### Detokenize
+
 ```ruby
 client.detokenize(
     tokens: [33555, 1114 , 34]
@@ -98,6 +144,7 @@ client.detokenize(
 ```
 
 ### Detect language
+
 ```ruby
 client.detect_language(
     texts: ["Здравствуй, Мир"]
@@ -105,6 +152,7 @@ client.detect_language(
 ```
 
 ### Summarize
+
 ```ruby
 client.summarize(
     text: "..."

data/lib/cohere/client.rb

@@ -16,6 +16,7 @@ module Cohere
       message:,
       model: nil,
       stream: false,
+      preamble: nil,
       preamble_override: nil,
       chat_history: [],
       conversation_id: nil,
@@ -29,12 +30,18 @@ module Cohere
       k: nil,
       p: nil,
       frequency_penalty: nil,
-      presence_penalty: nil
+      presence_penalty: nil,
+      tools: [],
+      &block
     )
       response = connection.post("chat") do |req|
         req.body = {message: message}
         req.body[:model] = model if model
-        req.body[:stream] = stream if stream
+        if stream || block
+          req.body[:stream] = true
+          req.options.on_data = block if block
+        end
+        req.body[:preamble] = preamble if preamble
         req.body[:preamble_override] = preamble_override if preamble_override
         req.body[:chat_history] = chat_history if chat_history
         req.body[:conversation_id] = conversation_id if conversation_id
@@ -49,6 +56,7 @@ module Cohere
         req.body[:p] = p if p
         req.body[:frequency_penalty] = frequency_penalty if frequency_penalty
         req.body[:presence_penalty] = presence_penalty if presence_penalty
+        req.body[:tools] = tools if tools
       end
       response.body
     end

data/lib/cohere/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Cohere
-  VERSION = "0.9.8"
+  VERSION = "0.9.9"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: cohere-ruby
 version: !ruby/object:Gem::Version
-  version: 0.9.8
+  version: 0.9.9
 platform: ruby
 authors:
 - Andrei Bondarev
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-02-11 00:00:00.000000000 Z
+date: 2024-04-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday
@@ -70,7 +70,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.3
+rubygems_version: 3.4.1
 signing_key:
 specification_version: 4
 summary: Cohere API client for Ruby.