cohere-ruby 0.9.11 → 1.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5f1c3798500d154b882bd6bb2e0e43904fea4bf69db4d982da4abc6265bc461c
-  data.tar.gz: d91ed933c24f2dacc5f2604639d307fd389f3774aaed8082df07e03ac2961704
+  metadata.gz: 0c65ac65e6fa623709455907a71697f721311d2933e96a10a5c49f907f356f02
+  data.tar.gz: 553fd85c32a2a99b2a8d1a75f1a2aa0a0c9abec290f305ed26caa855515f717b
 SHA512:
-  metadata.gz: b989c60be670a143848367a7be84f53fc6adf97f31e1f3739113a7a7caec988450e4e889bcbc7f2733783623977dc4f6e1eb7484470259935e30e7dc6d5d1414
-  data.tar.gz: 00a2e3e1a19bfb1980d06ebc1fd6117262bcdf226bc95bfe0edff8ea8e1397791eaf7ef37b78ad158e0ef5a8363fa809c9215fc325d771e72abf22d7919c54c0
+  metadata.gz: bb7dd10864d789f91225f9370011e622dc6353a55a7039b8ef509610980c1503c3916f210189361cc1dcc20cb1db880d7fd4d80f967867fdcecd6803e85e8bdb
+  data.tar.gz: 89eb351c192abfda2c905810210b21be5d6a5b18aa6e3d5a4ac28efaba9fc23e1a0525510b51a4399c5e7086fab12a854035388806c4401122c3463277941990

data/.env.example

@@ -0,0 +1 @@
+COHERE_API_KEY=

data/CHANGELOG.md

@@ -1,5 +1,10 @@
 ## [Unreleased]
 
+## [1.0.1] - 2024-11-22
+
+## [1.0.0] - 2024-11-22
+- Migrate to v2 APIs
+
 ## [0.9.11] - 2024-08-01
 - New `rerank()` method
 

data/Gemfile

@@ -9,3 +9,4 @@ gem "rake", "~> 13.0"
 
 gem "rspec", "~> 3.0"
 gem "standard", "~> 1.28.0"
+gem "dotenv", "~> 2.8.1"

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    cohere-ruby (0.9.11)
+    cohere-ruby (1.0.1)
       faraday (>= 2.0.1, < 3.0)
 
 GEM
@@ -9,6 +9,7 @@ GEM
   specs:
     ast (2.4.2)
     diff-lcs (1.5.0)
+    dotenv (2.8.1)
     faraday (2.7.10)
       faraday-net_http (>= 2.0, < 3.1)
       ruby2_keywords (>= 0.0.4)
@@ -74,6 +75,7 @@ PLATFORMS
 
 DEPENDENCIES
   cohere-ruby!
+  dotenv (~> 2.8.1)
   rake (~> 13.0)
   rspec (~> 3.0)
   standard (~> 1.28.0)

data/README.md

@@ -8,12 +8,12 @@
 
 Cohere API client for Ruby.
 
-Part of the [Langchain.rb](https://github.com/andreibondarev/langchainrb) stack.
+Part of the [Langchain.rb](https://github.com/patterns-ai-core/langchainrb) stack.
 
-![Tests status](https://github.com/andreibondarev/cohere-ruby/actions/workflows/ci.yml/badge.svg)
+![Tests status](https://github.com/patterns-ai-core/cohere-ruby/actions/workflows/ci.yml/badge.svg)
 [![Gem Version](https://badge.fury.io/rb/cohere-ruby.svg)](https://badge.fury.io/rb/cohere-ruby)
 [![Docs](http://img.shields.io/badge/yard-docs-blue.svg)](http://rubydoc.info/gems/cohere-ruby)
-[![License](https://img.shields.io/badge/license-MIT-green.svg)](https://github.com/andreibondarev/cohere-ruby/blob/main/LICENSE.txt)
+[![License](https://img.shields.io/badge/license-MIT-green.svg)](https://github.com/patterns-ai-core/cohere-ruby/blob/main/LICENSE.txt)
 [![](https://dcbadge.vercel.app/api/server/WDARp7J2n8?compact=true&style=flat)](https://discord.gg/WDARp7J2n8)
 
 ## Installation
@@ -50,14 +50,18 @@ client.generate(
 
 ```ruby
 client.chat(
-  message: "Hey! How are you?"
+  model: "command-r-plus-08-2024",
+  messages: [{role:"user", content: "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|
+client.chat(
+  model: "command-r-plus-08-2024",
+  messages: [{role:"user", content: "Hey! How are you?"}]
+) do |chunk, overall_received_bytes|
   puts "Received #{overall_received_bytes} bytes: #{chunk.force_encoding(Encoding::UTF_8)}"
 end
 ```
@@ -68,25 +72,25 @@ end
 
 ```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
-       }
-     }
-   }
+  {
+    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,
+  messages: [{ role:"user", content: message }],
+  tools: tools
 )
 ```
 
@@ -94,7 +98,10 @@ client.chat(
 
 ```ruby
 client.embed(
-  texts: ["hello!"]
+  model: "embed-english-v3.0",
+  texts: ["hello", "goodbye"],
+  input_type: "classification",
+  embedding_types: ["float"]
 )
 ```
 
@@ -110,11 +117,12 @@ docs = [
 ]
 
 client.rerank(
-  texts: ["hello!"]
+  model: "rerank-english-v3.0",
+  query: "What is the capital of the United States?",
+  documents: docs
 )
 ```
 
-
 ### Classify
 
 ```ruby
@@ -137,8 +145,9 @@ inputs = [
 ]
 
 client.classify(
-  examples: examples,
-  inputs: inputs
+  model: "embed-multilingual-v2.0",
+  inputs: inputs,
+  examples: examples
 )
 ```
 
@@ -146,7 +155,8 @@ client.classify(
 
 ```ruby
 client.tokenize(
-  text: "hello world!"
+  model: "command-r-plus-08-2024",
+  text: "Hello, world!"
 )
 ```
 
@@ -154,7 +164,8 @@ client.tokenize(
 
 ```ruby
 client.detokenize(
-  tokens: [33555, 1114 , 34]
+  model: "command-r-plus-08-2024",
+  tokens: [33555, 1114, 34]
 )
 ```
 

data/lib/cohere/client.rb

@@ -6,62 +6,56 @@ module Cohere
   class Client
     attr_reader :api_key, :connection
 
-    ENDPOINT_URL = "https://api.cohere.ai/v1"
-
     def initialize(api_key:, timeout: nil)
       @api_key = api_key
       @timeout = timeout
     end
 
+    # Generates a text response to a user message and streams it down, token by token
     def chat(
-      message: nil,
-      model: nil,
+      model:,
+      messages:,
       stream: false,
-      preamble: nil,
-      preamble_override: nil,
-      chat_history: [],
-      conversation_id: nil,
-      prompt_truncation: nil,
-      connectors: [],
-      search_queries_only: false,
+      tools: [],
       documents: [],
-      citation_quality: nil,
-      temperature: nil,
+      citation_options: nil,
+      response_format: nil,
+      safety_mode: nil,
       max_tokens: nil,
-      k: nil,
-      p: nil,
+      stop_sequences: nil,
+      temperature: nil,
       seed: nil,
       frequency_penalty: nil,
       presence_penalty: nil,
-      tools: [],
+      k: nil,
+      p: nil,
+      logprops: nil,
       &block
     )
-      response = connection.post("chat") do |req|
+      response = v2_connection.post("chat") do |req|
         req.body = {}
 
-        req.body[:message] = message if message
-        req.body[:model] = model if model
-        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
-        req.body[:prompt_truncation] = prompt_truncation if prompt_truncation
-        req.body[:connectors] = connectors if connectors
-        req.body[:search_queries_only] = search_queries_only if search_queries_only
-        req.body[:documents] = documents if documents
-        req.body[:citation_quality] = citation_quality if citation_quality
-        req.body[:temperature] = temperature if temperature
+        req.body[:model] = model
+        req.body[:messages] = messages if messages
+        req.body[:tools] = tools if tools.any?
+        req.body[:documents] = documents if documents.any?
+        req.body[:citation_options] = citation_options if citation_options
+        req.body[:response_format] = response_format if response_format
+        req.body[:safety_mode] = safety_mode if safety_mode
         req.body[:max_tokens] = max_tokens if max_tokens
-        req.body[:k] = k if k
-        req.body[:p] = p if p
+        req.body[:stop_sequences] = stop_sequences if stop_sequences
+        req.body[:temperature] = temperature if temperature
         req.body[:seed] = seed if seed
         req.body[:frequency_penalty] = frequency_penalty if frequency_penalty
         req.body[:presence_penalty] = presence_penalty if presence_penalty
-        req.body[:tools] = tools if tools
+        req.body[:k] = k if k
+        req.body[:p] = p if p
+        req.body[:logprops] = logprops if logprops
+
+        if stream || block
+          req.body[:stream] = true
+          req.options.on_data = block if block
+        end
       end
       response.body
     end
@@ -84,7 +78,7 @@ module Cohere
       logit_bias: nil,
       truncate: nil
     )
-      response = connection.post("generate") do |req|
+      response = v1_connection.post("generate") do |req|
         req.body = {prompt: prompt}
         req.body[:model] = model if model
         req.body[:num_generations] = num_generations if num_generations
@@ -104,36 +98,44 @@ module Cohere
       response.body
     end
 
+    # This endpoint returns text embeddings. An embedding is a list of floating point numbers that captures semantic information about the text that it represents.
     def embed(
-      texts:,
-      model: nil,
-      input_type: nil,
+      model:,
+      input_type:,
+      embedding_types:,
+      texts: nil,
+      images: nil,
       truncate: nil
     )
-      response = connection.post("embed") do |req|
-        req.body = {texts: texts}
-        req.body[:model] = model if model
-        req.body[:input_type] = input_type if input_type
+      response = v2_connection.post("embed") do |req|
+        req.body = {
+          model: model,
+          input_type: input_type,
+          embedding_types: embedding_types
+        }
+        req.body[:texts] = texts if texts
+        req.body[:images] = images if images
         req.body[:truncate] = truncate if truncate
       end
       response.body
     end
 
+    # This endpoint takes in a query and a list of texts and produces an ordered array with each text assigned a relevance score.
     def rerank(
+      model:,
       query:,
       documents:,
-      model: nil,
       top_n: nil,
       rank_fields: nil,
       return_documents: nil,
       max_chunks_per_doc: nil
     )
-      response = connection.post("rerank") do |req|
+      response = v2_connection.post("rerank") do |req|
         req.body = {
+          model: model,
           query: query,
           documents: documents
         }
-        req.body[:model] = model if model
         req.body[:top_n] = top_n if top_n
         req.body[:rank_fields] = rank_fields if rank_fields
         req.body[:return_documents] = return_documents if return_documents
@@ -142,41 +144,44 @@ module Cohere
       response.body
     end
 
+    # This endpoint makes a prediction about which label fits the specified text inputs best.
     def classify(
+      model:,
       inputs:,
-      examples:,
-      model: nil,
-      present: nil,
+      examples: nil,
+      preset: nil,
       truncate: nil
     )
-      response = connection.post("classify") do |req|
+      response = v1_connection.post("classify") do |req|
         req.body = {
-          inputs: inputs,
-          examples: examples
+          model: model,
+          inputs: inputs
         }
-        req.body[:model] = model if model
-        req.body[:present] = present if present
+        req.body[:examples] = examples if examples
+        req.body[:preset] = preset if preset
         req.body[:truncate] = truncate if truncate
       end
       response.body
     end
 
-    def tokenize(text:, model: nil)
-      response = connection.post("tokenize") do |req|
-        req.body = model.nil? ? {text: text} : {text: text, model: model}
+    # This endpoint splits input text into smaller units called tokens using byte-pair encoding (BPE).
+    def tokenize(text:, model:)
+      response = v1_connection.post("tokenize") do |req|
+        req.body = {text: text, model: model}
       end
       response.body
     end
 
-    def detokenize(tokens:, model: nil)
-      response = connection.post("detokenize") do |req|
-        req.body = model.nil? ? {tokens: tokens} : {tokens: tokens, model: model}
+    # This endpoint takes tokens using byte-pair encoding and returns their text representation.
+    def detokenize(tokens:, model:)
+      response = v1_connection.post("detokenize") do |req|
+        req.body = {tokens: tokens, model: model}
       end
       response.body
     end
 
     def detect_language(texts:)
-      response = connection.post("detect-language") do |req|
+      response = v1_connection.post("detect-language") do |req|
         req.body = {texts: texts}
       end
       response.body
@@ -191,7 +196,7 @@ module Cohere
       temperature: nil,
       additional_command: nil
     )
-      response = connection.post("summarize") do |req|
+      response = v1_connection.post("summarize") do |req|
         req.body = {text: text}
         req.body[:length] = length if length
         req.body[:format] = format if format
@@ -205,17 +210,22 @@ module Cohere
 
     private
 
-    # standard:disable Lint/DuplicateMethods
-    def connection
-      @connection ||= Faraday.new(url: ENDPOINT_URL, request: {timeout: @timeout}) do |faraday|
-        if api_key
-          faraday.request :authorization, :Bearer, api_key
-        end
+    def v1_connection
+      @connection ||= Faraday.new(url: "https://api.cohere.ai/v1", request: {timeout: @timeout}) do |faraday|
+        faraday.request :authorization, :Bearer, api_key
+        faraday.request :json
+        faraday.response :json, content_type: /\bjson$/
+        faraday.adapter Faraday.default_adapter
+      end
+    end
+
+    def v2_connection
+      @connection ||= Faraday.new(url: "https://api.cohere.com/v2", request: {timeout: @timeout}) do |faraday|
+        faraday.request :authorization, :Bearer, api_key
         faraday.request :json
         faraday.response :json, content_type: /\bjson$/
         faraday.adapter Faraday.default_adapter
       end
     end
-    # standard:enable Lint/DuplicateMethods
   end
 end

data/lib/cohere/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: cohere-ruby
 version: !ruby/object:Gem::Version
-  version: 0.9.11
+  version: 1.0.1
 platform: ruby
 authors:
 - Andrei Bondarev
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-08-01 00:00:00.000000000 Z
+date: 2024-11-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday
@@ -37,6 +37,7 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- ".env.example"
 - ".rspec"
 - CHANGELOG.md
 - Gemfile