foobara-anthropic-api 0.0.2

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 460564625ca99478e4533807393b31319cdbb1fbd88ec3bbf3c2fdad2553ade1
+  data.tar.gz: 539d6b7ec69c3f4b8ee858e327927980b112f79b46fb9f841d1415b73a993dc7
+SHA512:
+  metadata.gz: 50c3d406b8b139e2447051c549bc657defc1a85d13d71faaaffa0ec23443af91d2925fcea58c93950a88679a77108aa8e29ab2154f16e280f79d2a6d3e0185e9
+  data.tar.gz: b8a0972cbd2eac9f490462b119cb7ea316306876c8ebf7e7d65b705af2ebbed0649aabb90bfdac7625707dd66cdbeef328b5940a25799c2acfb3d5ee31f64a87

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2024 Miles Georgi
+
+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/lib/foobara/anthropic_api.rb

@@ -0,0 +1,5 @@
+require "foobara/all"
+
+require_relative "../../src/foobara/ai/anthropic_api"
+
+Foobara::Util.require_directory "#{__dir__}/../../src"

data/src/foobara/ai/anthropic_api/create_message.rb

@@ -0,0 +1,71 @@
+module Foobara
+  module Ai
+    module AnthropicApi
+      class CreateMessage < Foobara::Command
+        URL = "https://api.anthropic.com/v1/messages".freeze
+
+        inputs do
+          # TODO: come up with some kind of execution context object to wrap all of these calls upon which
+          # data wrapping a "request" can live instead of being passed around?
+          api_token :string
+          system :string
+          max_tokens :integer, default: 1024
+          model :model, default: Types::ModelEnum::CLAUDE_3_5_SONNET
+          messages [Types::Message]
+        end
+
+        result Types::MessageResult
+
+        def execute
+          build_request_body
+          build_request_headers
+          issue_http_request
+          parse_response
+          build_message_result
+
+          message_result
+        end
+
+        attr_accessor :request_body, :request_headers, :response, :response_body, :message_result
+
+        def build_request_body
+          self.request_body = { model:, max_tokens:, messages: }
+        end
+
+        def build_request_headers
+          self.request_headers = {
+            "Content-Type" => "application/json",
+            "X-Api-Key" => api_token,
+            "Anthropic-Version" => "2023-06-01"
+          }
+        end
+
+        def api_token
+          inputs[:api_token] || ENV.fetch("ANTHROPIC_API_KEY", nil)
+        end
+
+        def issue_http_request
+          url = URI.parse(URL)
+          self.response = Net::HTTP.post(url, JSON.generate(request_body), request_headers)
+        end
+
+        def parse_response
+          json = if response.is_a?(Net::HTTPSuccess)
+                   response.body
+                 else
+                   # :nocov:
+                   raise "Could not get message result from #{URL}: " \
+                         "#{response.code} #{response.message}"
+                   # :nocov:
+                 end
+
+          self.response_body = JSON.parse(json)
+        end
+
+        def build_message_result
+          self.message_result = Types::MessageResult.new(response_body)
+        end
+      end
+    end
+  end
+end

data/src/foobara/ai/anthropic_api/types/message.rb

@@ -0,0 +1,14 @@
+module Foobara
+  module Ai
+    module AnthropicApi
+      module Types
+        class Message < Foobara::Model
+          attributes do
+            role :string, one_of: %w[assistant user]
+            content :string
+          end
+        end
+      end
+    end
+  end
+end

data/src/foobara/ai/anthropic_api/types/message_result/content.rb

@@ -0,0 +1,16 @@
+module Foobara
+  module Ai
+    module AnthropicApi
+      module Types
+        class MessageResult < Foobara::Model
+          class Content < Foobara::Model
+            attributes do
+              text :string, :required
+              type :string, :required, one_of: %w[text]
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/src/foobara/ai/anthropic_api/types/message_result/usage.rb

@@ -0,0 +1,16 @@
+module Foobara
+  module Ai
+    module AnthropicApi
+      module Types
+        class MessageResult < Foobara::Model
+          class Usage < Foobara::Model
+            attributes do
+              input_tokens :integer, :required
+              output_tokens :integer, :required
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/src/foobara/ai/anthropic_api/types/message_result.rb

@@ -0,0 +1,23 @@
+require_relative "model"
+require_relative "role"
+
+module Foobara
+  module Ai
+    module AnthropicApi
+      module Types
+        class MessageResult < Foobara::Model
+          attributes do
+            content [Content]
+            id :string, :required
+            model :model, :required
+            role :role, :required
+            stop_reason :string, :required, one_of: %w[end_turn max_tokens stop_sequence]
+            stop_sequence :string, :allow_nil
+            type :string, :required, one_of: %w[message]
+            usage Usage
+          end
+        end
+      end
+    end
+  end
+end

data/src/foobara/ai/anthropic_api/types/model.rb

@@ -0,0 +1,16 @@
+module Foobara
+  module Ai
+    module AnthropicApi
+      module Types
+        module ModelEnum
+          CLAUDE_3_5_SONNET = "claude-3-5-sonnet-20240620".freeze
+          CLAUDE_3_OPUS = "claude-3-opus-20240229".freeze
+          CLAUDE_3_SONNET = "claude-3-sonnet-20240229".freeze
+          CLAUDE_3_HAIKU = "claude-3-haiku-20240307".freeze
+        end
+
+        Model = AnthropicApi.foobara_register_type(:model, :string, one_of: ModelEnum)
+      end
+    end
+  end
+end

data/src/foobara/ai/anthropic_api/types/role.rb

@@ -0,0 +1,14 @@
+module Foobara
+  module Ai
+    module AnthropicApi
+      module Types
+        module RoleEnum
+          ASSISTANT = :assistant
+          USER = :user
+        end
+      end
+
+      Types::Role = foobara_register_type(:role, :symbol, one_of: Types::RoleEnum)
+    end
+  end
+end

data/src/foobara/ai/anthropic_api.rb

@@ -0,0 +1,12 @@
+require "net/http"
+require "uri"
+
+module Foobara
+  module Ai
+    foobara_organization!
+
+    module AnthropicApi
+      foobara_domain!
+    end
+  end
+end

metadata

@@ -0,0 +1,71 @@
+--- !ruby/object:Gem::Specification
+name: foobara-anthropic-api
+version: !ruby/object:Gem::Version
+  version: 0.0.2
+platform: ruby
+authors:
+- Miles Georgi
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2024-08-20 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: foobara
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description:
+email:
+- azimux@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- LICENSE.txt
+- lib/foobara/anthropic_api.rb
+- src/foobara/ai/anthropic_api.rb
+- src/foobara/ai/anthropic_api/create_message.rb
+- src/foobara/ai/anthropic_api/types/message.rb
+- src/foobara/ai/anthropic_api/types/message_result.rb
+- src/foobara/ai/anthropic_api/types/message_result/content.rb
+- src/foobara/ai/anthropic_api/types/message_result/usage.rb
+- src/foobara/ai/anthropic_api/types/model.rb
+- src/foobara/ai/anthropic_api/types/role.rb
+homepage: https://github.com/foobara/anthropic-api
+licenses:
+- None specified yet
+metadata:
+  homepage_uri: https://github.com/foobara/anthropic-api
+  source_code_uri: https://github.com/foobara/anthropic-api
+  changelog_uri: https://github.com/foobara/anthropic-api/blob/main/CHANGELOG.md
+  rubygems_mfa_required: 'true'
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.2.2
+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: No description. Add one.
+test_files: []