Add agent and gate (#2)

* Add Agent and Gate

Author: andresg4

lib/mars/agent.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+module Mars
+  class Agent < Runnable
+    def initialize(name:, options:, tools: [], schema: nil)
+      @name = name
+      @tools = Array(tools)
+      @schema = schema
+      @options = options
+    end
+
+    def run(input)
+      chat.ask(input)
+    end
+
+    private
+
+    attr_reader :name, :tools, :schema, :options
+
+    def chat
+      @chat ||= RubyLLM::Chat.new(**options).with_tools(tools).with_schema(schema)
+    end
+  end
+end

lib/mars/gate.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+module Mars
+  class Gate < Runnable
+    def initialize(name:, condition:, branches:)
+      @name = name
+      @condition = condition
+      @branches = branches
+    end
+
+    def run(input)
+      result = condition.call(input)
+
+      raise "Invalid condition result: #{result}" unless branches.key?(result)
+
+      branches[result].run(input)
+    end
+
+    private
+
+    attr_reader :name, :condition, :branches
+  end
+end

mars.gemspec

@@ -34,7 +34,7 @@ Gem::Specification.new do |spec|
   spec.require_paths = ["lib"]
 
   # Uncomment to register a new dependency of your gem
-  # spec.add_dependency "example-gem", "~> 1.0"
+  spec.add_dependency "ruby_llm", "~> 1.0"
 
   # For more information and examples about making a new gem, check out our
   # guide at: https://bundler.io/guides/creating_gem.html