ad-agent_architecture 0.0.6 → 0.0.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: daa2d7eacfdb1c2650cf23106d3f001040d765d7aa3ccf44081a989361e9ad48
-  data.tar.gz: b3fe1ef5e9d26a809d13ddb6200d125a17549b6af88b5995ba05e35788fcb608
+  metadata.gz: 4f028f265371e8a636e96ebbd94eb147d157133faac830fb212e6d7863cff2e0
+  data.tar.gz: cb6231c112e8788b82f19325ad54716b03fed23905435716aa1592967b9f8f48
 SHA512:
-  metadata.gz: b09d423f903b09b704e958627b4a2cc4d1dbae248aee623e0750623e366f13247ab943043742702f8e0353ce4fe8a22ea7303f0f267a2ccce90c365239970947
-  data.tar.gz: be77db10d9e1fc55d9d18464714b57a8039b194a5c53ec963d7becdc743533ef60b06af6d18cf9ba13048446d09031d679fe239c3c5c3352870a8b3cbeac706f
+  metadata.gz: 873afb3fa43cfba38266711436db58c76b8d83202f7926c6be23be7799c059b8de94715a8f68df3ef71c20a519215e1173bcd8eb3ddfb8c0cffe70d958270fb4
+  data.tar.gz: 794d569604a03b43a85aeadcfd42f566c7bba8e746445d69d30cae7bf6362c2d1a2e48a2967d8018b3fe3ce0aa9d1b539787f7a73a7d908a12a6d2343b5429a4

data/.rubocop.yml

@@ -14,7 +14,7 @@ AllCops:
   Exclude:
     - ".builders/**/*"
     - "spec/samples/**/*"
-    - "lib/ad/agent_architecture/db/*"
+    - "lib/ad/agent_architecture/database/*"
 
 Metrics/BlockLength:
   Exclude:
@@ -42,6 +42,9 @@ Metrics/BlockLength:
 
 Metrics/MethodLength:
   Max: 25
+  Exclude:
+    - "lib/ad/agent_architecture/dsl/*"
+    - "lib/ad/agent_architecture/database/*"
 
 Layout/LineLength:
   Max: 200
@@ -87,6 +90,14 @@ Lint/AmbiguousBlockAssociation:
 
 Style/AccessorGrouping:
   Enabled: false
+Metrics/AbcSize:
+  Exclude:
+    - "lib/ad/agent_architecture/dsl/*"
+    - "lib/ad/agent_architecture/database/*"
+    - "lib/ad/agent_architecture/report/*"
+Metrics/NoExpectationExample:
+  Exclude:
+    - "spec/ad/agent_architecture/dsl/*"
 
 Layout/SpaceBeforeComma:
   Enabled: false
@@ -105,3 +116,7 @@ RSpec/SpecFilePathSuffix:
 RSpec/NamedSubject:
   Exclude:
     - "**/spec/**/*"
+
+RSpec/ExampleLength:
+  Exclude:
+    - "**/spec/**/*"

data/CHANGELOG.md

@@ -1,3 +1,17 @@
+## [0.0.7](https://github.com/appydave/ad-agent_architecture/compare/v0.0.6...v0.0.7) (2024-06-29)
+
+
+### Bug Fixes
+
+* add basic support for database persistence to DSL ([632c815](https://github.com/appydave/ad-agent_architecture/commit/632c815133b2cd31ab95dccb3c0b8564e25aa605))
+
+## [0.0.6](https://github.com/appydave/ad-agent_architecture/compare/v0.0.5...v0.0.6) (2024-06-28)
+
+
+### Bug Fixes
+
+* update requirements documentation ([e80d072](https://github.com/appydave/ad-agent_architecture/commit/e80d0728652c7a4fdeb014324f196e0044cb13af))
+
 ## [0.0.5](https://github.com/appydave/ad-agent_architecture/compare/v0.0.4...v0.0.5) (2024-06-28)
 
 

data/lib/ad/agent_architecture/{db → database}/models.rb

@@ -1,12 +1,12 @@
 # frozen_string_literal: true
 
-require 'sequel'
-
 module Ad
   module AgentArchitecture
     module Database
       # Workflow model represents a workflow entity in the database.
       class Workflow < Sequel::Model
+        set_primary_key :id
+
         one_to_many :sections, class: 'Ad::AgentArchitecture::Database::Section'
         one_to_many :attributes, class: 'Ad::AgentArchitecture::Database::Attribute'
         one_to_many :workflow_runs, class: 'Ad::AgentArchitecture::Database::WorkflowRun'

data/lib/ad/agent_architecture/database/save_workflow_graph.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+module Ad
+  module AgentArchitecture
+    module Database
+      # Save workflow graph to database
+      class SaveWorkflowGraph
+        def initialize(workflow_hash)
+          @workflow_hash = workflow_hash
+        end
+
+        def save
+          DB.transaction do
+            # Save workflow
+            workflow_record = Ad::AgentArchitecture::Database::Workflow.create(name: @workflow_hash[:name])
+
+            # Save attributes
+            attribute_records = @workflow_hash[:attributes].map do |_name, attr|
+              Ad::AgentArchitecture::Database::Attribute.create(
+                name: attr[:name], type: attr[:type], is_array: attr[:is_array], workflow: workflow_record
+              )
+            end
+            attribute_map = attribute_records.to_h { |ar| [ar.name.to_sym, ar] }
+
+            # Save sections and steps
+            @workflow_hash[:sections].each do |section|
+              section_record = Ad::AgentArchitecture::Database::Section.create(
+                name: section[:name], order: section[:order], workflow: workflow_record
+              )
+
+              section[:steps].each do |step|
+                step_record = Ad::AgentArchitecture::Database::Step.create(
+                  name: step[:name], order: step[:order], prompt: step[:prompt], section: section_record
+                )
+
+                step[:input_attributes].each do |attr_name|
+                  Ad::AgentArchitecture::Database::InputAttribute.create(
+                    step: step_record, attribute: attribute_map[attr_name.to_sym]
+                  )
+                end
+
+                step[:output_attributes].each do |attr_name|
+                  Ad::AgentArchitecture::Database::OutputAttribute.create(
+                    step: step_record, attribute: attribute_map[attr_name.to_sym]
+                  )
+                end
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ad/agent_architecture/database/sql_query.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+module Ad
+  module AgentArchitecture
+    module Database
+      class SQLQuery
+        SQL_DIR = File.expand_path('../sql', __dir__)
+
+        def self.query(sql_filename, params = {})
+          sql_path = File.join(SQL_DIR, "#{sql_filename}.sql")
+
+          # puts "SQL_DIR: #{SQL_DIR}"
+          # puts "SQL Path: #{sql_path}"
+          raise "SQL file not found: #{sql_path}" unless File.exist?(sql_path)
+
+          sql = File.read(sql_path)
+          DB.fetch(sql, params)
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,97 @@
+# frozen_string_literal: true
+
+module Ad
+  module AgentArchitecture
+    module Dsl
+      # DSL for defining agent workflows
+      class AgentWorkflowDsl
+        def self.create(name:, &block)
+          new(name).tap do |dsl|
+            dsl.instance_eval(&block) if block_given?
+            dsl.save
+          end
+        end
+
+        def initialize(name)
+          @workflow = { name: name, sections: [], attributes: {} }
+          @current_section_order = 1
+        end
+
+        def attributes(&block)
+          instance_eval(&block) if block_given?
+        end
+
+        def attribute(name, type:, is_array: false)
+          @workflow[:attributes][name] = { name: name, type: type, is_array: is_array }
+        end
+
+        def section(name:, &block)
+          @current_step_order = 1
+          @current_section = { name: name, order: @current_section_order, steps: [] }
+          @current_section_order += 1
+          instance_eval(&block) if block_given?
+          @workflow[:sections] << @current_section
+        end
+
+        def step(name:, &block)
+          @current_step = { name: name, order: @current_step_order, input_attributes: [], output_attributes: [] }
+          @current_step_order += 1
+          instance_eval(&block) if block_given?
+          @current_section[:steps] << @current_step
+        end
+
+        def input(attr_name, **_opts)
+          @current_step[:input_attributes] << attr_name
+        end
+
+        def output(attr_name, **_opts)
+          @current_step[:output_attributes] << attr_name
+        end
+
+        def prompt(prompt_text, **_opts)
+          @current_step[:prompt] = prompt_text
+        end
+
+        def save
+          DB.transaction do
+            # Save workflow
+            workflow_record = Ad::AgentArchitecture::Database::Workflow.create(name: @workflow[:name])
+
+            # Save attributes
+            attribute_records = @workflow[:attributes].map do |_name, attr|
+              Ad::AgentArchitecture::Database::Attribute.create(
+                name: attr[:name], type: attr[:type], is_array: attr[:is_array], workflow: workflow_record
+              )
+            end
+            attribute_map = attribute_records.to_h { |ar| [ar.name.to_sym, ar] }
+
+            # Save sections and steps
+            @workflow[:sections].each do |section|
+              section_record = Ad::AgentArchitecture::Database::Section.create(
+                name: section[:name], order: section[:order], workflow: workflow_record
+              )
+
+              section[:steps].each do |step|
+                step_record = Ad::AgentArchitecture::Database::Step.create(
+                  name: step[:name], order: step[:order], prompt: step[:prompt], section: section_record
+                )
+
+                step[:input_attributes].each do |attr_name|
+                  Ad::AgentArchitecture::Database::InputAttribute.create(
+                    step: step_record, attribute: attribute_map[attr_name.to_sym]
+                  )
+                end
+
+                step[:output_attributes].each do |attr_name|
+                  Ad::AgentArchitecture::Database::OutputAttribute.create(
+                    step: step_record, attribute: attribute_map[attr_name.to_sym]
+                  )
+                end
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+require 'k_log'
+
+module Ad
+  module AgentArchitecture
+    module Report
+      # Print workflow details
+      class WorkflowDetailReport
+        include KLog::Logging
+
+        def print(workflow)
+          log.section_heading 'Workflow Details Report'
+          log.kv 'Name', workflow.name
+          log.kv 'Description', workflow.description
+
+          workflow.sections.each do |section|
+            log.section_heading "Section: #{section.name}"
+            log.kv 'Order', section.order
+            section.steps.each do |step|
+              log.section_heading "Step: #{step.name}"
+              log.kv 'Order', step.order
+              log.kv 'Prompt', step.prompt
+              log.kv 'Input Attributes', step.input_attributes.map { |ia| ia.attribute.name }.join(', ')
+              log.kv 'Output Attributes', step.output_attributes.map { |oa| oa.attribute.name }.join(', ')
+            end
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+require 'k_log'
+
+module Ad
+  module AgentArchitecture
+    module Report
+      # Print workflow details
+      class WorkflowListReport
+        include KLog::Logging
+
+        def print
+          # tp query, :workflow_name, :workflow_description, :section_name, :section_description, :section_order, :step_name, :step_order, :step_prompt, :inputs, :outputs
+          tp query, :workflow_name, :workflow_description, :section_name, :step_name, :step_prompt, :inputs, :outputs
+        end
+
+        def query
+          Ad::AgentArchitecture::Database::SQLQuery.query(:workflow_details).map do |row|
+            row[:inputs] = JSON.parse(row[:inputs]) if row[:inputs]
+            row[:outputs] = JSON.parse(row[:outputs]) if row[:outputs]
+            row
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ad/agent_architecture/sql/workflow_details.sql

@@ -0,0 +1,35 @@
+SELECT
+  workflows.id AS workflow_id,
+  workflows.name AS workflow_name,
+  workflows.description AS workflow_description,
+  sections.id AS section_id,
+  sections.name AS section_name,
+  sections.description AS section_description,
+  sections."order" AS section_order,
+  steps.id AS step_id,
+  steps.name AS step_name,
+  steps."order" AS step_order,
+  steps.prompt AS step_prompt,
+  json_group_array(
+    json_object(
+      'name', input_attr.name,
+      'type', input_attr.type,
+      'is_array', input_attr.is_array
+    )
+  ) AS inputs,
+  json_group_array(
+    json_object(
+      'name', output_attr.name,
+      'type', output_attr.type,
+      'is_array', output_attr.is_array
+    )
+  ) AS outputs
+FROM workflows
+LEFT JOIN sections ON workflows.id = sections.workflow_id
+LEFT JOIN steps ON sections.id = steps.section_id
+LEFT JOIN input_attributes ON steps.id = input_attributes.step_id
+LEFT JOIN attributes AS input_attr ON input_attributes.attribute_id = input_attr.id
+LEFT JOIN output_attributes ON steps.id = output_attributes.step_id
+LEFT JOIN attributes AS output_attr ON output_attributes.attribute_id = output_attr.id
+GROUP BY workflows.id, sections.id, steps.id
+ORDER BY workflows.name, sections."order", steps."order";

data/lib/ad/agent_architecture/version.rb

@@ -2,6 +2,6 @@
 
 module Ad
   module AgentArchitecture
-    VERSION = '0.0.6'
+    VERSION = '0.0.8'
   end
 end

data/lib/ad/agent_architecture.rb

@@ -2,15 +2,21 @@
 
 require 'sequel'
 require 'sqlite3'
+require 'k_log'
 
 require 'ad/agent_architecture/version'
-require 'ad/agent_architecture/db/create_schema'
+require 'ad/agent_architecture/database/create_schema'
 
 DB = Sequel.sqlite
 
 Ad::AgentArchitecture::Database::CreateSchema.new(DB).execute
 
-require 'ad/agent_architecture/db/models'
+require 'ad/agent_architecture/database/models'
+require 'ad/agent_architecture/database/sql_query'
+require 'ad/agent_architecture/database/save_workflow_graph'
+require 'ad/agent_architecture/dsl/agent_workflow_dsl'
+require 'ad/agent_architecture/report/workflow_detail_report'
+require 'ad/agent_architecture/report/workflow_list_report'
 
 module Ad
   module AgentArchitecture

data/package-lock.json

@@ -1,12 +1,12 @@
 {
   "name": "ad-agent_architecture",
-  "version": "0.0.6",
+  "version": "0.0.8",
   "lockfileVersion": 3,
   "requires": true,
   "packages": {
     "": {
       "name": "ad-agent_architecture",
-      "version": "0.0.6",
+      "version": "0.0.8",
       "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.6",
+  "version": "0.0.8",
   "description": "Architecture/Schema for AI Agents",
   "scripts": {
     "release": "semantic-release"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ad-agent_architecture
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.0.8
 platform: ruby
 authors:
 - David Cruwys
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-06-28 00:00:00.000000000 Z
+date: 2024-06-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: k_log
@@ -80,8 +80,14 @@ files:
 - docs/requirements.md
 - docs/structure-youtube-script.yaml
 - lib/ad/agent_architecture.rb
-- lib/ad/agent_architecture/db/create_schema.rb
-- lib/ad/agent_architecture/db/models.rb
+- lib/ad/agent_architecture/database/create_schema.rb
+- lib/ad/agent_architecture/database/models.rb
+- lib/ad/agent_architecture/database/save_workflow_graph.rb
+- lib/ad/agent_architecture/database/sql_query.rb
+- lib/ad/agent_architecture/dsl/agent_workflow_dsl.rb
+- lib/ad/agent_architecture/report/workflow_detail_report.rb
+- lib/ad/agent_architecture/report/workflow_list_report.rb
+- lib/ad/agent_architecture/sql/workflow_details.sql
 - lib/ad/agent_architecture/version.rb
 - package-lock.json
 - package.json