foobara-agent 0.0.1

data/src/foobara/agent/list_commands.rb

@@ -0,0 +1,39 @@
+module Foobara
+  class Agent
+    class ListCommands < Foobara::Command
+      inputs do
+        command_connector :duck, :required, "Connector to end"
+      end
+
+      result do
+        user_provided_commands [:string], :required,
+                               "List of commands the user would like you to choose from to accomplish the goal"
+        agent_specific_commands [:string], :required,
+                                "Commands available to all agents to introspect or end the session"
+      end
+
+      def execute
+        construct_command_lists
+
+        command_lists
+      end
+
+      attr_accessor :command_lists
+
+      def construct_command_lists
+        user_provided_commands = []
+        agent_specific_commands = []
+
+        command_connector.all_transformed_command_classes.each do |command_class|
+          if command_class.domain == Foobara::Agent
+            agent_specific_commands
+          else
+            user_provided_commands
+          end << command_class.full_command_name
+        end
+
+        self.command_lists = { user_provided_commands:, agent_specific_commands: }
+      end
+    end
+  end
+end

data/src/foobara/agent/list_types.rb

@@ -0,0 +1,23 @@
+module Foobara
+  class Agent
+    class ListTypes < Foobara::Command
+      inputs do
+        command_connector :duck, :required, "Connector to fetch types from"
+      end
+
+      result [:string]
+
+      def execute
+        construct_type_list
+
+        type_list
+      end
+
+      attr_accessor :type_list
+
+      def construct_type_list
+        self.type_list = command_connector.all_exposed_type_names
+      end
+    end
+  end
+end

data/src/foobara/agent/types/command_log_entry.rb

@@ -0,0 +1,16 @@
+module Foobara
+  class Agent
+    class CommandLogEntry < Foobara::Model
+      attributes do
+        command_name :string, :required, "Name of the command that was run"
+        inputs :duck, :required, "Inputs to the command" # TODO: Allow :attributes to be used as a type succesfully
+        outcome :required do
+          success :boolean, :required, "Whether the command succeeded or not"
+          result :duck, :allow_nil, "Result of the command"
+          # TODO: create a type for error hash structure
+          errors_hash :duck, :allow_nil, "Errors that occurred during the command"
+        end
+      end
+    end
+  end
+end

data/src/foobara/agent/types/context.rb

@@ -0,0 +1,11 @@
+module Foobara
+  class Agent
+    class Context < Foobara::Model
+      attributes do
+        # TODO: why doesn't this default of [] work as expected on newly created models?
+        command_log [CommandLogEntry], default: [],
+                                       description: "Log of all commands run so far and their outcomes"
+      end
+    end
+  end
+end

data/src/foobara/agent.rb

@@ -0,0 +1,61 @@
+require "io/wait"
+
+module Foobara
+  class Agent
+    attr_accessor :context, :agent_command_connector, :agent_name, :llm_model
+
+    def initialize(
+      context: nil,
+      agent_name: nil,
+      command_classes: nil,
+      agent_command_connector: nil,
+      llm_model: nil
+    )
+      # TODO: shouldn't have to pass command_log here since it has a default, debug that
+      self.context = context
+      self.agent_command_connector = agent_command_connector
+      self.agent_name = agent_name if agent_name
+      self.llm_model = llm_model
+
+      build_initial_context
+      build_agent_command_connector
+
+      command_classes&.each do |command_class|
+        self.agent_command_connector.connect(command_class)
+      end
+    end
+
+    def accomplish_goal(goal, result_type: nil)
+      inputs = {
+        goal:,
+        final_result_type: result_type,
+        current_context: context,
+        existing_command_connector: agent_command_connector,
+        agent_name:
+      }
+
+      if llm_model
+        inputs[:llm_model] = llm_model
+      end
+
+      AccomplishGoal.run(inputs)
+    end
+
+    def build_initial_context
+      # TODO: shouldn't have to pass command_log here since it has a default, debug that
+      self.context ||= Context.new(command_log: [])
+    end
+
+    def build_agent_command_connector
+      self.agent_command_connector ||= Connector.new(
+        accomplish_goal_command: self,
+        llm_model:,
+        default_serializers: [
+          Foobara::CommandConnectors::Serializers::ErrorsSerializer,
+          Foobara::CommandConnectors::Serializers::AtomicSerializer,
+          Foobara::CommandConnectors::Serializers::JsonSerializer
+        ]
+      )
+    end
+  end
+end

metadata

@@ -0,0 +1,91 @@
+--- !ruby/object:Gem::Specification
+name: foobara-agent
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Miles Georgi
+bindir: bin
+cert_chain: []
+date: 1980-01-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: foobara
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.0.126
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.0.126
+- !ruby/object:Gem::Dependency
+  name: foobara-llm-backed-command
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.0.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.0.1
+email:
+- azimux@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- CHANGELOG.md
+- LICENSE-MPL-2.0.txt
+- LICENSE.txt
+- README.md
+- lib/foobara/agent.rb
+- src/foobara/agent.rb
+- src/foobara/agent/accomplish_goal.rb
+- src/foobara/agent/connector/connector.rb
+- src/foobara/agent/connector/set_command_connector_inputs_transformer.rb
+- src/foobara/agent/describe_command.rb
+- src/foobara/agent/describe_type.rb
+- src/foobara/agent/determine_inputs_for_next_command.rb
+- src/foobara/agent/determine_next_command.rb
+- src/foobara/agent/end_session_because_goal_has_been_accomplished.rb
+- src/foobara/agent/give_up.rb
+- src/foobara/agent/list_commands.rb
+- src/foobara/agent/list_types.rb
+- src/foobara/agent/types/command_log_entry.rb
+- src/foobara/agent/types/context.rb
+homepage: https://github.com/foobara/agent
+licenses:
+- MPL-2.0
+metadata:
+  homepage_uri: https://github.com/foobara/agent
+  source_code_uri: https://github.com/foobara/agent
+  changelog_uri: https://github.com/foobara/agent/blob/main/CHANGELOG.md
+  rubygems_mfa_required: 'true'
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.4.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.6.7
+specification_version: 4
+summary: An agent that uses whatever Foobara commands you wish to accomplish goals
+  of your choosing!
+test_files: []