superagi 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 477d9cd1d2f1dcdbdcc691730e62692e5ca732f555f7126d0814dc7e6b938f65
-  data.tar.gz: 38e0496835095a59ca7f825dc0125da3053f39c2bbbc5358690ad0da9f971082
+  metadata.gz: a210eeb7bb52e127cb2b9ab235b3a483eb779dbf16bed7802fadb5eae17bc0c5
+  data.tar.gz: b3287833b9493ccf4d34ff90e95e47d0e11623da94d568c956ebdd26cf1d92c1
 SHA512:
-  metadata.gz: ef9fa9c5e6ea03936a65dff236d338d3e7ffff9923149c86040175a0a1399b20a6e6545571765c386b21022cdb0c66baabbc42d3362a4011edc930e46279dad6
-  data.tar.gz: ad7df73c2df566c168c5e90eb5e3b2b1ff7a25e0b4bb681c7e8528d02d963d30087788a59c0ccca90aa5b68e8b4326f35081bfb4f91f14b2f47a66dbf3d21907
+  metadata.gz: a5e3d55aec92a571d6b88a28cc14b82beae6ddc5a44d10216e4e54d863f2141ca59aad221c6bdcc8aeea044c2acd84c1187acd1e25bd0b57ac183744252f83c8
+  data.tar.gz: 9cd0bab2daff3d82a80495c12fa2e2bb83b777cf7a23f76065715980486e1269c29baa48ba5b644fa906150fb18ae2c96a89dede4ac15a586ecba50fb7902c3b

data/CHANGELOG.md

@@ -5,6 +5,13 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [0.3.0] - 2023-10-30
+
+### Added
+
+- Add SuperAGI::Agent#pause and Agent#resume to pause and resume an agent run.
+- Add SuperAGI::Agent#resources to get the output of an agent run.
+
 ## [0.2.0] - 2023-10-28
 
 ### Added

data/Gemfile

@@ -5,7 +5,7 @@ gemspec
 
 gem "byebug", "~> 11.1.3"
 gem "dotenv", "~> 2.8.1"
-gem "rake", "~> 13.0"
+gem "rake", "~> 13.1"
 gem "rspec", "~> 3.12"
 gem "rubocop", "~> 1.50.2"
 gem "vcr", "~> 6.1.0"

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    superagi (0.2.0)
+    superagi (0.3.0)
       faraday (>= 1)
       faraday-multipart (>= 1)
 
@@ -32,7 +32,7 @@ GEM
       ast (~> 2.4.1)
     public_suffix (5.0.3)
     rainbow (3.1.1)
-    rake (13.0.6)
+    rake (13.1.0)
     regexp_parser (2.8.0)
     rexml (3.2.6)
     rspec (3.12.0)
@@ -75,7 +75,7 @@ PLATFORMS
 DEPENDENCIES
   byebug (~> 11.1.3)
   dotenv (~> 2.8.1)
-  rake (~> 13.0)
+  rake (~> 13.1)
   rspec (~> 3.12)
   rubocop (~> 1.50.2)
   superagi!

data/README.md

@@ -148,6 +148,29 @@ puts response
 # => {"run_id"=>29970}
 ```
 
+### Pause Agent
+
+To pause an agent:
+
+```ruby
+client.agent.run(id: 15312)
+response = client.agent.pause(id: 15312)
+puts response
+# => {"result"=>"success"}
+```
+
+### Resume Agent
+
+To resume an agent:
+
+```ruby
+client.agent.run(id: 15312)
+client.agent.pause(id: 15312)
+response = client.agent.resume(id: 15312)
+puts response
+# => {"result"=>"success"}
+```
+
 ### Agent Status
 
 To get the status of Agent runs:
@@ -158,6 +181,17 @@ puts response
 # => [{"run_id"=>29970,"status"=>"CREATED"}]
 ```
 
+### Agent Resources
+
+To get the resources output by Agent runs:
+
+```ruby
+  run_id = client.agent.run(id: 15312)["run_id"]
+  response = client.agent.resources(parameters: { run_ids: [run_id] })
+  puts response
+  # => {}
+```
+
 ## Development
 
 After checking out the repo, run `bin/setup` to install dependencies. You can run `bin/console` for an interactive prompt that will allow you to experiment.

data/lib/superagi/agent.rb

@@ -1,5 +1,7 @@
 module SuperAGI
   class Agent
+    include SuperAGI::AgentValidation
+
     def initialize(client:)
       @client = client
     end
@@ -10,7 +12,7 @@ module SuperAGI
     end
 
     def update(id:, parameters:)
-      parameters = DEFAULT_UPDATE_PARAMETERS.merge(parameters)
+      parameters = valid_parameters(method: :update, parameters: parameters)
       @client.json_put(path: "/agent/#{id}", parameters: parameters)
     end
 
@@ -18,76 +20,21 @@ module SuperAGI
       @client.json_post(path: "/agent/#{id}/run", parameters: {})
     end
 
-    # def pause(id:)
-    # end
-
-    # def resume(id:)
-    # end
-
-    def status(id:)
-      @client.json_post(path: "/agent/#{id}/run-status", parameters: {})
-    end
-
-    # def resources(id:)
-    # end
-
-    private
-
-    ARRAY_PARAMETERS = %w[
-      constraints
-      goal
-      tools
-    ].freeze
-
-    DEFAULT_CREATE_PARAMETERS = {
-      agent_workflow: "Goal Based Workflow",
-      model: "gpt-4"
-    }.freeze
-    REQUIRED_CREATE_PARAMETERS = (%w[
-      description
-      instruction
-      iteration_interval
-      max_iterations
-      name
-    ] + ARRAY_PARAMETERS + DEFAULT_CREATE_PARAMETERS.keys).freeze
-
-    # Update parameters need to always include any List types, even if they are empty,
-    # otherwise the API will return a NoneType error.
-    DEFAULT_UPDATE_PARAMETERS = {
-      constraints: [],
-      goal: [],
-      tools: []
-    }.freeze
-    REQUIRED_UPDATE_PARAMETERS = DEFAULT_UPDATE_PARAMETERS.keys.freeze
-
-    def valid_parameters(method:, parameters:)
-      parameters = default_parameters(method: method, parameters: parameters)
-      validate_presence(method: method, parameters: parameters)
-      validate_arrays(parameters: parameters)
-      parameters
+    def pause(id:)
+      @client.json_post(path: "/agent/#{id}/pause", parameters: {})
     end
 
-    def default_parameters(method:, parameters:)
-      case method
-      when :create then DEFAULT_CREATE_PARAMETERS.merge(parameters)
-      when :update then DEFAULT_UPDATE_PARAMETERS.merge(parameters)
-      end
+    def resume(id:)
+      @client.json_post(path: "/agent/#{id}/resume", parameters: {})
     end
 
-    def validate_presence(method:, parameters:)
-      required_parameters = case method
-                            when :create then REQUIRED_CREATE_PARAMETERS
-                            when :update then REQUIRED_UPDATE_PARAMETERS
-                            end
-      required_parameters.each do |key|
-        raise ArgumentError, "#{key} is required" unless parameters[key.to_sym]
-      end
+    def status(id:)
+      @client.json_post(path: "/agent/#{id}/run-status", parameters: {})
     end
 
-    def validate_arrays(parameters:)
-      ARRAY_PARAMETERS.each do |key|
-        raise ArgumentError, "#{key} must be an array" unless parameters[key.to_sym].is_a?(Array)
-      end
+    def resources(parameters:)
+      parameters = valid_parameters(method: :resources, parameters: parameters)
+      @client.json_post(path: "/agent/resources/output", parameters: parameters)
     end
   end
 end

data/lib/superagi/agent_validation.rb

@@ -0,0 +1,70 @@
+module SuperAGI
+  module AgentValidation
+    private
+
+    ARRAY_CREATE_PARAMETERS = %w[
+      constraints
+      goal
+      tools
+    ].freeze
+
+    DEFAULT_CREATE_PARAMETERS = {
+      agent_workflow: "Goal Based Workflow",
+      model: "gpt-4"
+    }.freeze
+    REQUIRED_CREATE_PARAMETERS = (%w[
+      description
+      instruction
+      iteration_interval
+      max_iterations
+      name
+    ] + ARRAY_CREATE_PARAMETERS + DEFAULT_CREATE_PARAMETERS.keys).freeze
+
+    # Update parameters need to always include any List types, even if they are empty,
+    # otherwise the API will return a NoneType error.
+    DEFAULT_UPDATE_PARAMETERS = {
+      constraints: [],
+      goal: [],
+      tools: []
+    }.freeze
+    REQUIRED_UPDATE_PARAMETERS = DEFAULT_UPDATE_PARAMETERS.keys.freeze
+
+    REQUIRED_RESOURCES_PARAMETERS = %w[run_ids].freeze
+
+    def valid_parameters(method:, parameters:)
+      parameters = default_parameters(method: method, parameters: parameters)
+      validate_presence(method: method, parameters: parameters)
+      validate_arrays(method: method, parameters: parameters)
+      parameters
+    end
+
+    def default_parameters(method:, parameters:)
+      case method
+      when :create then DEFAULT_CREATE_PARAMETERS.merge(parameters)
+      when :update then DEFAULT_UPDATE_PARAMETERS.merge(parameters)
+      else parameters
+      end
+    end
+
+    def validate_presence(method:, parameters:)
+      required_parameters = case method
+                            when :create then REQUIRED_CREATE_PARAMETERS
+                            when :update then REQUIRED_UPDATE_PARAMETERS
+                            when :resources then REQUIRED_RESOURCES_PARAMETERS
+                            end
+      required_parameters.each do |key|
+        raise ArgumentError, "#{key} is required" unless parameters[key.to_sym]
+      end
+    end
+
+    def validate_arrays(method:, parameters:)
+      array_parameters = case method
+                         when :create, :update then ARRAY_CREATE_PARAMETERS
+                         when :resources then REQUIRED_RESOURCES_PARAMETERS
+                         end
+      array_parameters.each do |key|
+        raise ArgumentError, "#{key} must be an array" unless parameters[key.to_sym].is_a?(Array)
+      end
+    end
+  end
+end

data/lib/superagi/version.rb

@@ -1,3 +1,3 @@
 module SuperAGI
-  VERSION = "0.2.0".freeze
+  VERSION = "0.3.0".freeze
 end

data/lib/superagi.rb

@@ -3,6 +3,7 @@ require "faraday/multipart"
 
 require_relative "superagi/http"
 require_relative "superagi/client"
+require_relative "superagi/agent_validation"
 require_relative "superagi/agent"
 
 module SuperAGI

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: superagi
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Alex
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2023-10-28 00:00:00.000000000 Z
+date: 2023-10-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday
@@ -69,6 +69,7 @@ files:
 - lib/ruby/superagi.rb
 - lib/superagi.rb
 - lib/superagi/agent.rb
+- lib/superagi/agent_validation.rb
 - lib/superagi/client.rb
 - lib/superagi/compatibility.rb
 - lib/superagi/http.rb