autoflux 0.4.0 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 85b6aa4923d77e37e9e4c0323b9d7b4ea211d1c826b832f439ffef41fc2838a4
-  data.tar.gz: 1f83f0a1a66cca3f03fdd0905625406e87fb5e43f9c8a8a467d4508d1fb2a450
+  metadata.gz: 2d8e62b48ee54fa55a2f7f86dfdbe3fee25458b0c68cefc6ad54b976414523fa
+  data.tar.gz: c5e255652c99c2b01dc3535219788110035b71bd71fdabcd5a97fdc52431165d
 SHA512:
-  metadata.gz: 5014bfa1497302ff5a839f2ed914ca483af355faf5c33f6f5f543e2e3a4729a09515c58e473b6ecb2e6e03dd345a7354371728475f2fbee5998d1db514b7a465
-  data.tar.gz: 95a4935d4ca50f9817412184781ce486b5273ba349be6441f5842f746b8f61543f849d5c3d23fb39375e02f265078e7596809d7599a4a6135d76c2b16725950b
+  metadata.gz: f176ff0cdd42e11b03f2137102482fd7de9401a1b389b896616f878c0e670c4e6dc74e81fc8126a21b5350468658cb47dc547785743f1004337d7b3949637b73
+  data.tar.gz: 7ec3e5dd178cb03687d396e9194c471a2c97bbee1e88ba97d3ed674410120f8d5b6418b35187f1c7fc1a1ca115738891e45c9af77df648b49fd2f7c6e767b393

data/README.md

@@ -49,31 +49,48 @@ When the `io` is `EOF`, the workflow will stop.
 
 ### Agent
 
-The agent is a interface implements `call` method to process the command.
+The agent is an object with have `#name` and `#call` methods.
 
 ```ruby
-agent = ->(prompt, **context) {
-  case prompt
-  when 'hello'
-    'Hello, how can I help you?'
-  when 'bye'
-    'Goodbye'
-  else
-    'I do not understand'
-  end
-}
+require 'autoflux/openai'
+
+agent = Autoflux::OpenAI.new(
+    name: "chat",
+    model: "gpt-4o-mini"
+)
 ```
 
 The workflow will pass itself as context to the agent.
 
 ```ruby
-agent = ->(prompt, workflow:, **) {
-    workflow.io.write("User: #{prompt}")
-}
+class MyAgent
+    attr_reader :name
+
+    def initialize(name:)
+      @name = name
+    end
+
+    def call(params, workflow:)
+      workflow.io.write("Hello, #{params[:name]}!")
+    end
+end
+
 ```
 
 Workflow never knows the how the agent works and which tool is used.
 
+The agent can switch by workflow if the workflow knows it. You can use it in the agent's tool to switch the agent.
+
+```ruby
+workflow = Autoflux::Workflow.new(
+    agent: agent1, # if not given the first agent in `agents` will be used
+    agents: [agent1, agent2],
+    io: io,
+)
+
+workflow.switch_agent("agent2")
+```
+
 ### IO
 
 The IO is an adapter to let the workflow interact with the user.

data/lib/autoflux/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Autoflux
-  VERSION = "0.4.0"
+  VERSION = "0.5.0"
 end

data/lib/autoflux/workflow.rb

@@ -18,14 +18,17 @@ module Autoflux
 
     include Enumerable
 
-    attr_reader :id, :agent, :io
+    attr_reader :id, :agent, :io, :agents
 
     # @rbs state: State
-    def initialize(agent:, io:, id: nil, step: Step::Start.new)
+    def initialize(io:, agent: nil, agents: [], id: nil, step: Step::Start.new)
       @id = id || Workflow.next_id
-      @agent = agent
+      @agent = agent || agents.first
+      @agents = agents
       @io = io
       @step = step
+
+      raise Error, "No agent provided" unless @agent
     end
 
     def each
@@ -44,6 +47,15 @@ module Autoflux
       each(&block || ->(_workflow) {})
     end
 
+    # Switch the agent.
+    def switch_agent(name)
+      new_agent = agents.find { |agent| agent.name == name }
+      return false unless new_agent
+
+      @agent = new_agent
+      true
+    end
+
     # Stop the workflow.
     def stop
       @step = Step::Stop.new

data/sig/autoflux/agent.rbs

@@ -1,6 +1,7 @@
 module Autoflux
   # Agent implements the _Agent interface.
   interface _Agent
+    def name: () -> String
     def call: (String? prompt, workflow: Workflow) -> _ToS
   end
 end

data/sig/autoflux/workflow.rbs

@@ -5,6 +5,7 @@ module Autoflux
 
     @id: String
     @agent: _Agent
+    @agents: Array[_Agent]
     @io: _IO
     @step: _Step
 
@@ -12,11 +13,13 @@ module Autoflux
 
     attr_reader id: String
     attr_reader agent: _Agent
+    attr_reader agents: Array[_Agent]
     attr_reader io: _IO
 
-    def initialize: (agent: _Agent, io: _IO, ?id: String?, ?step: _Step) -> void
+    def initialize: (io: _IO, ?agent: _Agent, ?agents: Array[_Agent], ?id: String?, ?step: _Step) -> void
     def each: { (Workflow) -> void } -> void
     def run: () ?{ (Workflow) -> void } -> void
+    def switch_agent: (String name) -> bool
     def stop: () -> void
     def step: () -> _Step
     def to_h: () -> { id: String, step: String }

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: autoflux
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.5.0
 platform: ruby
 authors:
 - Aotokitsuruya