ad-agent_architecture 0.0.22 → 0.0.23

data/lib/ad/agent_architecture/dsl/workflow_dsl.rb

@@ -9,11 +9,16 @@ module Ad
 
         attr_reader :data
 
-        def initialize(name)
-          @data = { name: name, sections: [], attributes: {}, prompts: {}, settings: {} }
+        def initialize(name, description: nil)
+          @data = { name: name, description: description, sections: [], attributes: {}, prompts: {}, settings: {} }
           @current_section_order = 1
         end
 
+        def description(description)
+          @data[:description] = description
+          self
+        end
+
         def settings(&block)
           dsl = SettingsDsl.new(self)
           dsl.instance_eval(&block) if block_given?
@@ -32,8 +37,8 @@ module Ad
           dsl
         end
 
-        def section(name, &block)
-          dsl = SectionDsl.new(self, name, @current_section_order)
+        def section(name, description: nil, &block)
+          dsl = SectionDsl.new(self, name, @current_section_order, description: description)
           @current_section_order += 1
           dsl.instance_eval(&block) if block_given?
           dsl

data/lib/ad/agent_architecture/report/agent_data_builder.rb

@@ -0,0 +1,132 @@
+# frozen_string_literal: true
+
+module Ad
+  module AgentArchitecture
+    module Report
+      # This class is responsible for building the agent data for export or visualization
+      class AgentDataBuilder
+        attr_reader :workflow_id, :data
+
+        def initialize(workflow_id)
+          @workflow_id = workflow_id
+          @data = {}
+        end
+
+        def build
+          fetch_workflow
+          fetch_sections
+          fetch_attributes
+          fetch_prompts
+          data
+        end
+
+        private
+
+        def fetch_workflow
+          workflow = Ad::AgentArchitecture::Database::Workflow.where(id: workflow_id).first
+
+          @data[:name] = workflow.name
+          @data[:title] = Cmdlet::Case::Title.new.call(workflow.name)
+          @data[:description] = workflow.description
+          @data[:settings] = fetch_settings(workflow)
+        end
+
+        def fetch_sections
+          @data[:sections] = []
+          sections = Ad::AgentArchitecture::Database::Section.where(workflow_id: workflow_id).order(:order)
+          sections.each do |section|
+            section_data = {
+              name: section.name,
+              title: Cmdlet::Case::Title.new.call(section.name),
+              order: section.order,
+              description: section.description,
+              steps: fetch_steps(section.id)
+            }
+            @data[:sections] << section_data
+          end
+        end
+
+        def fetch_steps(section_id)
+          steps = Ad::AgentArchitecture::Database::Step.where(section_id: section_id).order(:order)
+          steps.map do |step|
+            {
+              name: step.name,
+              title: Cmdlet::Case::Title.new.call(step.name),
+              order: step.order,
+              description: step.description,
+              prompt: step.prompt,
+              input_attributes: fetch_input_attributes(step.id),
+              output_attributes: fetch_output_attributes(step.id)
+            }
+          end
+        end
+
+        def fetch_input_attributes(step_id)
+          Ad::AgentArchitecture::Database::InputAttribute.where(step_id: step_id).map do |input|
+            attribute = input.attribute
+            {
+              name: attribute.name,
+              type: attribute.type,
+              is_array: attribute.is_array,
+              description: attribute.description,
+              title: Cmdlet::Case::Title.new.call(attribute.name)
+            }
+          end
+        end
+
+        def fetch_output_attributes(step_id)
+          Ad::AgentArchitecture::Database::OutputAttribute.where(step_id: step_id).map do |output|
+            attribute = output.attribute
+            {
+              name: attribute.name,
+              type: attribute.type,
+              is_array: attribute.is_array,
+              description: attribute.description,
+              title: Cmdlet::Case::Title.new.call(attribute.name)
+            }
+          end
+        end
+
+        def fetch_attributes
+          @data[:attributes] = {}
+          attributes = Ad::AgentArchitecture::Database::Attribute.where(workflow_id: workflow_id)
+          attributes.each do |attr|
+            @data[:attributes][attr.name.to_sym] = {
+              name: attr.name,
+              type: attr.type,
+              is_array: attr.is_array,
+              description: attr.description,
+              title: Cmdlet::Case::Title.new.call(attr.name)
+            }
+          end
+        end
+
+        def fetch_prompts
+          @data[:prompts] = {}
+          prompts = Ad::AgentArchitecture::Database::Prompt.where(workflow_id: workflow_id)
+          prompts.each do |prompt|
+            @data[:prompts][prompt.name.to_sym] = {
+              name: prompt.name,
+              content: prompt.content,
+              description: prompt.description,
+              title: Cmdlet::Case::Title.new.call(prompt.name)
+            }
+          end
+        end
+
+        def fetch_settings(_workflow)
+          result = {}
+          settings = Ad::AgentArchitecture::Database::Setting.where(workflow_id: workflow_id)
+          settings.each do |setting|
+            result[setting.name.to_sym] = {
+              value: setting.value,
+              description: setting.description,
+              title: Cmdlet::Case::Title.new.call(setting.name)
+            }
+          end
+          result
+        end
+      end
+    end
+  end
+end

data/lib/ad/agent_architecture/version.rb

@@ -2,6 +2,6 @@
 
 module Ad
   module AgentArchitecture
-    VERSION = '0.0.22'
+    VERSION = '0.0.23'
   end
 end

data/lib/ad/agent_architecture.rb

@@ -3,6 +3,7 @@
 require 'sequel'
 require 'sqlite3'
 require 'k_log'
+require 'cmdlet'
 
 require 'ad/agent_architecture/version'
 require 'ad/agent_architecture/database/create_schema'
@@ -26,6 +27,7 @@ require 'ad/agent_architecture/dsl/actions/save_database'
 require 'ad/agent_architecture/dsl/actions/save_json'
 require 'ad/agent_architecture/dsl/actions/save_yaml'
 require 'ad/agent_architecture/report/dsl_generator'
+require 'ad/agent_architecture/report/agent_data_builder'
 require 'ad/agent_architecture/report/workflow_detail_report'
 require 'ad/agent_architecture/report/workflow_list_report'
 

data/package-lock.json

@@ -1,12 +1,12 @@
 {
   "name": "ad-agent_architecture",
-  "version": "0.0.22",
+  "version": "0.0.23",
   "lockfileVersion": 3,
   "requires": true,
   "packages": {
     "": {
       "name": "ad-agent_architecture",
-      "version": "0.0.22",
+      "version": "0.0.23",
       "devDependencies": {
         "@klueless-js/semantic-release-rubygem": "github:klueless-js/semantic-release-rubygem",
         "@semantic-release/changelog": "^6.0.3",

data/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ad-agent_architecture",
-  "version": "0.0.22",
+  "version": "0.0.23",
   "description": "Architecture/Schema for AI Agents",
   "scripts": {
     "release": "semantic-release"

metadata

@@ -1,15 +1,29 @@
 --- !ruby/object:Gem::Specification
 name: ad-agent_architecture
 version: !ruby/object:Gem::Version
-  version: 0.0.22
+  version: 0.0.23
 platform: ruby
 authors:
 - David Cruwys
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-07-04 00:00:00.000000000 Z
+date: 2024-07-06 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: cmdlet
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: k_log
   requirement: !ruby/object:Gem::Requirement
@@ -74,6 +88,8 @@ files:
 - LICENSE.txt
 - README.md
 - Rakefile
+- a1.json
+- a2.json
 - bin/console
 - bin/setup
 - bin/watch.rb
@@ -102,6 +118,7 @@ files:
 - lib/ad/agent_architecture/dsl/settings_dsl.rb
 - lib/ad/agent_architecture/dsl/step_dsl.rb
 - lib/ad/agent_architecture/dsl/workflow_dsl.rb
+- lib/ad/agent_architecture/report/agent_data_builder.rb
 - lib/ad/agent_architecture/report/dsl_generator.rb
 - lib/ad/agent_architecture/report/workflow_detail_report.rb
 - lib/ad/agent_architecture/report/workflow_list_report.rb