cohere-ruby 0.9.6 → 0.9.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 506af84fa7319c6087e3d3b5c2c3a4c29663d01fda670dd0c66556f49e2fb591
-  data.tar.gz: 455b150a1831b3bfb0a2793dc0de602b4aa3460faddb868ced8050fe347d19c2
+  metadata.gz: 83208ee6cbdf496180ac2e6ed258388a74a266d86188b81ceb011bf5f2b61f44
+  data.tar.gz: 9f0ff2d20ea3c988663bc1c104f46f0c94a6d8d833d4a0ebf1cb81f765d8502d
 SHA512:
-  metadata.gz: 3146c898ed6d61a010f7876939d116ee78c16c859a9cb9cf7f8206f5795fabd327eb16770cbf7c7a626a102d27efbe570c3a351ed9eae00357b5c90946404641
-  data.tar.gz: 8ceb823d98361cc736a9c31e569cd2453c0ce10b7fcd696ef2ee6cdb07effd8cc2d491cf2d8b4ff97b15214293ddbe3a1ada728ed65563966bdf51035be69b90
+  metadata.gz: a0425fe338748ba829c6d15b3fe319c23cf6fe47ab18c885476dcccffe969888c215162d9139768e53ed569b5e6799c9abc17ecb028551f3cd48c2f92cec49e6
+  data.tar.gz: 48add01bf4d9ee6a43ee27809efe2eb352eee73bceac26b6fcfff29ae060063b8e3f4dc5803e317d80a612d0f6890ac18ca1d90cf5fd62254e3d1bc6f1dce8de

data/CHANGELOG.md

@@ -1,5 +1,8 @@
 ## [Unreleased]
 
+## [0.9.8] - 2024-02-10
+- Adding method for /chat endpoint
+
 ## [0.1.0] - 2023-04-23
 
 - Initial release

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    cohere-ruby (0.9.6)
+    cohere-ruby (0.9.8)
       faraday (>= 2.0.1, < 3.0)
 
 GEM
@@ -67,6 +67,7 @@ GEM
     unicode-display_width (2.4.2)
 
 PLATFORMS
+  arm64-darwin-23
   x86_64-darwin-19
   x86_64-darwin-21
   x86_64-linux

data/README.md

@@ -33,7 +33,7 @@ If bundler is not being used to manage dependencies, install the gem by executin
 require "cohere"
 
 client = Cohere::Client.new(
-  api_key: ENV['COHERE_API_KEY']
+  ENV['COHERE_API_KEY']
 )
 ```
 ### Generate
@@ -43,6 +43,13 @@ client.generate(
 )
 ```
 
+### Chat
+```ruby
+client.chat(
+    message: "Hey! How are you?"
+)
+```
+
 ### Embed
 ```ruby
 client.embed(

data/lib/cohere/client.rb

@@ -8,10 +8,51 @@ module Cohere
 
     ENDPOINT_URL = "https://api.cohere.ai/v1"
 
-    def initialize(api_key:)
+    def initialize(api_key)
       @api_key = api_key
     end
 
+    def chat(
+      message:,
+      model: nil,
+      stream: false,
+      preamble_override: nil,
+      chat_history: [],
+      conversation_id: nil,
+      prompt_truncation: nil,
+      connectors: [],
+      search_queries_only: false,
+      documents: [],
+      citation_quality: nil,
+      temperature: nil,
+      max_tokens: nil,
+      k: nil,
+      p: nil,
+      frequency_penalty: nil,
+      presence_penalty: nil
+    )
+      response = connection.post("chat") do |req|
+        req.body = {message: message}
+        req.body[:model] = model if model
+        req.body[:stream] = stream if stream
+        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[:max_tokens] = max_tokens if max_tokens
+        req.body[:k] = k if k
+        req.body[:p] = p if p
+        req.body[:frequency_penalty] = frequency_penalty if frequency_penalty
+        req.body[:presence_penalty] = presence_penalty if presence_penalty
+      end
+      response.body
+    end
+
     # This endpoint generates realistic text conditioned on a given input.
     def generate(
       prompt:,
@@ -53,11 +94,13 @@ module Cohere
     def embed(
       texts:,
       model: nil,
+      input_type: 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
         req.body[:truncate] = truncate if truncate
       end
       response.body

data/lib/cohere/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: cohere-ruby
 version: !ruby/object:Gem::Version
-  version: 0.9.6
+  version: 0.9.8
 platform: ruby
 authors:
 - Andrei Bondarev
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-08-10 00:00:00.000000000 Z
+date: 2024-02-11 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.2.3
+rubygems_version: 3.5.3
 signing_key:
 specification_version: 4
 summary: Cohere API client for Ruby.