baml 0.1.7-arm64-darwin → 0.1.8-arm64-darwin

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d6fe61717296adb78f3096fb887c1d1cd50b1afc04231badea6e3f2a88120d01
-  data.tar.gz: 859ef8029df23e5d731e9c7b61f319e990e7a5555ff7f25a2c8b4aad7fe6be54
+  metadata.gz: c2d01cbd5ca6800a21ac72bb3f708ffb5b20ef86bd6e9102a5c9dcbf60874604
+  data.tar.gz: 2acb1f122149e41a10d95146c834ab34d264bb56a2028b8d50f8c7367b45bd0e
 SHA512:
-  metadata.gz: e32611dc7a4bc584b7b853f06392c2b8eb9a8c94308e430ad4a0339feb5f82a0dc2bebeee43db79a498475142a8d67c0746962ce00f1223b4c897c8a5dc6a54e
-  data.tar.gz: 8a5971a501ff6ac9ddd9b1f6b4f43f7f32ce3787f3a7b188c8bbab145817dd25a3f6866a32baf417ff52fd0aa518d594c2e9baddf1bd794b1522e5c1aa57e1ed
+  metadata.gz: a15a9e541d2d4de73737c72cc9ff443d8b07fee096dccf911337a547c2ce09b0b9658dc48893be76fff65ec1acfbd7ecc3d9a3698947f8b58b76c852a2722704
+  data.tar.gz: 67124711ac2be51312555d2593e9bcd87e48b1bb8dba1b6316c7e644bb7a82f51e85648b7bce9791b2e4eaac941989b8542717bca6986fa0e69a52948553e7f6

data/exe/baml-cli

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+require "baml"
+
+Baml::Ffi::invoke_runtime_cli(Process.argv0, ARGV)

data/lib/baml.rb

@@ -4,8 +4,12 @@ begin
 rescue LoadError
   require_relative "baml/ruby_ffi"
 end
+require_relative "stream"
 
 module Baml
+  # TODO: implement image support
+  class Image; end
+
   # Dynamically + idempotently define Baml::TypeConverter
   # NB: this does not respect raise_coercion_error = false
   def self.convert_to(type)

data/lib/stream.rb

@@ -0,0 +1,73 @@
+require "sorbet-runtime"
+
+module Baml
+  # class BamlStream
+  #   include Enumerable
+
+  #   def initialize(raw_stream:)
+  #     @raw_stream = raw_stream
+  #   end
+
+  #   def each(&block)
+  #     @raw_stream.each do |raw_msg|
+  #       yield Message.from(raw_msg)
+  #     end
+  #   end
+  # end
+
+  class BamlStream
+    extend T::Sig
+    extend T::Generic
+
+    include Enumerable
+
+    PartialType = type_member
+    FinalType = type_member
+
+    def initialize(
+      ffi_stream:,
+      partial_coerce:,
+      final_coerce:,
+      ctx_manager:
+    )
+      @ffi_stream = ffi_stream
+      @partial_coerce = partial_coerce
+      @final_coerce = final_coerce
+      @ctx_manager = ctx_manager
+
+      @final_response = nil
+    end
+
+    # Calls the given block once for each event in the stream, where event is a parsed
+    # partial response. Returns `self` to enable chaining `.get_final_response`.
+    #
+    # Must be called with a block.
+    #
+    # @yieldparam [PartialType] event the parsed partial response
+    # @return [BamlStream] self
+    def each(&block)
+      # Implementing this and include-ing Enumerable allows users to treat this as a Ruby
+      # collection: https://ruby-doc.org/3.1.6/Enumerable.html#module-Enumerable-label-Usage
+      if @final_response == nil
+        @final_response = @ffi_stream.done(@ctx_manager) do |event|
+          block.call @partial_coerce.new.from(event.parsed_using_types(Baml::PartialTypes))
+        end
+      end
+
+      self
+    end
+
+
+    # Gets the final response from the stream.
+    #
+    # @return [FinalType] the parsed final response
+    sig {returns(FinalType)}
+    def get_final_response
+      if @final_response == nil
+        @final_response = @ffi_stream.done(@ctx_manager)
+      end
+
+      @final_coerce.new.from(@final_response.parsed_using_types(Baml::Types))
+    end
+  end
+end

metadata

@@ -1,29 +1,32 @@
 --- !ruby/object:Gem::Specification
 name: baml
 version: !ruby/object:Gem::Version
-  version: 0.1.7
+  version: 0.1.8
 platform: arm64-darwin
 authors:
 - BoundaryML
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-04-30 00:00:00.000000000 Z
+date: 2024-06-07 00:00:00.000000000 Z
 dependencies: []
 description: A gem for users to interact with BoundaryML's Language Model clients
   (LLM) in Ruby.
 email:
 - contact@boundaryml.com
-executables: []
+executables:
+- baml-cli
 extensions: []
 extra_rdoc_files: []
 files:
+- exe/baml-cli
 - lib/baml.rb
 - lib/baml/2.7/ruby_ffi.bundle
 - lib/baml/3.0/ruby_ffi.bundle
 - lib/baml/3.1/ruby_ffi.bundle
 - lib/baml/3.2/ruby_ffi.bundle
 - lib/baml/3.3/ruby_ffi.bundle
+- lib/stream.rb
 homepage: https://github.com/BoundaryML/baml
 licenses:
 - MIT