Merge pull request #19 from alexrudall/0.3.0

Bump version

CHANGELOG.md

@@ -5,6 +5,12 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [0.3.0] - 2024-06-10
+
+### Added
+
+- Add chat streaming! Thank you to the inimitable [@swombat](https://github.com/swombat) for adding this vital functionality!
+
 ## [0.2.0] - 2024-04-25
 
 ### Added

Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    anthropic (0.2.0)
+    anthropic (0.3.0)
       event_stream_parser (>= 0.3.0, < 2.0.0)
       faraday (>= 1)
       faraday-multipart (>= 1)

lib/anthropic/http.rb

@@ -2,6 +2,7 @@
 
 require_relative "http_headers"
 
+# rubocop:disable Metrics/ModuleLength
 module Anthropic
   module HTTP
     include HTTPHeaders
@@ -14,6 +15,7 @@ def get(path:)
     end
 
     # This is currently the workhorse for all API calls.
+    # rubocop:disable Metrics/MethodLength
     def json_post(path:, parameters:)
       str_resp = {}
       response = conn.post(uri(path: path)) do |req|
@@ -30,6 +32,7 @@ def json_post(path:, parameters:)
 
       str_resp.empty? ? response.body : str_resp
     end
+    # rubocop:enable Metrics/MethodLength
 
     # Unused?
     def multipart_post(path:, parameters: nil)
@@ -64,6 +67,7 @@ def to_json(string)
     #
     # @param user_proc [Proc] The inner proc to call for each JSON object in the chunk.
     # @return [Proc] An outer proc that iterates over a raw stream, converting it to JSON.
+    # rubocop:disable Metrics/MethodLength
     def to_json_stream(user_proc:, response:, preprocess: nil)
       parser = EventStreamParser::Parser.new
       preprocess_stack = ""
@@ -82,7 +86,9 @@ def to_json_stream(user_proc:, response:, preprocess: nil)
         end
       end
     end
+    # rubocop:enable Metrics/MethodLength
 
+    # rubocop:disable Metrics/MethodLength
     def _handle_message_type(type, parsed_data, response, &block)
       case type
       when "message_start"
@@ -97,6 +103,7 @@ def _handle_message_type(type, parsed_data, response, &block)
         block.yield delta
       end
     end
+    # rubocop:enable Metrics/MethodLength
 
     # Decides whether to preprocess JSON or text and calls the appropriate method.
     def preprocess(directive, stack, delta, user_proc)
@@ -187,3 +194,4 @@ def try_parse_json(maybe_json)
     end
   end
 end
+# rubocop:enable Metrics/ModuleLength

lib/anthropic/version.rb

@@ -1,3 +1,3 @@
 module Anthropic
-  VERSION = "0.2.0".freeze
+  VERSION = "0.3.0".freeze
 end

spec/anthropic/client/messages_spec.rb

@@ -1,4 +1,6 @@
 RSpec.describe Anthropic::Client do
+  let(:json_prompt) { "Answer in the provided JSON format. Only include JSON." }
+
   describe "#messages" do
     context "with all required parameters (:model, :messages, :max_tokens)" do
       let(:model) { "claude-3-haiku-20240307" }
@@ -94,7 +96,7 @@
     context "streaming JSON" do
       let(:model) { "claude-3-haiku-20240307" }
       let(:messages) do
-        [{ role: "user", content: "Give me the heights of the 3 tallest mountains. Answer in the provided JSON format. Only include JSON." },
+        [{ role: "user", content: "Give me the height of the 3 tallest mountains. #{json_prompt}" },
          { role: "assistant",
            content: '[{"name": "Mountain Name", "height": "height in km"}]' }]
       end
@@ -131,7 +133,7 @@
     context "streaming preprocessed JSON" do
       let(:model) { "claude-3-haiku-20240307" }
       let(:messages) do
-        [{ role: "user", content: "Give me the heights of the 3 tallest mountains. Answer in the provided JSON format. Only include JSON." },
+        [{ role: "user", content: "Give me the height of the 3 tallest mountains. #{json_prompt}" },
          { role: "assistant",
            content: '[{"name": "Mountain Name", "height": "height in km"}]' }]
       end
@@ -170,7 +172,7 @@
     context "malformed streaming preprocessed JSON" do
       let(:model) { "claude-3-haiku-20240307" }
       let(:messages) do
-        [{ role: "user", content: "Give me the heights of the 3 tallest mountains. Answer in the provided JSON format. Only include JSON." },
+        [{ role: "user", content: "Give me the height of the 3 tallest mountains. #{json_prompt}" },
          { role: "assistant",
            content: '[{"name": "Mountain Name", "height": "height in km"}]' }]
       end