zuno 0.0.3 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fd6229b4b78dc25546114b25fdab7237c946dbb5c36e0c40c55f5c411b5cc2b6
-  data.tar.gz: c003ba3200b6236a6eec39d4b9a45b9737599a9c234150ef33ec0697bbfb5482
+  metadata.gz: 378df9dab08b062608ea23bba7f54f4abccf4604e76335d7f4a33a1fcb6e3f8e
+  data.tar.gz: 83e578be4532ad403e419a4419919ddde40692a7aa7778fdb3be0ac0dad2f26f
 SHA512:
-  metadata.gz: f0d6d53335fbebede4f20728e9ce3c067ca3e286980d1d9a978238375e19599c31830cc6fe616b82a4ab662e318a7c567979f6db9a614982141c443bff2f37c1
-  data.tar.gz: 59a861200fef2fc8b85b548d44866aa7530442ec2baac252b3220555d5cf81b6f5caa98b1c9f4944b613e9f16483e24da9e1b83f5b6a5c9bbfa7773da8797d1f
+  metadata.gz: 9a9b863540c88a01a0b33a1f5f6444f74a80644805d407b427169a8dbdc5760004d58e9dc66f70136bdc2df3e20717dae61a4101b5143367f5714fc351ca1256
+  data.tar.gz: 70ec8841a22fbefd2cf9e4926f088a6ff674ccef5277d33e32ff8658641b1c3d7b8ffb8b8c13f1add0cb2ce1759dcc69187307c474361dd5727dbd6ad2a6999e

data/lib/providers/anthropic.rb

@@ -1,3 +1,5 @@
+require 'ostruct'
+
 module Providers
   class Anthropic
     def initialize
@@ -7,20 +9,25 @@ module Providers
         faraday.adapter Faraday.default_adapter
       end
 
-      @api_key = Zuno.configuration.openai_api_key
+      @api_key = Zuno.configuration.anthropic_api_key
     end
 
     def chat_completion(messages, model, options = {})
-      response = @connection.post("/v1/chat/completions") do |req|
+      response = @connection.post("/v1/messages") do |req|
+        req.headers["x-api-key"] = @api_key
         req.headers["Content-Type"] = "application/json"
-        req.headers["Authorization"] = "Bearer #{@api_key}"
+        req.headers["anthropic-version"] = "2023-06-01"
         req.body = {
           model: model,
-          messages: messages
+          messages: messages,
         }.merge(options).to_json
       end
-
-      { response: response.body["choices"][0]["message"]["content"] }
+      
+      if response.body["error"]
+        raise response.body["error"]["message"]
+      else
+        OpenStruct.new(response: response.body["content"][0]["text"])
+      end
     end
   end
 end

data/lib/providers/groq_cloud.rb

@@ -1,5 +1,6 @@
 require 'httparty'
 require 'open-uri'
+require 'ostruct'
 
 module Providers
   class GroqCloud
@@ -24,7 +25,40 @@ module Providers
         }
       )
 
-      { text: JSON.parse(response.body)["text"] }
+      OpenStruct.new(text: JSON.parse(response.body)["text"])
+    end
+
+    def translate(audio)
+      response = self.class.post(
+        '/openai/v1/audio/translations',
+        multipart: true,
+        body: {
+          file: audio,
+          model: 'whisper-large-v3',
+          temperature: 0,
+          response_format: 'json',
+        }
+      )
+
+      OpenStruct.new(text: JSON.parse(response.body)["text"])
+    end
+
+    def chat_completion(messages, model, options = {})
+      response = self.class.post(
+        '/openai/v1/chat/completions',
+        body: {
+          model: model,
+          messages: messages
+        }.merge(options).to_json
+      )
+      
+      parsed_response = JSON.parse(response.body)
+
+      if parsed_response["error"]
+        raise parsed_response["error"]["message"]
+      else
+        OpenStruct.new(response: parsed_response["choices"][0]["message"]["content"])
+      end
     end
   end
 end

data/lib/providers/openai.rb

@@ -1,3 +1,5 @@
+require 'ostruct'
+
 module Providers
   class OpenAI
     def initialize
@@ -20,7 +22,11 @@ module Providers
         }.merge(options).to_json
       end
 
-      { response: response.body["choices"][0]["message"]["content"] }
+      if response.body["error"]
+        raise response.body["error"]["message"]
+      else
+        OpenStruct.new(response: response.body["choices"][0]["message"]["content"])
+      end
     end
   end
 end

data/lib/zuno/chat.rb

@@ -5,13 +5,14 @@ require "providers/anthropic"
 
 module Zuno
   OPENAI_MODELS = %w[gpt-3.5-turbo gpt-4-turbo gpt-4-turbo-preview].freeze
-  ANTHROPIC_MODELS = %w[claude-3-opus-20240229].freeze
+  ANTHROPIC_MODELS = %w[claude-3-5-sonnet-20240620 claude-3-opus-20240229 claude-3-sonnet-20240229 claude-3-haiku-20240307].freeze
+  GROQ_CLOUD_MODELS = %w[llama3-8b-8192 llama3-70b-8192 mixtral-8x7b-32768 gemma-7b-it gemma2-9b-it].freeze
 
   class << self
-    def chat(messages:, model:)
+    def chat(messages:, model: nil, **options)
       model ||= Zuno.configuration.chat_completion_model
       provider = provider_for_model(model)
-      provider.chat_completion(messages, model)
+      provider.chat_completion(messages, model, options)
     end
 
     private
@@ -20,6 +21,7 @@ module Zuno
       case model
       when *OPENAI_MODELS then Providers::OpenAI.new
       when *ANTHROPIC_MODELS then Providers::Anthropic.new
+      when *GROQ_CLOUD_MODELS then Providers::GroqCloud.new
       else
         raise ArgumentError, "Unsupported model: #{model}"
       end
@@ -28,7 +30,8 @@ module Zuno
     def model_providers_mapping
       @model_providers_mapping ||= {
         **OPENAI_MODELS.to_h { |model| [model, Providers::OpenAI.new] },
-        **ANTHROPIC_MODELS.to_h { |model| [model, Providers::Anthropic.new] }
+        **ANTHROPIC_MODELS.to_h { |model| [model, Providers::Anthropic.new] },
+        **GROQ_CLOUD_MODELS.to_h { |model| [model, Providers::GroqCloud.new] }
       }
     end
   end

data/lib/zuno/configuration.rb

@@ -3,7 +3,10 @@ module Zuno
     attr_accessor :chat_completion_model, :openai_api_key, :anthropic_api_key, :groq_cloud_api_key
 
     def initialize
+      @chat_completion_model = nil
       @openai_api_key = nil
+      @anthropic_api_key = nil
+      @groq_cloud_api_key = nil
     end
   end
 end

data/lib/zuno/translation.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+require "providers/groq_cloud"
+
+module Zuno
+  class << self
+    def translate(audio:)
+      Providers::GroqCloud.new.translate(audio)
+    end
+  end
+end

data/lib/zuno/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Zuno
-  VERSION = "0.0.3"
+  VERSION = "0.0.5"
 end

data/lib/zuno.rb

@@ -4,6 +4,7 @@ require_relative "zuno/version"
 require_relative "zuno/configuration"
 require "zuno/chat"
 require "zuno/transcription"
+require "zuno/translation"
 require "faraday"
 
 module Zuno

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: zuno
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.5
 platform: ruby
 authors:
 - John Paul
@@ -24,6 +24,7 @@ files:
 - lib/zuno/chat.rb
 - lib/zuno/configuration.rb
 - lib/zuno/transcription.rb
+- lib/zuno/translation.rb
 - lib/zuno/version.rb
 homepage: https://github.com/dqnamo
 licenses: