zuno 0.0.4 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8d1704ad858e219dc6216e0dd8ac6bdaf8b57cb0052c1a0e4d6dfaed794b75f1
-  data.tar.gz: 4580bf2697e8746aad1de8e4f9e42c87be944f00e81fc93758cb299c1fa9e76e
+  metadata.gz: 378df9dab08b062608ea23bba7f54f4abccf4604e76335d7f4a33a1fcb6e3f8e
+  data.tar.gz: 83e578be4532ad403e419a4419919ddde40692a7aa7778fdb3be0ac0dad2f26f
 SHA512:
-  metadata.gz: 13e4b4687664ad655c7125933e37e8adaa0e7440c41b2e9ff3452ba87446e889e00f3c2727189a466340d7756d2ce0f947c50274ec6a5f282b4308d14969b942
-  data.tar.gz: ee873e43504615c7ce040e3952ab68e5ab649eda573b035cf1b27e164a10a521657484895c7d93a481803517952eff3bce9180ee440711605c039edeea70cf3d
+  metadata.gz: 9a9b863540c88a01a0b33a1f5f6444f74a80644805d407b427169a8dbdc5760004d58e9dc66f70136bdc2df3e20717dae61a4101b5143367f5714fc351ca1256
+  data.tar.gz: 70ec8841a22fbefd2cf9e4926f088a6ff674ccef5277d33e32ff8658641b1c3d7b8ffb8b8c13f1add0cb2ce1759dcc69187307c474361dd5727dbd6ad2a6999e

data/lib/providers/groq_cloud.rb

@@ -27,5 +27,38 @@ module Providers
 
       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/zuno/chat.rb

@@ -6,9 +6,10 @@ 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-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:, **options)
+    def chat(messages:, model: nil, **options)
       model ||= Zuno.configuration.chat_completion_model
       provider = provider_for_model(model)
       provider.chat_completion(messages, model, options)
@@ -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/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.4"
+  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.4
+  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: