ruby_todo 1.0.0 → 1.0.3

data/lib/ruby_todo/database.rb

@@ -10,112 +10,86 @@ module RubyTodo
       def setup
         return if ActiveRecord::Base.connected?
 
-        db_path = File.expand_path("~/.ruby_todo/todo.db")
-        FileUtils.mkdir_p(File.dirname(db_path))
+        ensure_database_directory
+        establish_connection
+        create_tables
+      end
+
+      private
 
+      def ensure_database_directory
+        db_dir = File.expand_path("~/.ruby_todo")
+        FileUtils.mkdir_p(db_dir)
+      end
+
+      def establish_connection
+        db_path = File.expand_path("~/.ruby_todo/ruby_todo.db")
         ActiveRecord::Base.establish_connection(
           adapter: "sqlite3",
           database: db_path
         )
-
-        create_tables unless tables_exist?
-        check_schema_version
-      end
-
-      private
-
-      def tables_exist?
-        ActiveRecord::Base.connection.tables.include?("notebooks") &&
-          ActiveRecord::Base.connection.tables.include?("tasks") &&
-          ActiveRecord::Base.connection.tables.include?("schema_migrations")
       end
 
       def create_tables
+        connection = ActiveRecord::Base.connection
+
         ActiveRecord::Schema.define do
-          create_table :notebooks do |t|
-            t.string :name, null: false
-            t.timestamps
+          unless connection.table_exists?(:notebooks)
+            create_table :notebooks do |t|
+              t.string :name, null: false
+              t.boolean :is_default, default: false
+              t.timestamps
+            end
           end
 
-          create_table :tasks do |t|
-            t.references :notebook, null: false, foreign_key: true
-            t.string :title, null: false
-            t.string :status, null: false, default: "todo"
-            t.text :description
-            t.datetime :due_date
-            t.string :priority
-            t.string :tags
-            t.timestamps
+          unless connection.table_exists?(:tasks)
+            create_table :tasks do |t|
+              t.references :notebook, null: false
+              t.string :title, null: false
+              t.text :description
+              t.string :status, default: "todo"
+              t.datetime :due_date
+              t.string :priority
+              t.string :tags
+              t.timestamps
+            end
           end
 
-          create_table :templates do |t|
-            t.references :notebook, foreign_key: true
-            t.string :name, null: false
-            t.string :title_pattern, null: false
-            t.text :description_pattern
-            t.string :tags_pattern
-            t.string :priority
-            t.string :due_date_offset
-            t.timestamps
+          unless connection.table_exists?(:templates)
+            create_table :templates do |t|
+              t.string :name, null: false
+              t.string :title_pattern, null: false
+              t.text :description_pattern
+              t.string :priority
+              t.string :tags_pattern
+              t.string :due_date_offset
+              t.references :notebook
+              t.timestamps
+            end
           end
 
-          add_index :tasks, %i[notebook_id status]
-          add_index :tasks, :priority
-          add_index :tasks, :due_date
-          add_index :templates, :name, unique: true
-
-          create_table :schema_migrations do |t|
-            t.integer :version, null: false
+          # Add indexes if they don't exist
+          unless connection.index_exists?(:tasks, %i[notebook_id status])
+            add_index :tasks, %i[notebook_id status]
           end
-        end
 
-        # Set initial schema version
-        ActiveRecord::Base.connection.execute("INSERT INTO schema_migrations (version) VALUES (2)")
-      end
+          unless connection.index_exists?(:tasks, :priority)
+            add_index :tasks, :priority
+          end
 
-      def check_schema_version
-        # Get current schema version
-        current_version = ActiveRecord::Base.connection.select_value("SELECT MAX(version) FROM schema_migrations").to_i
+          unless connection.index_exists?(:tasks, :due_date)
+            add_index :tasks, :due_date
+          end
 
-        # If needed, perform migrations
-        if current_version < 1
-          upgrade_to_version_1
-        end
+          unless connection.index_exists?(:templates, :name)
+            add_index :templates, :name, unique: true
+          end
 
-        if current_version < 2
-          upgrade_to_version_2
+          unless connection.index_exists?(:notebooks, :is_default)
+            add_index :notebooks, :is_default
+          end
         end
       end
-
-      def upgrade_to_version_1
-        ActiveRecord::Base.connection.execute(<<-SQL)
-          ALTER TABLE tasks ADD COLUMN priority STRING;
-          ALTER TABLE tasks ADD COLUMN tags STRING;
-        SQL
-
-        ActiveRecord::Base.connection.execute("UPDATE schema_migrations SET version = 1")
-      end
-
-      def upgrade_to_version_2
-        ActiveRecord::Base.connection.execute(<<-SQL)
-          CREATE TABLE templates (
-            id INTEGER PRIMARY KEY AUTOINCREMENT,
-            notebook_id INTEGER,
-            name VARCHAR NOT NULL,
-            title_pattern VARCHAR NOT NULL,
-            description_pattern TEXT,
-            tags_pattern VARCHAR,
-            priority VARCHAR,
-            due_date_offset VARCHAR,
-            created_at DATETIME NOT NULL,
-            updated_at DATETIME NOT NULL,
-            FOREIGN KEY (notebook_id) REFERENCES notebooks(id)
-          );
-          CREATE UNIQUE INDEX index_templates_on_name ON templates (name);
-        SQL
-
-        ActiveRecord::Base.connection.execute("UPDATE schema_migrations SET version = 2")
-      end
     end
   end
 end

data/lib/ruby_todo/models/notebook.rb

@@ -5,8 +5,30 @@ require "active_record"
 module RubyTodo
   class Notebook < ActiveRecord::Base
     has_many :tasks, dependent: :destroy
+    has_many :templates
 
-    validates :name, presence: true, uniqueness: true
+    validates :name, presence: true
+    validates :name, uniqueness: true
+
+    scope :default, -> { where(is_default: true).first }
+
+    before_create :set_default_if_first
+    before_save :ensure_only_one_default
+
+    def self.default_notebook
+      find_by(is_default: true)
+    end
+
+    def make_default!
+      transaction do
+        Notebook.where.not(id: id).update_all(is_default: false)
+        update!(is_default: true)
+      end
+    end
+
+    def is_default?
+      is_default
+    end
 
     def tasks_by_status(status)
       tasks.where(status: status)
@@ -59,16 +81,28 @@ module RubyTodo
     def statistics
       {
         total: tasks.count,
-        todo: todo_tasks.count,
-        in_progress: in_progress_tasks.count,
-        done: done_tasks.count,
-        archived: archived_tasks.count,
-        overdue: overdue_tasks.count,
-        due_soon: due_soon_tasks.count,
-        high_priority: high_priority_tasks.count,
-        medium_priority: medium_priority_tasks.count,
-        low_priority: low_priority_tasks.count
+        todo: tasks.where(status: "todo").count,
+        in_progress: tasks.where(status: "in_progress").count,
+        done: tasks.where(status: "done").count,
+        archived: tasks.where(status: "archived").count,
+        overdue: tasks.select(&:overdue?).count,
+        due_soon: tasks.select(&:due_soon?).count,
+        high_priority: tasks.where(priority: "high").count,
+        medium_priority: tasks.where(priority: "medium").count,
+        low_priority: tasks.where(priority: "low").count
       }
     end
+
+    private
+
+    def set_default_if_first
+      self.is_default = true if Notebook.count.zero?
+    end
+
+    def ensure_only_one_default
+      return unless is_default_changed? && is_default?
+
+      Notebook.where.not(id: id).update_all(is_default: false)
+    end
   end
 end

data/lib/ruby_todo/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RubyTodo
-  VERSION = "1.0.0"
+  VERSION = "1.0.3"
 end

data/progress_ai_test.md

@@ -0,0 +1,126 @@
+# AI Assistant Command Tests
+
+This document tracks the testing of the `ruby_todo ai:ask` command with various natural language inputs to ensure all CLI functionalities work correctly.
+
+## Test Methodology
+
+For each test:
+1. Run `ruby delete_notebooks.rb` to reset the environment
+2. Run `echo "y" | ruby_todo import protectors_tasks.json` to restore test data
+3. Run the test command
+4. Verify the expected behavior
+
+## Test Results
+
+| Command Type | Natural Language Query | Expected Behavior | Result | Notes |
+|--------------|------------------------|-------------------|--------|-------|
+| Move Task | "move migrate arbitration-tf-shared to github actions" | Move task with ID 85 to in_progress status | ✅ Success | Successfully moved task to in_progress and provided confirmation message |
+| Move Multiple Tasks | "move migrate arbitration-tf-shared and awsappman-tf-accounts-management to github actions" | Move tasks with IDs 104 and 105 to in_progress status | ✅ Success | Successfully moved both tasks to in_progress and provided confirmation message |
+| Special Pattern Search | "move all migrate to barracuda org tasks to done" | Move all barracuda-related tasks (IDs 119-122) to done status | ✅ Success | Successfully identified and moved 4 barracuda-related tasks to done status |
+| Generic Repository Pattern | "move acme tasks to done" | Move task with "acme org" in title to done status | ✅ Success | Successfully identified and moved task with generic repository pattern |
+| Generic Repository Pattern (Different Syntax) | "move tasks related to quantum org to in_progress" | Move task with "quantum org" in title to in_progress status | ✅ Success | Successfully identified and moved task with different repository pattern syntax |
+| Generic Repository Pattern (GitHub) | "move task with Move app-repo/frontend to GitHub to todo" | Move task with "GitHub" in title to todo status | ✅ Success | Successfully identified and moved task with GitHub migration pattern |
+| Move to Archived | "move tappy-tf-shared to archived" | Move task with ID 149 to archived status | ✅ Success | Successfully moved task to archived status |
+| Status Mapping | "move tappy-aws-infrastructure to pending" | Move task with ID 166 to todo status | ✅ Success | Successfully mapped "pending" to "todo" status |
+| Partial Title Search | "move Add New Relic to in progress" | Move tasks with "New Relic" in their titles to in_progress status | ✅ Success | Successfully found and moved 4 matching tasks |
+| Task Listing | "show me all high priority tasks" | Show high priority tasks | ✅ Success | Now directly handles priority-filtered task listing |
+| Task Listing | "show me all medium priority tasks" | Show medium priority tasks | ✅ Success | Shows all medium priority tasks in default notebook |
+| Task Listing | "show me all todo tasks" | Show tasks with todo status | ✅ Success | Shows all tasks with todo status in default notebook |
+| Task Listing | "show me all in progress tasks" | Show tasks with in_progress status | ✅ Success | Shows all tasks with in_progress status in default notebook |
+| Notebook Listing | "show me all notebooks" | Show all notebooks | ✅ Success | Displays all notebooks in the system |
+| Statistics Request | "show statistics for protectors notebook" | Show notebook statistics | ✅ Success | Now directly handles statistics for specific notebooks |
+| Task Creation | "create a new task called 'Test task creation via AI' with high priority" | Create a new task | ✅ Success | Successfully creates new tasks with specified priority |
+| Move All Tasks | "move all tasks to todo" | Move all tasks to todo status | ✅ Success | Successfully moved all tasks in the notebook to todo status |
+| Move All Tasks (Different Status) | "move all tasks to in_progress" | Move all tasks to in_progress status | ✅ Success | Successfully moved all tasks in the notebook to in_progress status |
+
+## Final Smoke Test Results
+
+After implementing all direct command handlers, a comprehensive smoke test was conducted to verify the functionality of all supported commands:
+
+| Command Type | Test Command | Result | Notes |
+|--------------|--------------|--------|-------|
+| Task Movement | "move Add New Relic to in progress" | ✅ Success | Successfully moved 4 tasks to in_progress status |
+| Priority Filtering | "show me all high priority tasks" | ✅ Success | Correctly displayed all high priority tasks |
+| Priority Filtering | "show me all medium priority tasks" | ✅ Success | Correctly displayed all medium priority tasks |
+| Status Filtering | "show me all todo tasks" | ✅ Success | Correctly displayed all todo tasks |
+| Notebook Listing | "show all notebooks" | ✅ Success | Correctly displayed all notebooks |
+| Statistics | "show statistics for protectors notebook" | ✅ Success | Correctly displayed statistics for the notebook |
+| Task Creation (Quoted) | "create a new task called 'Smoke test AI assistant' with high priority" | ✅ Success | Successfully created new task with high priority |
+| Task Movement (Specific) | "move Smoke test AI assistant to in_progress" | ✅ Success | Successfully moved created task to in_progress |
+| Task Creation (Unquoted) | "add a task for final verification" | ✅ Success | Successfully created task without quoted title |
+| Status Filtering | "show me all in progress tasks" | ✅ Success | Correctly displayed all in-progress tasks including newly created ones |
+| Bulk Task Movement | "move all tasks to todo" | ✅ Success | Successfully moved all tasks in the default notebook to todo status |
+| Bulk Task Movement (Alt) | "move all tasks to in_progress" | ✅ Success | Successfully moved all tasks in the default notebook to in_progress status |
+| Repository Migration | "move migrate repositories to in progress" | ✅ Success | Successfully moved all repository migration tasks regardless of organization name |
+| Repository Migration (Other Org) | "move acme tasks to done" | ✅ Success | Successfully identified and moved repository migration tasks for non-barracuda organizations |
+| Repository Migration (Different Syntax) | "move tasks related to quantum org to in_progress" | ✅ Success | Successfully identified and moved task with "repo/name to org" pattern |
+| Repository Migration (GitHub) | "move task with Move app-repo/frontend to GitHub to todo" | ✅ Success | Successfully identified and moved task with GitHub migration pattern |
+
+The smoke test confirms that all implemented AI assistant functionality is working as expected. The direct command handlers successfully bypass the OpenAI API for common queries, making these operations faster and more reliable.
+
+## Conclusions
+
+The AI assistant has been significantly improved and now works effectively for:
+1. Task movement operations (as before)
+   - Single task movement
+   - Multiple task movement with compound search terms
+   - Special pattern recognition (repository migrations for any organization, not just "barracuda org")
+   - Support for different repository migration syntax patterns (both "to org" and "repo/name" formats)
+   - GitHub migration pattern recognition
+   - Status mappings (e.g., "pending" → "todo", "github actions" → "in_progress")
+   - Different statuses (todo, in_progress, done, archived)
+   - Bulk operations on all tasks ("move all tasks to...")
+
+2. Task listing operations
+   - Priority-filtered listings (high, medium)
+   - Status-filtered listings (todo, in_progress, done, archived)
+   - Notebook listings
+
+3. Statistics operations
+   - Notebook-specific statistics
+   - Global statistics
+
+4. Task creation
+   - Creating tasks with specified titles
+   - Setting priorities (high, medium, low)
+   - Adding to specific notebooks
+
+The direct handling approach bypasses the AI model for common queries, making these operations consistently reliable and efficient.
+
+Areas for further improvement:
+1. Add direct handling for task editing and deletion
+2. Add direct handling for template-related operations
+3. Add direct handling for due dates and tags in task creation
+4. Improve natural language understanding for complex queries
+
+These enhancements will require adding pattern recognition for each command type and implementing the corresponding CLI commands directly.
+
+## Implementation Notes
+
+The following approaches were used to improve the AI assistant:
+
+1. **Direct command handling**: Added pattern matching for common queries to bypass the AI and execute CLI commands directly
+2. **Improved error handling**: Enhanced command execution to better handle formatting issues and provide detailed debugging
+3. **Status mapping**: Fixed status mapping so that terms like "pending" correctly map to "todo"
+4. **Enhanced OpenAI system prompt**: Updated to provide clearer examples of command formats and expected outputs
+5. **Response cleaning**: Improved JSON parsing to handle malformed responses from the AI model
+6. **Intelligent pattern extraction**: Added logic to extract task titles, priorities, and notebook names from natural language queries
+7. **Contextual defaults**: Used the default notebook when specific notebooks aren't mentioned in queries
+8. **Special case handling**: Added special token pattern recognition for bulk operations like "move all tasks"
+9. **Generic pattern matching**: Updated repository migration pattern to work with any organization name, not just "barracuda"
+10. **Flexible syntax support**: Enhanced pattern matching to recognize different formats of repository migration commands
+11. **GitHub integration**: Added support for GitHub migration patterns in task search and movement
+
+These changes have greatly improved the reliability and consistency of the AI assistant, particularly for common task and notebook operations. The AI-powered system now provides a robust natural language interface that makes task management more intuitive and efficient.
+
+## Next Steps
+
+For future development, consider:
+
+1. **Expanding direct command handling**: Add handlers for remaining command types
+2. **Improving natural language processing**: Enhance pattern extraction for more complex queries
+3. **Adding conversational context**: Support follow-up queries that reference previous interactions
+4. **Implementing task editing**: Allow users to modify existing tasks via natural language
+5. **Supporting batch operations**: Enable operations across multiple notebooks at once
+
+The combination of direct command handling and AI-based fallback provides a robust architecture that balances reliability with flexibility.

data/protectors_tasks.json

@@ -0,0 +1,159 @@
+{
+  "name": "cox",
+  "created_at": "2024-04-01T00:00:00Z",
+  "updated_at": "2024-04-01T00:00:00Z",
+  "tasks": [
+    {
+      "title": "Add New Relic to tappy api",
+      "description": "Implement New Relic monitoring for tappy api",
+      "status": "todo",
+      "priority": "high",
+      "tags": "newrelic,monitoring,tappy",
+      "due_date": null
+    },
+    {
+      "title": "Add New Relic alerts and Pager Duty to Mission Control",
+      "description": "Implement New Relic alerts with low urgency Pager Duty integration for Mission Control",
+      "status": "todo",
+      "priority": "medium",
+      "tags": "newrelic,alerts,pagerduty,mission-control",
+      "due_date": null
+    },
+    {
+      "title": "Update assurance-postsale infrastructure",
+      "description": "Migrate to application load balancer, add New Relic infrastructure and alerts, update to Amazon Linux 2023, and update from OpenJDK8 to OpenJDK21 (EOL). Lock Docker image version instead of using latest.",
+      "status": "todo",
+      "priority": "high",
+      "tags": "migration,newrelic,infra,amazon-linux,openjdk,docker",
+      "due_date": null
+    },
+    {
+      "title": "Update apms-db-converter infrastructure",
+      "description": "Migrate to application load balancer, add New Relic infrastructure and alerts, update to Amazon Linux 2023, and update from OpenJDK8 to OpenJDK21 (EOL). Lock Docker image version instead of using latest.",
+      "status": "todo",
+      "priority": "high",
+      "tags": "migration,newrelic,infra,amazon-linux,openjdk,docker",
+      "due_date": null
+    },
+    {
+      "title": "Migrate MAN-Vintage/questions-engine-frontend to barracuda org",
+      "description": "Repository migration from MAN-Vintage to barracuda organization",
+      "status": "todo",
+      "priority": "medium",
+      "tags": "migration,repository,barracuda,vintage",
+      "due_date": null
+    },
+    {
+      "title": "Migrate MAN-Vintage/question-engine-api to barracuda org",
+      "description": "Repository migration from MAN-Vintage to barracuda organization",
+      "status": "todo",
+      "priority": "medium",
+      "tags": "migration,repository,barracuda,vintage",
+      "due_date": null
+    },
+    {
+      "title": "Migrate MAN-Fireflies/cr-popout-ui to barracuda org",
+      "description": "Repository migration from MAN-Fireflies to barracuda organization",
+      "status": "todo",
+      "priority": "medium",
+      "tags": "migration,repository,barracuda,fireflies",
+      "due_date": null
+    },
+    {
+      "title": "Migrate MAN-Fireflies/cr-popout-api to barracuda org",
+      "description": "Repository migration from MAN-Fireflies to barracuda organization",
+      "status": "todo",
+      "priority": "medium",
+      "tags": "migration,repository,barracuda,fireflies",
+      "due_date": null
+    },
+    {
+      "title": "Migrate arbitration-tf-shared to GitHub Actions",
+      "description": "Migrate terraform repository from current CI/CD to GitHub Actions",
+      "status": "todo",
+      "priority": "medium",
+      "tags": "migration,terraform,github-actions",
+      "due_date": null
+    },
+    {
+      "title": "Migrate awsappman-tf-accounts-management to GitHub Actions",
+      "description": "Migrate terraform repository from current CI/CD to GitHub Actions",
+      "status": "todo",
+      "priority": "medium",
+      "tags": "migration,terraform,github-actions",
+      "due_date": null
+    },
+    {
+      "title": "Migrate surcharge-tf-environment to GitHub Actions",
+      "description": "Migrate terraform repository from current CI/CD to GitHub Actions",
+      "status": "todo",
+      "priority": "medium",
+      "tags": "migration,terraform,github-actions",
+      "due_date": null
+    },
+    {
+      "title": "Migrate tapp-aurora-mysql to GitHub Actions",
+      "description": "Migrate terraform repository from current CI/CD to GitHub Actions",
+      "status": "todo",
+      "priority": "medium",
+      "tags": "migration,terraform,github-actions,aurora-mysql",
+      "due_date": null
+    },
+    {
+      "title": "Migrate tapp-aurora-postgresql to GitHub Actions",
+      "description": "Migrate terraform repository from current CI/CD to GitHub Actions",
+      "status": "todo",
+      "priority": "medium",
+      "tags": "migration,terraform,github-actions,aurora-postgresql",
+      "due_date": null
+    },
+    {
+      "title": "Migrate tappy-aws-infrastructure to GitHub Actions",
+      "description": "Migrate terraform repository from current CI/CD to GitHub Actions",
+      "status": "todo",
+      "priority": "medium",
+      "tags": "migration,terraform,github-actions,tappy-aws",
+      "due_date": null
+    },
+    {
+      "title": "Migrate tappy-mission-control-tf-environment to GitHub Actions",
+      "description": "Migrate terraform repository from current CI/CD to GitHub Actions",
+      "status": "todo",
+      "priority": "medium",
+      "tags": "migration,terraform,github-actions,mission-control",
+      "due_date": null
+    },
+    {
+      "title": "Migrate tappy-tf-shared to GitHub Actions",
+      "description": "Migrate terraform repository from current CI/CD to GitHub Actions",
+      "status": "todo",
+      "priority": "medium",
+      "tags": "migration,terraform,github-actions,tappy-shared",
+      "due_date": null
+    },
+    {
+      "title": "Migrate tappy-wordpress-tf-environment to GitHub Actions",
+      "description": "Migrate terraform repository from current CI/CD to GitHub Actions",
+      "status": "todo",
+      "priority": "medium",
+      "tags": "migration,terraform,github-actions,wordpress",
+      "due_date": null
+    },
+    {
+      "title": "Migrate tappy2-client-tf-environment to GitHub Actions",
+      "description": "Migrate terraform repository from current CI/CD to GitHub Actions",
+      "status": "todo",
+      "priority": "medium",
+      "tags": "migration,terraform,github-actions,tappy2-client",
+      "due_date": null
+    },
+    {
+      "title": "Migrate tappy_mysql to GitHub Actions",
+      "description": "Migrate terraform repository from current CI/CD to GitHub Actions",
+      "status": "todo",
+      "priority": "medium",
+      "tags": "migration,terraform,github-actions,tappy-mysql",
+      "due_date": null
+    }
+  ]
+} 

data/test_ai_assistant.rb

@@ -0,0 +1,55 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require_relative "lib/ruby_todo"
+
+# Create notebooks and tasks
+puts "Creating test notebooks and tasks..."
+notebook = RubyTodo::Notebook.create(name: "Test Notebook")
+notebook.update!(is_default: true)
+
+notebook.tasks.create(
+  title: "Migrate to the barracua org",
+  status: "todo",
+  description: "Migration task for repository"
+)
+
+notebook.tasks.create(
+  title: "Task 2",
+  status: "todo"
+)
+
+notebook.tasks.create(
+  title: "Regular task not related to migrations",
+  status: "in_progress"
+)
+
+# Create the AI Assistant
+ai_command = RubyTodo::AIAssistantCommand.new([], { verbose: true })
+
+# Test the problematic case
+puts "\n\n=== Testing problematic case ==="
+prompt = "move all migrate to the barracua org tasks to in porgress"
+
+puts "Running test with prompt: '#{prompt}'"
+context = {}
+
+ai_command.send(:handle_task_request, prompt, context)
+
+# Display results
+puts "\n=== Results ==="
+if context[:matching_tasks]&.any?
+  puts "Found #{context[:matching_tasks].size} matching tasks:"
+  context[:matching_tasks].each do |task|
+    puts "- #{task[:title]} (ID: #{task[:task_id]}, Notebook: #{task[:notebook]})"
+  end
+  puts "Target status: #{context[:target_status]}"
+else
+  puts "No matching tasks found"
+end
+
+# Clean up
+puts "\n=== Cleaning up ==="
+RubyTodo::Task.delete_all
+RubyTodo::Notebook.delete_all
+puts "Test completed"

data/test_migration.rb

@@ -0,0 +1,55 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require_relative "lib/ruby_todo"
+
+# Create notebooks and tasks
+puts "Creating test notebooks and tasks..."
+notebook = RubyTodo::Notebook.create(name: "Test Notebook")
+notebook.update!(is_default: true)
+
+notebook.tasks.create(
+  title: "Migrate to the barracua org",
+  status: "todo",
+  description: "Migration task for repository"
+)
+
+notebook.tasks.create(
+  title: "Task 2",
+  status: "todo"
+)
+
+notebook.tasks.create(
+  title: "Regular task not related to migrations",
+  status: "in_progress"
+)
+
+# Create the AI Assistant
+ai_command = RubyTodo::AIAssistantCommand.new([], { verbose: true })
+
+# Test the problematic case
+puts "\n\n=== Testing problematic case ==="
+prompt = "move all migrate to the barracua org tasks to in porgress"
+
+puts "Running test with prompt: '#{prompt}'"
+context = {}
+
+ai_command.send(:handle_task_request, prompt, context)
+
+# Display results
+puts "\n=== Results ==="
+if context[:matching_tasks]&.any?
+  puts "Found #{context[:matching_tasks].size} matching tasks:"
+  context[:matching_tasks].each do |task|
+    puts "- #{task[:title]} (ID: #{task[:task_id]}, Notebook: #{task[:notebook]})"
+  end
+  puts "Target status: #{context[:target_status]}"
+else
+  puts "No matching tasks found or no target status detected"
+end
+
+# Clean up
+puts "\n=== Cleaning up ==="
+RubyTodo::Task.delete_all
+RubyTodo::Notebook.delete_all
+puts "Test completed"