aigency 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c413cf37072efae1aa6df896884d291ee9bac175a1154e579d85833692f55959
-  data.tar.gz: feaacbce278ae242f36bd8f2b3024d014bb200dad97679f6fb023986def493ec
+  metadata.gz: 7b9b5b24d49d87f69a17cec9420b4815db4646b37d8759951e4ced8431765800
+  data.tar.gz: 04ecc1ddb1433cfcd2c18b7c3af420df7dd52a417ef6ab90a2a3de35adb2b99b
 SHA512:
-  metadata.gz: '09ba427def6dcce5abcb433c76d44276c58af597ad15f703e1cfb3503e8a527f2944951314c6a2a53e3b91cc54005d54c996e659288d4726e71d0565a33e7cef'
-  data.tar.gz: e4ef1fea768cd49fad82edf2ea4086b792118c545c7d12ea8772e0ba7c694793fa45c3024552ca35b9677d15a744a133a6f94bcf5f98a952008a6e5fefd9834b
+  metadata.gz: 6ab68e574725ebb40bad60ff6e807e480733c91f406d75c8e45b6678e16952e0cd67dde124df13ef9036ef82437bcdae97a4182169cbc615b30116f29c89667c
+  data.tar.gz: 6a4e0866fc61258b3fee11e76baeadc72ade5d5d66b87f4d8d41ad2127ac27a50a5f7ef394d0d940e03d7cc389e567f4d7e814ebb1792ef56c125904a75c4dcc

data/CHANGELOG.md

@@ -0,0 +1,11 @@
+## 0.2.0 - October 2, 2026
+- Log token usage
+- Stale code
+- Fix kwarg
+- Log costs
+- DirectoryReader includes hidden files
+- Prohibited path spec
+- Workspace::Helpers
+
+## 0.1.0 - September 12, 2026
+- First release

data/lib/aigency/aigent.rb

@@ -38,26 +38,30 @@ module Aigency
     # Initialize a RubyLLM::Chat.
     #
     # @param extra [Array<String>] Additional instructions
+    # @param tools [Array<Class<Tool>>] Available tools (default is all registered tools)
     # @return [RubyLLM::Chat]
-    def brief(*extra)
+    def brief(*extra, tools: Tools.registry)
       chat = rubyllm_context.chat(provider: :openai)
-      Tools.registry.each { |klass| chat.with_tool(klass.new(self)) }
+      tools.each { |klass| chat.with_tool(klass.new(self)) }
       chat.with_instructions(build_instructions(extra), append: true)
-      # @todo This should be configurable
       chat.with_params(max_tokens: max_tokens)
     end
 
-    def load filename, *extra, out: $stdout
-      stream File.read(filename), *extra, out: out
+    def load filename, *extra, tools: Tools.registry, out: $stdout
+      stream File.read(filename), *extra, tools: tools, out: out
     end
 
-    def stream text, *extra, with: [], out: $stdout
+    def stream text, *extra, with: [], tools: Tools.registry, out: $stdout
       original_log = @log
       @log = out
-      brief(*extra).ask text, with: with do |chunk|
+      response = brief(*extra, tools: tools).ask text, with: with do |chunk|
         @log.print chunk.thinking&.text
         @log.print chunk.content
-      end.tap { @log = original_log }
+      end
+      @log.puts "Input tokens:    #{response.tokens.input}"
+      @log.puts "Output tokens:   #{response.tokens.output}"
+      @log = original_log
+      response
     end
 
     # @param name [String]

data/lib/aigency/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Aigency
-  VERSION = "0.1.0"
+  VERSION = "0.2.0"
 end

data/lib/aigency/workspace/directory_deleter.rb

@@ -3,19 +3,18 @@
 module Aigency
   module Workspace
     class DirectoryDeleter < Tool
-      desc "Deletes a directory from your private workspace"
+      include Helpers
 
-      def execute(dirname:)
-        puts "Deleting directory #{dirname}"
+      desc "Delete a directory and all of its contents from your private workspace"
 
-        resolved = File.absolute_path(dirname, aigent.data('workspace'))
-        return { dirname: dirname, success: false, error: "Permission denied" } unless resolved.start_with?(aigent.data('workspace'))
-        return { dirname: dirname, success: false, error: "Permission denied" } if resolved.include?('/.git/') || resolved.end_with?('/.git')
-        return { dirname: dirname, success: false, error: "Not a directory" } if File.file?(resolved)
-        return { dirname: dirname, sucess: false, error: "Directory not found" } unless File.directory?(resolved)
+      def execute(directory_name:)
+        aigent.log.puts "[Workspace::DirectoryDeleter] #{directory_name}"
 
+        resolved = expand_to_workspace(directory_name)
         FileUtils.rm_rf(resolved)
-        { dirname: dirname, success: true }
+        { dirname: directory_name, success: true }
+      rescue StandardError => e
+        { success: false, error: "[#{e.class}] #{e.message}"}
       end
     end
   end

data/lib/aigency/workspace/directory_reader.rb

@@ -3,21 +3,18 @@
 module Aigency
   module Workspace
     class DirectoryReader < Tool
-      desc "Gets the contents of a directory from your private workspace"
+      include Helpers
+
+      desc "Get the contents of a directory from your private workspace"
 
       def execute(directory_name:)
         aigent.log.puts "[Workspace::DirectoryReader] Reading #{directory_name}"
-        directory = aigent.data('workspace')
-        resolved = File.absolute_path(directory_name, directory)
-        return { directory: directory_name,
-                 success: false,
-                 error: "Permission denied. Path must be local to #{directory}" } unless resolved.start_with?(directory)
-
+        resolved = expand_to_workspace(directory_name)
         if File.directory?(resolved)
           {
             directory: directory_name,
             success: true,
-            entries: Dir[File.join(resolved, '*')].map { |dir| dir[resolved.length+1..]}
+            entries: Dir.entries(resolved)[2..]
           }
         else
           {
@@ -26,6 +23,8 @@ module Aigency
             error: "Not a directory"
           }
         end
+      rescue StandardError => e
+        { success: false, error: "[#{e.class}] #{e.message}" }
       end
     end
   end

data/lib/aigency/workspace/file_deleter.rb

@@ -3,19 +3,17 @@
 module Aigency
   module Workspace
     class FileDeleter < Tool
-      desc "Deletes a file from your private workspace"
+      include Helpers
+
+      desc "Delete a file from your private workspace"
 
       def execute(filename:)
         aigent.log.puts "[Workspace::FileDeleter] Deleting file #{filename}"
-
-        directory = aigent.data('directory')
-        resolved = File.absolute_path(filename, directory)
-        return { filename: filename, error: "Permission denied" } unless resolved.start_with?(directory)
-        return { filename: filename, error: "Permission denied" } if resolved.include?('/.git/')
-        return { filename: filename, error: "File does not exist" } unless File.file?(resolved)
-
+        resolved = expand_to_workspace(filename)
         File.delete resolved
         { filename: filename, success: true }
+      rescue StandardError => e
+        { success: false, error: "[#{e.class}] #{e.message}" }
       end
     end
   end

data/lib/aigency/workspace/file_reader.rb

@@ -3,28 +3,22 @@
 module Aigency
   module Workspace
     class FileReader < Tool
-      desc "Reads a file from your private workspace"
+      include Helpers
+
+      desc "Read a file from your private workspace"
 
       def execute(filename:)
         aigent.log.puts "[Workspace::FileReader] Reading #{filename}"
 
-        directory = aigent.data('workspace')
-        resolved = File.absolute_path(filename, directory)
-        return { filename: filename, success: false, error: "Permission denied" } unless resolved.start_with?(directory)
+        resolved = expand_to_workspace(filename)
 
-        if File.file?(resolved)
-          {
-            filename: filename,
-            success: true,
-            content: File.read(resolved)
-          }
-        else
-          {
-            filename: filename,
-            success: false,
-            error: "File not found"
-          }
-        end
+        {
+          success: true,
+          filename: filename,
+          content: File.read(resolved)
+        }
+      rescue StandardError => e
+        { success: false, error: "[#{e.class}] #{e.message}"}
       end
     end
   end

data/lib/aigency/workspace/file_writer.rb

@@ -5,18 +5,19 @@ require 'fileutils'
 module Aigency
   module Workspace
     class FileWriter < Tool
-      desc "Writes a file to your private workspace"
+      include Helpers
+
+      desc "Write a file to your private workspace"
 
       def execute(filename:, content:)
         aigent.log.puts "[Workspace::FileWriter] Writing #{filename}"
 
-        directory = aigent.data('workspace')
-        resolved = File.absolute_path(filename, directory)
-        return { filename: filename, success: false, error: "Permission denied" } unless resolved.start_with?(directory)
-
+        resolved = expand_to_workspace(filename)
         FileUtils.mkdir_p File.dirname(resolved)
         File.write resolved, content
         return { filename: filename, success: true }
+      rescue StandardError => e
+        { success: false, error: "[#{e.class}] #{e.message}" }
       end
     end
   end

data/lib/aigency/workspace/helpers.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+module Aigency
+  module Workspace
+    module Helpers
+      # @!method aigent
+      #   @abstract
+      #   @return [Aigency::Aigent]
+
+      # Expand a path relative to the aigent's workspace directory.
+      #
+      # @raise [PathError] if the expanded path is not inside the workspace
+      # @param path [String]
+      # @return [String]
+      def expand_to_workspace(path)
+        workspace = aigent.data('workspace')
+        resolved = File.expand_path(path, workspace)
+        return resolved if resolved == workspace || resolved.start_with?("#{workspace}/")
+
+        raise PathError, "Permission denied: `#{path}` is not relative to your root workspace"
+      end
+    end
+  end
+end

data/lib/aigency/workspace.rb

@@ -2,9 +2,13 @@
 
 module Aigency
   module Workspace
+    require 'aigency/workspace/helpers'
+    require 'aigency/workspace/directory_deleter'
     require 'aigency/workspace/directory_reader'
     require 'aigency/workspace/file_reader'
     require 'aigency/workspace/file_writer'
+
+    class PathError < RuntimeError; end
   end
 
   Tools.mkdir_p 'data/workspace'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: aigency
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Fred Snyder
@@ -58,6 +58,7 @@ executables:
 extensions: []
 extra_rdoc_files: []
 files:
+- CHANGELOG.md
 - README.md
 - Rakefile
 - exe/aigency
@@ -74,6 +75,7 @@ files:
 - lib/aigency/workspace/file_deleter.rb
 - lib/aigency/workspace/file_reader.rb
 - lib/aigency/workspace/file_writer.rb
+- lib/aigency/workspace/helpers.rb
 - scaffold/.gitignore
 - scaffold/Gemfile
 - scaffold/IDENTITY.md