zuno 0.0.4 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8d1704ad858e219dc6216e0dd8ac6bdaf8b57cb0052c1a0e4d6dfaed794b75f1
-  data.tar.gz: 4580bf2697e8746aad1de8e4f9e42c87be944f00e81fc93758cb299c1fa9e76e
+  metadata.gz: 77955f7ab70bc212f516b3a7590a00eb03c3790df12a3e2eecb823dd7d1ffea8
+  data.tar.gz: 6e50ab43d348817f32c2b589204bc582a0f071d3eeb483bc294f759211e05509
 SHA512:
-  metadata.gz: 13e4b4687664ad655c7125933e37e8adaa0e7440c41b2e9ff3452ba87446e889e00f3c2727189a466340d7756d2ce0f947c50274ec6a5f282b4308d14969b942
-  data.tar.gz: ee873e43504615c7ce040e3952ab68e5ab649eda573b035cf1b27e164a10a521657484895c7d93a481803517952eff3bce9180ee440711605c039edeea70cf3d
+  metadata.gz: 98da00fecffe4ea2a97b71bc6425c69b0da202e57665f58c4b557a1740b7c44555c2e5508b7aaf63a3e83d1647137e482a5133ef034220d7402cbe9566a166c9
+  data.tar.gz: c51e1ee3944c3c63233cfbfdaee925406f169fc9d7467b4aaaa5ac33f5967255a0c5ce6117a4e1b50498dfcac764c1bf26c490428956edb00e0975d0f6e8cc8a

data/lib/providers/groq_cloud.rb

@@ -27,5 +27,40 @@ 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"]
+      elsif parsed_response["choices"][0]["message"]["content"]
+        OpenStruct.new(response: parsed_response["choices"][0]["message"]["content"])
+      elsif parsed_response["choices"][0]["message"]["tool_calls"]
+        OpenStruct.new(tool_calls: parsed_response["choices"][0]["message"]["tool_calls"])
+      end
+    end
   end
 end

data/lib/providers/openai.rb

@@ -24,8 +24,10 @@ module Providers
 
       if response.body["error"]
         raise response.body["error"]["message"]
-      else
+      elsif response.body["choices"][0]["message"]["content"]
         OpenStruct.new(response: response.body["choices"][0]["message"]["content"])
+      elsif response.body["choices"][0]["message"]["tool_calls"]
+        OpenStruct.new(tool_calls: response.body["choices"][0]["message"]["tool_calls"])
       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.1.0"
 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,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: zuno
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.1.0
 platform: ruby
 authors:
 - John Paul
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-07-05 00:00:00.000000000 Z
+date: 2024-07-06 00:00:00.000000000 Z
 dependencies: []
 description: AI toolkit for Ruby
 email:
@@ -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: