zuno 0.0.2 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: cf1792df415cee7140b1c93937b432fa7dbcc1d9f36ac3f13484e32712f4ec50
-  data.tar.gz: f687effb232e1ac3f920102fe06be674709d740c073c85174ffe698abd918883
+  metadata.gz: 8d1704ad858e219dc6216e0dd8ac6bdaf8b57cb0052c1a0e4d6dfaed794b75f1
+  data.tar.gz: 4580bf2697e8746aad1de8e4f9e42c87be944f00e81fc93758cb299c1fa9e76e
 SHA512:
-  metadata.gz: 44f45a9d8f4067bb258436a4b9434c67629bb0c1a23049423b7039ec48fd923b5b0be327abb3547771caaa825407fe8c872e440158d931486848a6fb538498b4
-  data.tar.gz: 34dec504254b3149893db199bd149f319c95a427cb56d8d82bb7e4ebfd399dfebbc90781392a0d7b85475d64cf3ea144158a0b5ab6d001a61251718ca1d2c26b
+  metadata.gz: 13e4b4687664ad655c7125933e37e8adaa0e7440c41b2e9ff3452ba87446e889e00f3c2727189a466340d7756d2ce0f947c50274ec6a5f282b4308d14969b942
+  data.tar.gz: ee873e43504615c7ce040e3952ab68e5ab649eda573b035cf1b27e164a10a521657484895c7d93a481803517952eff3bce9180ee440711605c039edeea70cf3d

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,7 @@ module Providers
         }
       )
 
-      { text: JSON.parse(response.body)["text"] }
+      OpenStruct.new(text: JSON.parse(response.body)["text"])
     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,13 @@ 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
 
   class << self
-    def chat(messages:, model:)
+    def chat(messages:, model:, **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

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/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: zuno
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.4
 platform: ruby
 authors:
 - John Paul