gen-ai 0.4.0.alpha.3 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c449d6e75aa475881a045c2a03cf1b9fb0847c4d2c532478fbfa71f8829abaca
-  data.tar.gz: 3aae7fcb6cb92ebe5bd160ed541815e715cabdaecbe3def48b65c92ada744aaf
+  metadata.gz: 43e7d6407d84beec138ffb2a176910fd1cbe25b0f67cc1cf04f7cd462a615617
+  data.tar.gz: 3ed1891455618eb80e9410a455d954ab547092d3f3aaa3c2302857917456f9b8
 SHA512:
-  metadata.gz: a0c3b72951345e13c6840aa2982b325d54bea8c8ee4fdca7a6b8759fd7453f0d92c5405f3598da4542431b615293dec76099d08baee59239b8304d162cca5ec0
-  data.tar.gz: 10b9d0e1328e2d19adbad9cb92c1f86aa545634195bcb33db0e1a52ed1b877723b86c5acd8d13f8ff6eaa022aef2e46fb049b204cd21db654ce4de482061eca5
+  metadata.gz: d690ce67be4296396f0f07e0c0c8873c19c3477191f23a2043b95ef4d771f57da159e96c26c599da09c5b5ca5d65769b2156c7a625771b5ac7dab310c9ced46b
+  data.tar.gz: 355662f601a9bf54f45afa1e026004cee58561d5581bb621f74da11a479081566e2260f1db4a969a126f68e6b825a7ad6ee339233ec5b2bd81438110cb290b4e

data/README.md

@@ -32,10 +32,12 @@ require 'gen_ai'
 
 Language models capabilities
 
-| Provider         | Embedding | Completion | Conversation | Sentiment | Summarization |
-| ---------------- | :-------: | :--------: | :----------: | :-------: | :-----------: |
-| **OpenAI**       |    ✅     |     ✅     |      ✅      |    🛠️     |      🛠️       |
-| **Google Palm2** |    ✅     |     ✅     |      ✅      |    🛠️     |      🛠️       |
+| Provider          | Embedding | Completion | Conversation | Sentiment | Summarization |
+| ----------------- | :-------: | :--------: | :----------: | :-------: | :-----------: |
+| **OpenAI**        |    ✅     |     ✅     |      ✅      |    🛠️     |      🛠️       |
+| **Google Palm2**  |    ✅     |     ✅     |      ✅      |    🛠️     |      🛠️       |
+| **Google Gemini** |    ❌     |     🛠️     |      ✅      |    🛠️     |      🛠️       |
+| **Anthropic**     |    ❌     |     ✅     |      ✅      |    🛠️     |      🛠️       |
 
 Image generation model capabilities
 
@@ -86,21 +88,34 @@ result.values
 
 ```
 
-Have a **conversation** with Large Language Model.
+### Chat
+Have a **conversation** with Large Language Model and Build your own AI chatbot.
+
+Setting a context for the conversation is optional, but it helps the model to understand the topic of the conversation.
 
 ```ruby
-result = model.chat('Hi, how are you?')
+chat = GenAI::Chat.new(:open_ai, ENV['OPEN_AI_TOKEN'])
+chat.start(context: "You are a chat bot named Erl")
+chat.message("Hi, what's your name")
 # = >#<GenAI::Result:0x0000000106ff3d20...>
 
 result.value
-# => "Hello! I'm an AI, so I don't have feelings, but I'm here to help. How can I assist you today?"
+# => "I am a chatbot and you can call me Erl. How can I help you?""
+
+```
 
+
+Provider a history of the conversation to the model to help it understand the context of the conversation.
+
+```ruby
 history = [
     {role: 'user', content: 'What is the capital of Great Britain?'},
     {role: 'assistant', content: 'London'},
 ]
 
-result = model.chat("what about France?", history: history)
+chat = GenAI::Chat.new(:open_ai, ENV['OPEN_AI_TOKEN'])
+result = model.start(history: history)
+result = model.message("what about France?")
 # => #<GenAI::Result:0x00000001033c3bc0...>
 
 result.value

data/lib/gen_ai/api/client.rb

@@ -6,9 +6,10 @@ require 'faraday/multipart'
 module GenAI
   module Api
     class Client
-      def initialize(url:, token:)
+      def initialize(url:, token:, headers: {})
         @url = url
         @token = token
+        @headers = headers
       end
 
       def post(path, body, options = {})
@@ -28,19 +29,24 @@ module GenAI
 
       private
 
-      attr_reader :url, :token
+      attr_reader :url, :token, :headers
 
       def connection(multipart: false)
-        Faraday.new(url: url, headers: {
-          'Accept' => 'application/json',
-          'Content-Type' => multipart ? 'multipart/form-data' : 'application/json',
-          'Authorization' => "Bearer #{token}"
-        }) do |conn|
+        Faraday.new(url: url, headers: build_headers(token, headers, multipart)) do |conn|
           conn.request :multipart if multipart
           conn.request :url_encoded
         end
       end
 
+      def build_headers(token, headers, multipart)
+        hash = {
+          'Accept' => 'application/json',
+          'Content-Type' => multipart ? 'multipart/form-data' : 'application/json'
+        }
+        hash['Authorization'] = "Bearer #{token}" if token
+        hash.merge(headers)
+      end
+
       def handle_response
         response = yield
 

data/lib/gen_ai/api/format/anthropic.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module GenAI
+  module Api
+    module Format
+      module Anthropic
+        def format_messages(messages)
+          messages.map(&:deep_symbolize_keys)
+        end
+
+        def extract_completions(response)
+          if response['type'] == 'completion'
+            [response['completion'].strip]
+          else
+            response['content'].map { |item| item['text'] }
+          end
+        end
+      end
+    end
+  end
+end

data/lib/gen_ai/api/format/gemini.rb

@@ -4,12 +4,19 @@ module GenAI
   module Api
     module Format
       module Gemini
+        USER_ROLE = 'user'
+        ASSISTANT_ROLE = 'model'
+
         def format_messages(messages)
-          messages.map { |message| transform_message(message) }
+          messages.map { |message| transform_message(message.deep_symbolize_keys) }
         end
 
         def transform_message(message)
-          { role: role_for(message), parts: [text: message[:content]] }
+          if message.keys == %i[role content]
+            { role: role_for(message), parts: [text: message[:content]] }
+          else
+            message
+          end
         end
 
         def extract_completions(response)
@@ -19,7 +26,7 @@ module GenAI
         private
 
         def role_for(message)
-          message[:role] == 'user' ? self.class::USER_ROLE : self.class::ASSISTANT_ROLE
+          message[:role] == 'user' ? USER_ROLE : ASSISTANT_ROLE
         end
       end
     end

data/lib/gen_ai/chat/anthropic.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+module GenAI
+  class Chat
+    class Anthropic < Base
+      SYSTEM_ROLE = 'system'
+
+      private
+
+      def build_history(messages, context, examples)
+        @default_options[:system] = context
+        history = []
+        history.concat(examples)
+        history.concat(messages)
+        history
+      end
+
+      def role(message)
+        message[:role]
+      end
+
+      def transform_message(message)
+        message
+      end
+
+      def append_to_message(message)
+        @history.last[:content] = "#{@history.last[:content]}\n#{message}"
+      end
+    end
+  end
+end

data/lib/gen_ai/chat/base.rb

@@ -6,23 +6,26 @@ module GenAI
       USER_ROLE = 'user'
       ASSISTANT_ROLE = 'assistant'
 
+      attr_reader :history, :default_options
+
       def initialize(provider:, token:, options: {})
         @history = []
+        @default_options = {}
         @model = GenAI::Language.new(provider, token, options: options)
       end
 
       def start(history: [], context: nil, examples: [])
-        @history = build_history(history.map(&:deep_symbolize_keys!), context, examples.map(&:deep_symbolize_keys!))
+        @history = build_history(history.map(&:deep_symbolize_keys), context, examples.map(&:deep_symbolize_keys))
       end
 
       def message(message, options = {})
-        if @history.size == 1
+        if @history.size == 1 && @history.first[:role] != 'system'
           append_to_message(message)
         else
           append_to_history({ role: USER_ROLE, content: message })
         end
 
-        response = @model.chat(@history.dup, options)
+        response = @model.chat(@history.dup, default_options.merge(options).compact)
         append_to_history({ role: ASSISTANT_ROLE, content: response.value })
         response
       end

data/lib/gen_ai/chat/gemini.rb

@@ -5,9 +5,6 @@ module GenAI
     class Gemini < Base
       include GenAI::Api::Format::Gemini
 
-      USER_ROLE = 'user'
-      ASSISTANT_ROLE = 'model'
-
       private
 
       def build_history(messages, context, examples)

data/lib/gen_ai/language/anthropic.rb

@@ -0,0 +1,52 @@
+# frozen_string_literal: true
+
+require 'faraday'
+
+module GenAI
+  class Language
+    class Anthropic < Base
+      include GenAI::Api::Format::Anthropic
+
+      BASE_API_URL = 'https://api.anthropic.com'
+      ANTHROPIC_VERSION = '2023-06-01'
+      ANTHROPIC_BETA = 'messages-2023-12-15'
+      COMPLETION_MODEL = 'claude-2.1'
+      DEFAULT_MAX_TOKENS = 1024
+
+      def initialize(token:, options: {})
+        @token = token
+        build_client(token)
+      end
+
+      def complete(prompt, options = {})
+        response = client.post '/v1/complete', {
+          prompt: "\n\nHuman: #{prompt}\n\nAssistant:",
+          model: options.delete(:model) || COMPLETION_MODEL,
+          max_tokens_to_sample: options.delete(:max_tokens_to_sample) || DEFAULT_MAX_TOKENS
+        }.merge(options)
+
+        build_result(model: COMPLETION_MODEL, raw: response, parsed: extract_completions(response))
+      end
+
+      def chat(messages, options = {})
+        response = client.post '/v1/messages', {
+          messages: format_messages(messages),
+          model: options.delete(:model) || COMPLETION_MODEL,
+          max_tokens: options.delete(:max_tokens) || DEFAULT_MAX_TOKENS
+        }.merge(options)
+
+        build_result(model: COMPLETION_MODEL, raw: response, parsed: extract_completions(response))
+      end
+
+      private
+
+      def build_client(token)
+        @client = GenAI::Api::Client.new(url: BASE_API_URL, token: nil, headers: {
+          'anthropic-beta' => ANTHROPIC_BETA,
+          'anthropic-version' => ANTHROPIC_VERSION,
+          'x-api-key' => token
+        })
+      end
+    end
+  end
+end

data/lib/gen_ai/language/gemini.rb

@@ -2,6 +2,7 @@
 
 require 'faraday'
 
+require 'pry'
 module GenAI
   class Language
     class Gemini < Base
@@ -18,7 +19,7 @@ module GenAI
 
       def chat(messages, options = {})
         response = client.post "/v1beta/models/gemini-pro:generateContent?key=#{@token}", {
-          contents: messages.map(&:deep_symbolize_keys!),
+          contents: format_messages(messages),
           generationConfig: options.except(:model)
         }
 

data/lib/gen_ai/language/open_ai.rb

@@ -32,7 +32,7 @@ module GenAI
 
       def chat(messages, options = {})
         parameters = {
-          messages: messages.map(&:deep_symbolize_keys!),
+          messages: messages.map(&:deep_symbolize_keys),
           model: options.delete(:model) || COMPLETION_MODEL
         }.merge(options)
 

data/lib/gen_ai/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module GenAI
-  VERSION = '0.4.0.alpha.3'
+  VERSION = '0.4.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: gen-ai
 version: !ruby/object:Gem::Version
-  version: 0.4.0.alpha.3
+  version: 0.4.0
 platform: ruby
 authors:
 - Alex Chaplinsky
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-01-03 00:00:00.000000000 Z
+date: 2024-01-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -113,10 +113,12 @@ files:
 - lib/gen/ai.rb
 - lib/gen_ai.rb
 - lib/gen_ai/api/client.rb
+- lib/gen_ai/api/format/anthropic.rb
 - lib/gen_ai/api/format/gemini.rb
 - lib/gen_ai/api/format/open_ai.rb
 - lib/gen_ai/base.rb
 - lib/gen_ai/chat.rb
+- lib/gen_ai/chat/anthropic.rb
 - lib/gen_ai/chat/base.rb
 - lib/gen_ai/chat/gemini.rb
 - lib/gen_ai/chat/google_palm.rb
@@ -127,6 +129,7 @@ files:
 - lib/gen_ai/image/open_ai.rb
 - lib/gen_ai/image/stability_ai.rb
 - lib/gen_ai/language.rb
+- lib/gen_ai/language/anthropic.rb
 - lib/gen_ai/language/base.rb
 - lib/gen_ai/language/gemini.rb
 - lib/gen_ai/language/google_palm.rb
@@ -153,9 +156,9 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: 2.7.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
-  - - ">"
+  - - ">="
     - !ruby/object:Gem::Version
-      version: 1.3.1
+      version: '0'
 requirements: []
 rubygems_version: 3.3.7
 signing_key: