baml 0.1.0-aarch64-linux → 0.1.9-aarch64-linux

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ac1a499103bfe1463f22871d3d97348876d6b0500a4751684b8000c0b736809c
-  data.tar.gz: 9f1644242fc6448bcbba689411edcdea7eef4e126127136792dd0aa859eb80e5
+  metadata.gz: fa539a467be79364e6f2c0c84ac82e3058b1fc94185e5e45d0dd55adffc4ad89
+  data.tar.gz: ab456ce0b9cc38ce274a975c7864237bce6e2a3e2caac28c4eaea7956a6ab33f
 SHA512:
-  metadata.gz: c2f427540d45fdd09aa2418f010f5e3cb1f8f95cf530173134a043f57dbb1bd069cf8c111abbfb67c478cb1207d2f7c71fbf5bf7c88da5213952aaafa598faff
-  data.tar.gz: c7334b5aeccb57c995a529fd843557c1f6f81a20e34de25793a139ac693023e77f50d65abaed39757a59738497b8bda9c0f46b6c3cce10a95349a410dd9f8f33
+  metadata.gz: dff0778d274963081e565ba405137de3dbd76a962be45599a968248ff3e8c6ea34b276956e986e5a63f6e1dd8e5eea4324efa3b1c63cb22bb8416c6cd354b21f
+  data.tar.gz: 7ceb0c5d2d754df0e0892363b2f6f816f8fd4a6634828b9b85b50d23620f7b5b653680fc3abc32c87b3a2b4e4179bc2c6d431655ab303b99aaf3158d10112042

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

@@ -1,9 +1,53 @@
 begin
   ruby_version = /(\d+\.\d+)/.match(RUBY_VERSION)
-  require_relative "#{ruby_version}/baml"
+  require_relative "baml/#{ruby_version}/ruby_ffi"
 rescue LoadError
-  require_relative "baml"
+  require_relative "baml/ruby_ffi"
 end
+require_relative "stream"
 
 module Baml
-end
+  # 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)
+    if !Baml.const_defined?(:TypeConverter)
+      Baml.const_set(:TypeConverter, Class.new(TypeCoerce::Converter) do
+        def initialize(type)
+          super(type)
+        end
+        
+        def _convert(value, type, raise_coercion_error, coerce_empty_to_nil)
+          # make string handling more strict
+          if type == String
+            if value.is_a?(String)
+              return value
+            end
+
+            raise TypeCoerce::CoercionError.new(value, type)
+          end
+
+          # add unions
+          if type.is_a?(T::Types::Union)
+            type.types.each do |t|
+              # require raise_coercion_error on the recursive union call,
+              # so that we can suppress the error if it fails
+              converted = _convert(value, t, true, coerce_empty_to_nil)
+              return converted
+            rescue
+              # do nothing - try every instance of the union
+            end
+
+            raise TypeCoerce::CoercionError.new(value, type)
+          end
+
+          super(value, type, raise_coercion_error, coerce_empty_to_nil)
+        end
+      end)
+    end
+
+    Baml.const_get(:TypeConverter).new(type)
+  end
+end

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,25 +1,32 @@
 --- !ruby/object:Gem::Specification
 name: baml
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.9
 platform: aarch64-linux
 authors:
 - BoundaryML
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-04-24 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/baml.so
+- lib/baml/2.7/ruby_ffi.so
+- lib/baml/3.0/ruby_ffi.so
+- lib/baml/3.1/ruby_ffi.so
+- lib/baml/3.2/ruby_ffi.so
+- lib/baml/3.3/ruby_ffi.so
+- lib/stream.rb
 homepage: https://github.com/BoundaryML/baml
 licenses:
 - MIT
@@ -32,7 +39,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '3.3'
+      version: '2.7'
   - - "<"
     - !ruby/object:Gem::Version
       version: 3.4.dev