aiagent 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: f54f71f8b08b031f351313bcf6fd253fdd2a588ef7d17da74115ff2b72561bb0
+  data.tar.gz: 2a8004689fc829d08872840029e8729be303aaf236f33500e8f979d8b8c4e92d
+SHA512:
+  metadata.gz: bf0f1f4447b4608769fdffe194bf210cd9f409bcf00c387c0e370466b2ca3188e3e3880075ac14267f62597ed62b2389632e7943e410004d80cedf82a0cab3f0
+  data.tar.gz: 44a6f23bfe09e4ce4e3a32ccbf8eecaa935d7b4b63aa59ffd98bb237a574731804036c51a79b42b14fd8a9113a23e4c8ad0b91484f1f938277eeb1f235049c68

data/CHANGELOG.md

@@ -0,0 +1,3 @@
+## [0.1.0] - 2024-07-09
+
+- Initial release

data/Gemfile

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+source "https://rubygems.org"
+
+# Specify your gem's dependencies in the gemspec file
+gemspec
+
+gem "rake", "~> 13.0"
+
+gem "minitest", "~> 5.0"
+
+gem "rubocop", "~> 1.21"
+
+group :test do
+    gem 'simplecov', require: false
+    gem 'simplecov_json_formatter', require: false
+end
+  
+  

data/Gemfile.lock

@@ -0,0 +1,65 @@
+PATH
+  remote: .
+  specs:
+    aiagent (0.1.0)
+      httparty
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    ast (2.4.2)
+    docile (1.4.0)
+    httparty (0.21.0)
+      mini_mime (>= 1.0.0)
+      multi_xml (>= 0.5.2)
+    json (2.7.1)
+    language_server-protocol (3.17.0.3)
+    mini_mime (1.1.5)
+    minitest (5.22.2)
+    multi_xml (0.6.0)
+    parallel (1.24.0)
+    parser (3.3.0.5)
+      ast (~> 2.4.1)
+      racc
+    racc (1.7.3)
+    rainbow (3.1.1)
+    rake (13.1.0)
+    regexp_parser (2.9.0)
+    rexml (3.2.6)
+    rubocop (1.62.0)
+      json (~> 2.3)
+      language_server-protocol (>= 3.17.0)
+      parallel (~> 1.10)
+      parser (>= 3.3.0.2)
+      rainbow (>= 2.2.2, < 4.0)
+      regexp_parser (>= 1.8, < 3.0)
+      rexml (>= 3.2.5, < 4.0)
+      rubocop-ast (>= 1.31.1, < 2.0)
+      ruby-progressbar (~> 1.7)
+      unicode-display_width (>= 2.4.0, < 3.0)
+    rubocop-ast (1.31.2)
+      parser (>= 3.3.0.4)
+    ruby-progressbar (1.13.0)
+    simplecov (0.22.0)
+      docile (~> 1.1)
+      simplecov-html (~> 0.11)
+      simplecov_json_formatter (~> 0.1)
+    simplecov-html (0.12.3)
+    simplecov_json_formatter (0.1.4)
+    unicode-display_width (2.5.0)
+
+PLATFORMS
+  arm64-darwin-22
+  ruby
+
+DEPENDENCIES
+  aiagent!
+  httparty
+  minitest (~> 5.0)
+  rake (~> 13.0)
+  rubocop (~> 1.21)
+  simplecov
+  simplecov_json_formatter
+
+BUNDLED WITH
+   2.5.3

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2024 Web Ventures Ltd (NZ)
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/README.md

@@ -0,0 +1,61 @@
+# aiagent
+
+[![Gem Version](https://badge.fury.io/rb/aiagent.svg)](https://badge.fury.io/rb/aiagent) [![License](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+
+** Example usage coming soon in the next version of this gem **
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'aiagent'
+```
+
+And then execute:
+
+```ruby
+$ bundle install
+```
+
+Or install it yourself as:
+
+```ruby
+$ gem install aiagent
+```
+
+## Initializer
+
+Create an initializer called config/initializers/ai_agent.rb \
+And in that file simply require the agents that you'll use in your project.
+
+E.g. \
+require "ai_agent/ai_agent/claude"
+
+## Usage
+
+To use this gem you'll need an API key for the agents that you want to use.
+
+Set your API keys as environment variables.
+
+## Changelog
+
+For a detailed list of changes for each version of this project, please see the [CHANGELOG](CHANGELOG.md).
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake test` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+
+To install this gem onto your local machine, run `bundle exec rake install`.
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/webventures/aiagent.
+
+## License
+
+The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).
+
+## Trademarks
+
+This gem refers to agents that have been developed by third parties. Ownership of the respective trademarks remains with those parties.

data/Rakefile

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+require "bundler/gem_tasks"
+require "rake/testtask"
+
+Rake::TestTask.new(:test) do |t|
+  t.libs << "test"
+  t.pattern = 'test/**/*_test.rb'
+  t.libs << "lib"
+  t.test_files = FileList["test/**/test_*.rb"]
+end
+
+require "rubocop/rake_task"
+
+RuboCop::RakeTask.new
+
+task default: %i[test rubocop]

data/lib/ai_agent/ai_agent/claude.rb

@@ -0,0 +1,39 @@
+require "ai_agent/base"
+if Gem.loaded_specs.has_key?('claude-ruby')
+  require 'claude/client'
+else
+  Rails.logger.warn "claude-ruby gem is not loaded"
+end
+
+module AiAgent
+  CLAUDE = 'claude'.freeze
+
+  class Claude < Base
+    def initialize
+      super
+      self.agent = CLAUDE
+    end
+
+    def client
+      claude if agent == CLAUDE
+    end
+
+    def send_messages(messages, options)
+      client.messages(messages, options)
+    end
+
+    def format_response(response)
+      client.parse_response(response) rescue "ERROR: Couldn't extract text from Claude response"
+    end
+
+    private
+
+    def claude
+      @claude ||= ::Claude::Client.new(anthropic_api_key)
+    end
+
+    def anthropic_api_key
+      @anthropic_api_key ||= ENV['ANTHROPIC_API_KEY']
+    end
+  end
+end

data/lib/ai_agent/base.rb

@@ -0,0 +1,63 @@
+require 'httparty'
+require 'json'
+
+module AiAgent
+  class Base
+    attr_accessor :agent
+
+    def initialize
+      # be sure to set agent in the subclass initialize method
+    end
+
+    def client
+      raise NotImplementedError, "Subclasses must implement the client method"
+    end
+
+    def send_messages(messages, options)
+      raise NotImplementedError, "Subclasses must implement the send_message method"
+    end
+
+    def format_response(response)
+      raise NotImplementedError, "Subclasses must implement the format_response method"
+    end
+
+    def analyze_sentiment(text, strict: true, options: {})
+      prompt = "Analyze the sentiment of the following text and classify it as positive, negative, or neutral:\n\n#{text}\n\nSentiment: "
+      if strict
+        system = "If you are asked to return a word, then return only that word with no preamble or postamble. " if strict
+        max_tokens = 2
+      else
+        max_tokens = 100
+      end
+
+      send_messages([ { 'role': 'user', 'content': prompt } ],
+                    options.reverse_merge( system: system, max_tokens: max_tokens ))
+    end
+
+    def recognize_entities(text, strict: true, options: {})
+      prompt = "Identify and list the named entities in the following text:\n\n#{text}\n\nEntities: "
+      if strict
+        system = "Be specific in your answer, with no preamble or postamble. If you are asked to list some names, then return only a list of those names, nothing else. "
+        max_tokens = 100
+      else
+        max_tokens = 500
+      end
+
+      send_messages([ { 'role': 'user', 'content': prompt } ],
+                    options.reverse_merge( system: system, max_tokens: max_tokens ))
+    end
+
+    def summarize_text(text, strict: true, options: {})
+      prompt = "Summarize the following text:\n\n#{text}\n\nSummary: "
+      if strict
+        system = "Be specific in your answer, with no preamble or postamble. I.e. return only what the user asks for, nothing else. "
+        max_tokens = 100
+      else
+        max_tokens = 500
+      end
+
+      send_messages([ { 'role': 'user', 'content': prompt } ],
+                    options.reverse_merge( system: system, max_tokens: max_tokens ))
+    end
+  end
+end

data/lib/ai_agent/ruby/version.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+module AiAgent
+  module Ruby
+    VERSION = "0.1.0"
+  end
+end

data/lib/ai_agent/ruby.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+require_relative "ruby/version"
+
+module AiAgent
+  module Ruby
+    class Error < StandardError; end
+  end
+end

data/sig/ai_agent/ruby.rbs

@@ -0,0 +1,6 @@
+module AiAgent
+  module Ruby
+    VERSION: String
+    # See the writing guide of rbs: https://github.com/ruby/rbs#guides
+  end
+end

metadata

@@ -0,0 +1,58 @@
+--- !ruby/object:Gem::Specification
+name: aiagent
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Web Ventures Ltd
+autorequire:
+bindir: exe
+cert_chain: []
+date: 2024-07-08 00:00:00.000000000 Z
+dependencies: []
+description: Ruby SDK for interacting with LLM agents such as OpenAI's ChatGPT, Anthropic's
+  Claude, and Ollama.
+email:
+- webven@mailgab.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- CHANGELOG.md
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- lib/ai_agent/ai_agent/claude.rb
+- lib/ai_agent/base.rb
+- lib/ai_agent/ruby.rb
+- lib/ai_agent/ruby/version.rb
+- sig/ai_agent/ruby.rbs
+homepage: https://github.com/webventures/aiagent.git
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/webventures/aiagent.git
+  source_code_uri: https://github.com/webventures/aiagent.git
+  changelog_uri: https://github.com/webventures/aiagent/blob/main/CHANGELOG.md
+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.5.3
+signing_key:
+specification_version: 4
+summary: A Ruby SDK for interacting with AI Agents such as ChatGPT and Claude
+test_files: []