Cohere

require "cohere"

client = Cohere::Client.new(
  api_key: ENV['COHERE_API_KEY']
)

client.generate(
  prompt: "Once upon a time in a magical land called"
)

client.chat(
  message: "Hey! How are you?"
)

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

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,
)

client.embed(
  texts: ["hello!"]
)

docs = [
  "Carson City is the capital city of the American state of Nevada.",
  "The Commonwealth of the Northern Mariana Islands is a group of islands in the Pacific Ocean. Its capital is Saipan.",
  "Capitalization or capitalisation in English grammar is the use of a capital letter at the start of a word. English usage varies from capitalization in other languages.",
  "Washington, D.C. (also known as simply Washington or D.C., and officially as the District of Columbia) is the capital of the United States. It is a federal district.",
  "Capital punishment (the death penalty) has existed in the United States since beforethe United States was a country. As of 2017, capital punishment is legal in 30 of the 50 states.",
]

client.rerank(
  query: "What is the capital of the United States?",
  documents: docs
)

examples = [
  { text: "Dermatologists don't like her!", label: "Spam" },
  { text: "Hello, open to this?", label: "Spam" },
  { text: "I need help please wire me $1000 right now", label: "Spam" },
  { text: "Nice to know you ;)", label: "Spam" },
  { text: "Please help me?", label: "Spam" },
  { text: "Your parcel will be delivered today", label: "Not spam" },
  { text: "Review changes to our Terms and Conditions", label: "Not spam" },
  { text: "Weekly sync notes", label: "Not spam" },
  { text: "Re: Follow up from today's meeting", label: "Not spam" },
  { text: "Pre-read for tomorrow", label: "Not spam" }
]

inputs = [
  "Confirm your email address",
  "hey i need u to send some $",
]

client.classify(
  examples: examples,
  inputs: inputs
)

client.tokenize(
  text: "hello world!"
)

client.detokenize(
  tokens: [33555, 1114 , 34]
)

client.detect_language(
  texts: ["Здравствуй, Мир"]
)

client.summarize(
  text: "..."
)