zuno 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 51b2206418e88cbbfcd5f130d8e7d0f8486773317bd463e639ddcdf1ded31dd9
+  data.tar.gz: db8b41989dbec0e5f07405a567c56de04b0ed6badb4537514845574ab1afbc37
+SHA512:
+  metadata.gz: 37a62efd042c8511fbe7553a8dcf4ab880886d816317ed31bc5f0800b1554d6205af300394841e6c238700a8b6018ba4be33ec55af8a87c9c983dac773346368
+  data.tar.gz: '028768a79656ab3cb2695ee2d475a0627317eb6cbc87817a782dd0aa6048081c77cc24e4a124d72a00cffae0217c6a516076ef30a33376f86bc8702e04edcec0'

data/lib/providers/anthropic.rb

@@ -0,0 +1,26 @@
+module Providers
+  class Anthropic
+    def initialize
+      @connection = Faraday.new(url: "https://api.anthropic.com") do |faraday|
+        faraday.request :json
+        faraday.response :json
+        faraday.adapter Faraday.default_adapter
+      end
+
+      @api_key = Zuno.configuration.openai_api_key
+    end
+
+    def chat_completion(messages, model, options = {})
+      response = @connection.post("/v1/chat/completions") do |req|
+        req.headers["Content-Type"] = "application/json"
+        req.headers["Authorization"] = "Bearer #{@api_key}"
+        req.body = {
+          model: model,
+          messages: messages
+        }.merge(options).to_json
+      end
+
+      { response: response.body["choices"][0]["message"]["content"] }
+    end
+  end
+end

data/lib/providers/openai.rb

@@ -0,0 +1,26 @@
+module Providers
+  class OpenAI
+    def initialize
+      @connection = Faraday.new(url: "https://api.openai.com") do |faraday|
+        faraday.request :json
+        faraday.response :json
+        faraday.adapter Faraday.default_adapter
+      end
+
+      @api_key = Zuno.configuration.openai_api_key
+    end
+
+    def chat_completion(messages, model, options = {})
+      response = @connection.post("/v1/chat/completions") do |req|
+        req.headers["Content-Type"] = "application/json"
+        req.headers["Authorization"] = "Bearer #{@api_key}"
+        req.body = {
+          model: model,
+          messages: messages
+        }.merge(options).to_json
+      end
+
+      { response: response.body["choices"][0]["message"]["content"] }
+    end
+  end
+end

data/lib/zuno/chat.rb

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+require "providers/openai"
+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
+
+  class << self
+    def chat(messages:, model:)
+      model ||= Zuno.configuration.chat_completion_model
+      provider = provider_for_model(model)
+      provider.chat_completion(messages, model)
+    end
+
+    private
+
+    def provider_for_model(model)
+      case model
+      when *OPENAI_MODELS then Providers::OpenAI.new
+      when *ANTHROPIC_MODELS then Providers::Anthropic.new
+      else
+        raise ArgumentError, "Unsupported model: #{model}"
+      end
+    end
+
+    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] }
+      }
+    end
+  end
+end

data/lib/zuno/configuration.rb

@@ -0,0 +1,9 @@
+module Zuno
+  class Configuration
+    attr_accessor :chat_completion_model, :openai_api_key, :anthropic_api_key
+
+    def initialize
+      @openai_api_key = nil
+    end
+  end
+end

data/lib/zuno/version.rb

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

data/lib/zuno.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+require_relative "zuno/version"
+require_relative "zuno/configuration"
+require "zuno/chat"
+require "faraday"
+
+module Zuno
+  class << self
+    attr_accessor :configuration
+  end
+
+  def self.configure
+    self.configuration ||= Configuration.new
+    yield(configuration)
+  end
+end

metadata

@@ -0,0 +1,50 @@
+--- !ruby/object:Gem::Specification
+name: zuno
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- John Paul
+autorequire:
+bindir: exe
+cert_chain: []
+date: 2024-06-24 00:00:00.000000000 Z
+dependencies: []
+description: AI toolkit for Ruby
+email:
+- johnarpaul@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/providers/anthropic.rb
+- lib/providers/openai.rb
+- lib/zuno.rb
+- lib/zuno/chat.rb
+- lib/zuno/configuration.rb
+- lib/zuno/version.rb
+homepage: https://github.com/dqnamo
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/dqnamo
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.0.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.4.10
+signing_key:
+specification_version: 4
+summary: AI toolkit for Ruby
+test_files: []