llamafile 0.1.9 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: eecffa6e74da830d2293c3d25d0b94e96d142aab138086c7d11f65d90d148c5b
-  data.tar.gz: fa0736d62ab201d4691c86e073d77e442d19a24ca7fc1437ba8767230f8905b5
+  metadata.gz: 8f65be502a0aaa502ab94785114d66f057e9af629cda650851a39c73d3b78c97
+  data.tar.gz: 376add4a170db17299c3e11b1f2cc20313433894c1264c1ee729827b240ec08d
 SHA512:
-  metadata.gz: '046846dcb6eae1b18503f64fab042bbe29a8e0de757c14055dc4932fd45f8792c1484307f4f7f00133790ff2ae607d37d23b5858a2b5e752ac1bd0ceca6fcddb'
-  data.tar.gz: 399a97efd2be9d9b7af4afaf2cd7447524b5f502cc6c75c9f182853d710ec38b3360063f91fa6b7b7d7711d43714ccbab12c0e965719895804567746b7a5b0e2
+  metadata.gz: 13ccc9565d30515cafe42c61ee8e32c270e89399d1189b0f834285f277f70e31351b8ff0fdcdcacec0b5f78f16960768115918a4f3ee60ad92a03e16fda94e74
+  data.tar.gz: 3ef5540c172c6bdb6b021b20bfa0cbba2420f70059903a9961eb83586aaf2ae91ed069101d6b92100de396313825cbf8f4a767a79ff0b6b547ac430bba90b7f1

data/lib/llamafile/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Llamafile
-  VERSION = "0.1.9"
+  VERSION = "0.2.1"
 end

data/lib/llamafile.rb

@@ -21,19 +21,40 @@ module Lf
 	end
 	@@C = Hash.new { |h,k| h[k] = C.new(k) }
 	class C
+		attr_accessor :character, :actor, :example, :format
+		attr_reader :output, :input
 		def initialize k
 			@id = k
 			@input = ""
 			@output = ""
+			@character = %[a helpful and honest personal assistant.]
+			@actor = %[an honest person.]
+			@format = %[Format your response as properly formatted markdown.]
+			@example = %[# this is a heading\nThis is some general text about the heading.\n1. list item one.\n2. list item two.\n3. list item three.]
+			@convo = []
+		end
+		def fact q, a
+			@convo << [ q, a ]
+		end
+		def hist
+			@convo
+		end
+		def convo
+			o = []; @convo[-5..-1].each { |e| o << %[User: #{e[0]}\nLlama: #{e[1]}] };
+			return o.join("\n\n")
+		end
+		def prompt
+			%[Llama is #{@character}\nUser is #{@actor}\n#{@format}\nUse this example to guide your response:\n#{@example}\n#{convo}]
 		end
 		def << i
-			@output = Lf.prompt(output: %[User: #{@input}\nLlama: #{@output}], input: i)
-			@input = i
-			return @output
+			chain i
 		end
 		def chain *p
 			[p].flatten.compact.each { |e|
-				@output = Lf.prompt(output: %[User: #{@input}\nLlama: #{@output}], input: e)
+				puts %[<-- #{e}]
+				@convo << [ e, Lf.prompt(output: prompt, input: e)]
+				@output = @convo[-1][1]
+				puts %[--> #{@output}]
 				@input = e
 			}
 			return @output
@@ -49,10 +70,8 @@ module Lf
 		@@C.delete(k)
 	end
 	def self.chain *p
-		s = ""
-		[p].flatten.compact.each { |e|
-			s = LLAMA << %[#{s}#{e}]
-		}
+		s = {}
+		[p].flatten.compact.each { |e| o = []; s.each_pair {|k,v| o << %[User: #{k}\nLlama: #{v}] }; s[e] = Lf.prompt(output: o.join("\n"), input: e); }
 		return s
 	end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: llamafile
 version: !ruby/object:Gem::Version
-  version: 0.1.9
+  version: 0.2.1
 platform: ruby
 authors:
 - Erik Olson