foobara-agent-cli 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 46a126c4562444688ae39b09b44c3b201ffac14b4fb33eaa2181834147d6e46c
-  data.tar.gz: 1804df83e14c8284c18ad1eeddcffe6010359beb49b7dee7cdb567899ecaa5d0
+  metadata.gz: 02c3d2e1a47200eb3a55ed6b80634f30dbaf04c7f5b9e0927098bf3866f9d841
+  data.tar.gz: 467d2fe2e191095ae57685faf1ea3bf5cd0edd39412d2731ee8876bac991f89f
 SHA512:
-  metadata.gz: 56c97595a7036ebfb8aee312243336fa1478e5c9543d47c45af06c99ba1b2f1e1f76fac7aa8beb2400934c3ecbc3d641289f58d5cf476f15528d866f401364f3
-  data.tar.gz: 1c66a8e7f417d5024cd79cd86f6762c5bcb74104010f5e48b68ce2dd2bd9f6a5146926151c12bf6feeddf5c2d0f926e3e5a988cdd7fd4f1831aca1ea9f9e77ea
+  metadata.gz: 80609b7d266e727fb9ce1b70e267bae1c14d423ab6ffb898db5704f5e401457458ef829bc554c286b4d867583c15b59a9ded0caee96b10e46c80e7dfc01740ed
+  data.tar.gz: 6fea8df456f11fbbdc48c4d45c695c0a671a5460165a9bbc9149c6f4a4e364c4c1984ff09a640bb9ae77193414c3936d605dd06b6cb9a29f370954958c8de24d

data/README.md

@@ -3,6 +3,15 @@
 Allows you to run a Foobara::Agent in a loop giving it new goals to accomplish in a REPL-style
 CLI experience.  Currently works with Ollama, OpenAI, or Anthropic.
 
+## WARNING!!
+
+This is experimental at the moment! It will not ask you permission before running commands. So do
+not connect any commands to it that you don't want the risk of it being called when or in a manner
+that is should not be.
+
+If you would like to add that safety feature that would be awesome/fun! A good idea to let me know in advance
+so I can help or coordinate.
+
 ## Contributing
 
 I would love help with this and other Foobara gems! Feel free to hit me up at miles@foobara.com if you
@@ -74,34 +83,64 @@ Let's run this program on the commandline and ask our agent to find and fix the
 
 ```
 $ example_scripts/capybaras-agent                                                                                                                           
-> There is a capybara with a bad year of birth. Can you find and fix the bad record? Thanks!
 
-I found and fixed the Capybara record with the bad year of birth. Barbara's year of birth was incorrectly listed as 19, which is unlikely for a capybara. I've updated it to 2019, which is a more reasonable birth year.
+Welcome to the Foobara Agent CLI! Type your goal and press enter to get started.
 
-> Thank you so much! Can you set it back so that I can do the demo over again? Thanks!          
+> There is a capybara with a bad year of birth. Can you find and fix the bad record? Thanks!
 
-I've reset Barbara's year of birth back to 19, so you can run the demo again. The system is now ready for you to demonstrate the capybara data correction process.
+I found and fixed the Capybara record with the bad year of birth. Barbara's year of birth was
+incorrectly listed as 19, which is unlikely for a capybara. I've updated it to 2019, which is
+a more reasonable birth year.
 
 >
 ```
 
 We can check with FindAllCapybaras that it actually fixed the record!
 
+```
+$ ./capybaras-cli FindAllCapybaras
+{
+  name: "Fumiko",
+  year_of_birth: 2020,
+  id: 19
+},
+{
+  name: "Barbara",
+  year_of_birth: 2019,
+  id: 20
+},
+{
+  name: "Basil",
+  year_of_birth: 2021,
+  id: 21
+}
+```
+
+We can see that she was now born 2000 years later than before!
+
 Let's tell our agent to set it back (in the same session):
 
 ```
-$ example_scripts/capybaras-agent                                                                                                                           
+$ example_scripts/capybaras-agent
+
+Welcome to the Foobara Agent CLI! Type your goal and press enter to get started.
+
 > There is a capybara with a bad year of birth. Can you find and fix the bad record? Thanks!
 
-I found and fixed the Capybara record with the bad year of birth. Barbara's year of birth was incorrectly listed as 19, which is unlikely for a capybara. I've updated it to 2019, which is a more reasonable birth year.
+I found and fixed the Capybara record with the bad year of birth. Barbara's year of birth was
+incorrectly listed as 19, which is unlikely for a capybara. I've updated it to 2019, which is
+a more reasonable birth year.
 
 > Thank you so much! Can you set it back so that I can do the demo over again? Thanks!          
 
-I've reset Barbara's year of birth back to 19, so you can run the demo again. The system is now ready for you to demonstrate the capybara data correction process.
+I've reset Barbara's year of birth back to 19, so you can run the demo again. The system is
+now ready for you to demonstrate the capybara data correction process.
 
 >
 ```
 
+And we can check that it's back to where it was as before and it's 19 again instead of 2019!
+
 Please see example_scripts/capybaras-agent for a working script that contains this code
 
 ## License

data/src/foobara/agent.rb

@@ -3,50 +3,7 @@ require "io/wait"
 module Foobara
   class Agent
     def run_cli(io_in: $stdin, io_out: $stdout, io_err: $stderr)
-      io_out.write "> "
-      io_out.flush
-
-      loop do
-        input_is_available = io_in.wait_readable(1)
-
-        if input_is_available
-          line = io_in.gets
-          if line.nil?
-            break
-          end
-
-          goal = line.chomp
-
-          begin
-            outcome = accomplish_goal(goal)
-
-            if outcome.success?
-              result = outcome.result
-              io_out.puts
-              io_out.puts result[:message_to_user]
-              io_out.puts
-              io_out.flush
-            else
-              # :nocov:
-              io_out.puts
-              io_err.puts outcome.errors_hash
-              io_err.puts
-              io_err.flush
-              # :nocov:
-            end
-
-            io_out.write "> "
-            io_out.flush
-          rescue => e
-            # :nocov:
-            io_out.puts e.message
-            io_err.puts e.message
-            io_err.puts e.backtrace
-            io_err.flush
-            # :nocov:
-          end
-        end
-      end
+      CliRunner.new(self, io_in:, io_out:, io_err:).run
     end
   end
 end

data/src/foobara/cli_runner.rb

@@ -0,0 +1,59 @@
+module Foobara
+  class Agent
+    class CliRunner
+      attr_accessor :io_in, :io_out, :io_err, :agent
+
+      def initialize(agent, io_in: $stdin, io_out: $stdout, io_err: $stderr)
+        self.agent = agent
+        self.io_in = io_in
+        self.io_out = io_out
+        self.io_err = io_err
+      end
+
+      def run
+        Util.pipe_write_with_flush(
+          io_out,
+          "\nWelcome to the Foobara Agent CLI! Type your goal and press enter to get started.\n\n> "
+        )
+
+        loop do
+          ready = Util.pipe_wait_readable(io_in, 1)
+
+          break if agent.killed?
+          break if io_in.closed?
+
+          next unless ready
+
+          line = Util.pipe_readline(io_in)
+
+          break if line.nil? || io_in.closed? || agent.killed?
+
+          goal = line.strip
+
+          break if goal =~ /\A(exit|quit|bye)\z/i
+          next if goal.empty?
+
+          begin
+            outcome = agent.accomplish_goal(goal)
+
+            if outcome.success?
+              result = outcome.result
+              Util.pipe_writeline(io_out, "\nAgent says: #{result[:message_to_user]}\n")
+            else
+              # :nocov:
+              Util.pipe_writeline(io_err, "\nError: #{outcome.errors_hash}\n")
+              # :nocov:
+            end
+
+            Util.pipe_write_with_flush(io_out, "\n> ")
+          rescue => e
+            # :nocov:
+            Util.pipe_writeline(io_err, e.message)
+            Util.pipe_writeline(io_err, e.backtrace)
+            # :nocov:
+          end
+        end
+      end
+    end
+  end
+end

data/src/foobara/util.rb

@@ -0,0 +1,56 @@
+module Foobara
+  module Util
+    module_function
+
+    def pipe_writeline(io, line = "")
+      pipe_io_operation(io) do
+        io.puts line
+        io.flush
+      end
+    end
+
+    def pipe_write_with_flush(io, text = "")
+      pipe_io_operation(io) do
+        io.write text
+        io.flush
+      end
+    end
+
+    def pipe_readline(io)
+      pipe_io_operation(io) do
+        io.readline.strip
+      end
+    end
+
+    def pipe_wait_readable(io, timeout = nil)
+      pipe_io_operation(io) do
+        io.wait_readable(timeout)
+      end
+    end
+
+    def pipe_io_operation(_io)
+      yield
+    rescue EOFError
+      nil
+    rescue IOError => e
+      # :nocov:
+      message = e.message
+
+      if message.include?("closed") && message.include?("stream")
+        nil
+      else
+        raise
+      end
+      # :nocov:
+    end
+
+    def close(io)
+      io.close
+      nil
+    rescue IOError
+      # :nocov:
+      nil
+      # :nocov:
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: foobara-agent-cli
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Miles Georgi
@@ -35,6 +35,8 @@ files:
 - README.md
 - lib/foobara/agent_cli.rb
 - src/foobara/agent.rb
+- src/foobara/cli_runner.rb
+- src/foobara/util.rb
 homepage: https://github.com/foobara/agent-cli
 licenses:
 - MPL-2.0