rcrewai 0.3.0 → 0.4.0

data/lib/rcrewai/task.rb

@@ -2,12 +2,14 @@
 
 require_relative 'async_executor'
 require_relative 'human_input'
+require_relative 'output_schema'
 
 module RCrewAI
   class Task
     include AsyncExtensions
     include HumanInteractionExtensions
-    attr_reader :name, :description, :agent, :context, :expected_output, :tools, :async
+    attr_reader :name, :description, :agent, :context, :expected_output, :tools, :async,
+                :raw_result, :structured_output
     attr_accessor :result, :status, :start_time, :end_time, :execution_time
 
     def initialize(name:, description:, agent: nil, **options)
@@ -20,6 +22,16 @@ module RCrewAI
       @async = options[:async] || false   # Whether task can run asynchronously
       @callback = options[:callback]      # Callback function after completion
 
+      # Output processing (0.4.0)
+      @output_schema = options[:output_schema]            # JSON-schema for structured output
+      @guardrail = options[:guardrail]                    # ->(output) { [ok, value_or_error] }
+      @guardrail_max_retries = options.fetch(:guardrail_max_retries, 3)
+      @output_file = options[:output_file]                # Path to write result to
+      @create_directory = options.fetch(:create_directory, true)
+      @markdown = options.fetch(:markdown, false)
+      @raw_result = nil                                   # Unprocessed string content
+      @structured_output = nil                            # Parsed object when output_schema set
+
       # Human interaction options
       @human_input_enabled = options[:human_input] || false
       @require_human_confirmation = options[:require_confirmation] || false
@@ -62,7 +74,7 @@ module RCrewAI
             )
           end
 
-          @result = agent.execute_task(self)
+          @result = run_agent_with_output_processing
 
           # Post-execution human review if configured
           if @human_input_enabled && @human_review_points.include?(:completion)
@@ -166,6 +178,12 @@ module RCrewAI
       @context << task unless @context.include?(task)
     end
 
+    # Appends supplementary guidance (e.g. a planning step) to the task's
+    # description without discarding the original instructions.
+    def enrich_description(text)
+      @description = "#{@description}\n\n#{text}"
+    end
+
     def add_tool(tool)
       @tools << tool unless @tools.include?(tool)
     end
@@ -190,6 +208,74 @@ module RCrewAI
 
     private
 
+    # Runs the agent, then applies guardrail validation and schema coercion.
+    # Guardrail/schema failures re-run the agent (up to @guardrail_max_retries)
+    # with the failure fed back in, rather than raising immediately.
+    def run_agent_with_output_processing
+      attempts = 0
+      feedback = nil
+
+      loop do
+        attempts += 1
+        raw = extract_content(agent.execute_task(self))
+        @raw_result = raw
+
+        begin
+          apply_guardrail!(raw)
+          @structured_output = OutputSchema.coerce(raw, @output_schema) if @output_schema
+        rescue OutputProcessingError, OutputSchemaError => e
+          if attempts <= @guardrail_max_retries
+            feedback = e.message
+            append_feedback_to_description(feedback)
+            next
+          end
+          raise
+        end
+
+        write_output_file(guardrail_value_or(raw)) if @output_file
+        return guardrail_value_or(raw)
+      end
+    end
+
+    # Accepts either the legacy plain-string return or the 0.3.0 result hash.
+    def extract_content(agent_result)
+      return agent_result[:content].to_s if agent_result.is_a?(Hash)
+
+      agent_result.to_s
+    end
+
+    def apply_guardrail!(raw)
+      return unless @guardrail
+
+      ok, value = @guardrail.call(raw)
+      raise OutputProcessingError, "guardrail rejected output: #{value}" unless ok
+
+      @guardrail_value = value
+    end
+
+    def guardrail_value_or(raw)
+      @guardrail ? @guardrail_value : raw
+    end
+
+    def append_feedback_to_description(feedback)
+      @description = "#{@description}\n\n[Retry] Previous attempt was rejected: #{feedback}. " \
+                     'Please correct the output.'
+    end
+
+    def write_output_file(content)
+      dir = File.dirname(@output_file)
+      if @create_directory
+        require 'fileutils'
+        FileUtils.mkdir_p(dir)
+      elsif !Dir.exist?(dir)
+        raise OutputProcessingError, "output directory does not exist: #{dir}"
+      end
+
+      body = content.to_s
+      body = "# #{name}\n\n#{body}" if @markdown && !body.lstrip.start_with?('#')
+      File.write(@output_file, body)
+    end
+
     def confirm_task_execution
       message = "Confirm execution of task: #{name}"
       context = "Description: #{description}\nExpected Output: #{expected_output || 'Not specified'}\nAssigned Agent: #{agent&.name || 'No agent'}"
@@ -365,4 +451,5 @@ module RCrewAI
 
   class TaskExecutionError < Error; end
   class TaskDependencyError < TaskExecutionError; end
+  class OutputProcessingError < TaskExecutionError; end
 end

data/lib/rcrewai/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RCrewAI
-  VERSION = '0.3.0'
+  VERSION = '0.4.0'
 end

data/lib/rcrewai.rb

@@ -23,6 +23,7 @@ require_relative 'rcrewai/sse_parser'
 require_relative 'rcrewai/pricing'
 require_relative 'rcrewai/llm_client'
 require_relative 'rcrewai/memory'
+require_relative 'rcrewai/knowledge'
 require_relative 'rcrewai/human_input'
 require_relative 'rcrewai/tool_schema'
 require_relative 'rcrewai/provider_schema'
@@ -41,4 +42,5 @@ require_relative 'rcrewai/async_executor'
 require_relative 'rcrewai/agent'
 require_relative 'rcrewai/task'
 require_relative 'rcrewai/crew'
+require_relative 'rcrewai/flow'
 require_relative 'rcrewai/mcp'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rcrewai
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.4.0
 platform: ruby
 authors:
 - gkosmo
@@ -302,6 +302,7 @@ files:
 - CHANGELOG.md
 - LICENSE
 - README.md
+- ROADMAP.md
 - Rakefile
 - bin/rcrewai
 - docs/_config.yml
@@ -361,7 +362,16 @@ files:
 - lib/rcrewai/configuration.rb
 - lib/rcrewai/crew.rb
 - lib/rcrewai/events.rb
+- lib/rcrewai/flow.rb
+- lib/rcrewai/flow/state.rb
+- lib/rcrewai/flow/state_store.rb
 - lib/rcrewai/human_input.rb
+- lib/rcrewai/knowledge.rb
+- lib/rcrewai/knowledge/base.rb
+- lib/rcrewai/knowledge/chunker.rb
+- lib/rcrewai/knowledge/embedder.rb
+- lib/rcrewai/knowledge/sources.rb
+- lib/rcrewai/knowledge/store.rb
 - lib/rcrewai/legacy_react_runner.rb
 - lib/rcrewai/llm_client.rb
 - lib/rcrewai/llm_clients/anthropic.rb
@@ -376,6 +386,8 @@ files:
 - lib/rcrewai/mcp/transport/http.rb
 - lib/rcrewai/mcp/transport/stdio.rb
 - lib/rcrewai/memory.rb
+- lib/rcrewai/output_schema.rb
+- lib/rcrewai/planning.rb
 - lib/rcrewai/pricing.rb
 - lib/rcrewai/process.rb
 - lib/rcrewai/provider_schema.rb