job-workflow 0.6.0 → 0.6.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fbf6231d334745f4814282d93fa8c7b54461680e6bb57edc6632cd39b86cb71b
-  data.tar.gz: c627082defb65e2a4cd79cb2b88d3f54ce8b082a88d9ccaff3d96029ee49ea65
+  metadata.gz: 54b1b4244da651dd7a48b083dbff5194d414964a2269fa53580d3d6b68f9428e
+  data.tar.gz: bad5b87fe9e7f60699841de280554e05d8e07025d0cc01bdaa972b1ca24f37d0
 SHA512:
-  metadata.gz: f8caf717541c7f549d3957a5939d636881f8ad98389e5e53759dde43e561c8bbdeda0211feab0843f2625fbc974fb217b479bc2421e0bc828b9f6d910b749cd5
-  data.tar.gz: 466d87d40597d0d3c68de4cec3f2f6d158faab74100d098420cd376108fb88667f15cf1ecccdf2b810e41df402a9db5f7d84945a71de2dd756e71ae99588fe58
+  metadata.gz: 445ed81a6feb664a4f9ef139f7178e74701c9507b2cb0e91b017b9998b076bb8d7000bf2ce169a965ae797002ebe7e82596e18bb03200a06b22fd2ebac29ee5c
+  data.tar.gz: 35242db3bb8c3d0cf11f35be438eae04f929af4c77268326990512ca124acfe0b63876dd3618f9e24d44c1e04ab4014293f14ff90d540c4810b594b88e57a0a8

data/.agents/instructions/coding-style.md

@@ -0,0 +1,38 @@
+# Coding Style
+
+## Primary sources of truth
+
+- `.rubocop.yml` for Ruby style, complexity, and naming constraints
+- `Steepfile`, `rbs_collection.yaml`, and inline RBS usage for type-checking expectations
+- `.github/copilot-instructions.md` for repository-specific design and testing rules
+
+## Mechanical style constraints
+
+- Ruby string literals use double quotes.
+- RuboCop line length is capped at 120 characters.
+- The repository excludes `examples/**/*` from the root RuboCop config because the example app validates itself separately.
+
+## Design rules
+
+- Keep code small and focused on a single responsibility.
+- Prefer composable behavior over large monolithic methods.
+- Favor immutable workflow inputs through `Arguments`.
+- Use `Output` as the main way to communicate task side effects and downstream data.
+- Preserve clear boundaries between `Workflow`, `Task`, and `Runner`.
+- Avoid breaking public APIs without a documented migration or deprecation path.
+
+## Type and signature rules
+
+- Prefer `rbs-inline` comments in implementation files when type information changes.
+- Do not edit generated `.rbs` files or `sig/` artifacts directly unless maintainers explicitly approve it.
+
+## Test-adjacent implementation rules
+
+- Add only the code necessary for the requested behavior and its directly related fixes.
+- Reuse existing helpers and patterns before adding new abstractions.
+- Comment code only when the behavior is non-obvious.
+- Do not change established naming conventions without approval.
+
+## Unconfirmed items
+
+- Any additional hand-written naming rules beyond Ruby conventions and RuboCop enforcement

data/.agents/instructions/domain.md

@@ -0,0 +1,37 @@
+# Domain
+
+## Repository purpose
+
+JobWorkflow is a workflow orchestration library for Ruby on Rails applications. It provides a declarative DSL on top of ActiveJob so applications can define task graphs, dependencies, fan-out work, retries, throttling, and monitoring.
+
+## Core concepts
+
+- `Workflow`: the ordered graph of tasks for a job class
+- `Task`: one unit of work in a workflow, optionally depending on other tasks
+- `Arguments`: immutable workflow inputs exposed to tasks through `Context`
+- `Context`: the task-facing object used to read arguments, outputs, runtime helpers, and execution state
+- `Output`: the structured result store for completed task outputs
+- `Runner`: the orchestration layer that executes tasks and updates workflow state
+- `SubTaskJob`: the dedicated async job class used for `enqueue: true` fan-out work
+- `WorkflowStatus` / `JobStatus`: read models for workflow and sub-task execution state
+- Monitoring UI: the Rails engine that visualizes workflow definitions and execution DAGs
+
+## Common terminology
+
+- `each task`: a task that fans out over a collection
+- `dependency_wait`: waiting behavior for dependent async work
+- `fan-out`: splitting work into sub-jobs
+- `root execution`: the top-level workflow job, excluding sub-task jobs
+
+## Detailed references
+
+- `guides/GETTING_STARTED.md` for the high-level workflow model
+- `guides/DSL_BASICS.md` for task definitions and dependency rules
+- `guides/TASK_OUTPUTS.md` for output flow and downstream consumption
+- `guides/PARALLEL_PROCESSING.md` for fan-out and aggregation behavior
+- `guides/WORKFLOW_STATUS_QUERY.md` for workflow vs job status read models
+- `guides/MONITORING_UI.md` for the monitoring view of root executions and sub-task jobs
+
+## Unconfirmed items
+
+- Additional external domain vocabulary used by downstream adopters

data/.agents/instructions/environment.md

@@ -0,0 +1,44 @@
+# Environment
+
+## Required local tools
+
+- Ruby and Bundler for gem development
+- SQLite3 for local development and the example app
+- A shell environment that can run the root and example-app Rake tasks
+
+## Root setup
+
+```bash
+bin/setup
+bundle install
+bundle exec rake rbs:install
+```
+
+## Example app setup
+
+From `examples/rails_8_1/`:
+
+```bash
+bundle install
+bundle exec rails db:prepare
+bundle exec rake rbs:install
+```
+
+## Running the example app
+
+- Start the Rails server from `examples/rails_8_1/` with `bin/rails server`
+- Start background job processing with `bin/jobs`
+- Open `/job_workflow` for the monitoring UI and `/jobs` for Mission Control Jobs in the example app
+- For a denser monitoring DAG preview, enqueue `AcceptanceComplexMonitoringDagJob` from `examples/rails_8_1/`
+
+## Useful notes
+
+- The example app is the main place to verify real monitoring UI behavior.
+- The example app has its own lockfile and validation commands; treat it as an additional maintained surface.
+- Use `examples/rails_8_1/README.md` for the full preview sequence.
+- Use `guides/GETTING_STARTED.md` for first-time library setup and `guides/MONITORING_UI.md` for monitoring behavior.
+
+## Unconfirmed items
+
+- Preferred Ruby version manager for maintainers
+- Any editor or shell configuration the team expects

data/.agents/instructions/general.md

@@ -0,0 +1,29 @@
+# General
+
+## Core behavior
+
+- Respond to repository collaborators in Japanese, even though these agent resource files are written in English.
+- Interpret user instructions literally and keep changes scoped to the requested task.
+- Ask for clarification before making risky, ambiguous, or behavior-changing edits.
+- Prefer existing repository conventions over personal defaults.
+
+## Always do
+
+- Read the relevant repository context before editing.
+- Keep changes focused and avoid unrelated cleanup.
+- Report what changed, what was validated, and any unresolved points when finishing work.
+
+## Do not do
+
+- Do not create unrelated refactors while working on a user request.
+- Do not assume undocumented behavior when repository sources do not confirm it.
+- Do not treat generated signatures in `sig/generated/` as the primary editing surface.
+
+## Communication
+
+- Use concise Japanese in user-facing chat.
+- State uncertainty explicitly instead of presenting guesses as facts.
+
+## Unconfirmed items
+
+- Additional organization-wide communication rules beyond this repository

data/.agents/instructions/security.md

@@ -0,0 +1,20 @@
+# Security
+
+## Secrets and sensitive data
+
+- Do not commit API keys, tokens, passwords, private keys, or other secrets.
+- Stop and ask if a value looks like a secret or production credential.
+
+## Restricted operations
+
+- Do not modify files outside the repository root.
+
+## Dependency and code safety
+
+- Prefer existing repository dependencies and tools over introducing new ones.
+- Avoid risky shell operations, destructive git commands, and history rewriting unless explicitly requested.
+- Surface errors instead of silently swallowing them.
+
+## Unconfirmed items
+
+- External secret-management system used by maintainers

data/.agents/instructions/structure.md

@@ -0,0 +1,43 @@
+# Structure
+
+## Repository layout
+
+```text
+.
+├── app/                     # Rails engine assets for the monitoring UI
+├── bin/                     # executable helpers
+├── config/                  # Rails engine and routing configuration
+├── examples/rails_8_1/      # acceptance Rails application used to verify integration
+├── guides/                  # user-facing documentation
+├── lib/                     # main JobWorkflow library code
+├── sig/                     # generated signatures and type artifacts
+├── sig-private/             # private type definitions used by Steep
+├── spec/                    # main RSpec suite
+├── .agents/instructions/    # agent instructions files referenced from AGENTS.md
+└── AGENTS.md                # root agent entry point
+```
+
+## Key areas
+
+- `lib/job_workflow/` holds the core DSL, runtime, adapters, monitoring models, and version file.
+- `lib/job_workflow/monitoring/` contains monitoring presenters, registries, and layout helpers.
+- `app/` and `config/routes.rb` support the monitoring UI engine.
+- `examples/rails_8_1/` is a separate validation surface with its own Rake tasks, lockfile, and specs.
+- `examples/rails_8_1/app/jobs/` contains acceptance workflow definitions.
+- `examples/rails_8_1/spec/jobs/` contains acceptance and integration-oriented example-app specs.
+- `guides/` is the documentation index for feature and operational guides.
+
+## Structure expectations
+
+- Treat this repository as a single package with an embedded example application, not as a monorepo of independent packages.
+- Keep agent-wide documentation at the root unless a future task explicitly introduces subdirectory-specific AGENTS.md files.
+
+## Documentation entry points
+
+- Start with `guides/README.md` to find the right feature guide.
+- Use `guides/MONITORING_UI.md` for monitoring-specific behavior.
+- Use `examples/rails_8_1/README.md` for example-app setup and preview flows.
+
+## Unconfirmed items
+
+- Whether additional future example applications should get their own AGENTS.md

data/.agents/instructions/tech-stack.md

@@ -0,0 +1,40 @@
+# Tech Stack
+
+## Languages and frameworks
+
+- Ruby `>= 3.1.0` for the gem runtime
+- Rails and ActiveJob for the workflow runtime and monitoring engine integration
+- Solid Queue in the example app for asynchronous execution scenarios
+- SQLite in local and example-app development flows
+- RBS Inline plus Steep for type checking
+- RSpec for tests
+- RuboCop with repository plugins for linting
+- SimpleCov for 100% line and branch coverage enforcement
+
+## Primary commands
+
+- Root validation:
+  - `bundle exec rake spec`
+  - `bundle exec rake lint`
+  - `bundle exec rake typecheck`
+- Useful root helpers:
+  - `bundle exec rake lint:fix`
+  - `bundle exec rake lint:fixall`
+  - `bundle exec rake rbs:install`
+  - `bundle exec rake rbs:update`
+  - `bundle exec rake rbs:inline`
+- Example app validation from `examples/rails_8_1/`:
+  - `bundle exec rake spec`
+  - `bundle exec rake lint`
+  - `bundle exec rake typecheck`
+
+## Architecture summary
+
+- `Workflow`, `Task`, `Runner`, `Arguments`, `Context`, and `Output` form the core execution model.
+- Queue adapters isolate runtime-specific job lookup and persistence behavior.
+- Monitoring code lives under `lib/job_workflow/monitoring/` and is rendered through the Rails engine in `app/`.
+- The example Rails app is the main integration harness for real ActiveJob and Solid Queue behavior.
+
+## Important command rule
+
+- Use the repository Rake tasks instead of calling `rspec`, `rubocop`, or `steep` directly for final validation.

data/.agents/instructions/testing.md

@@ -0,0 +1,46 @@
+# Testing
+
+## Required validation
+
+- Every functional change must include new or updated specs that cover the changed behavior.
+- Root repository changes must pass:
+  - `bundle exec rake spec`
+  - `bundle exec rake lint`
+  - `bundle exec rake typecheck`
+- Root coverage must remain at `100%` line coverage and `100%` branch coverage.
+
+## Example app validation
+
+- If a change touches `examples/rails_8_1/`, also run inside that directory:
+  - `bundle exec rake spec`
+  - `bundle exec rake lint`
+  - `bundle exec rake typecheck`
+- Example app coverage must also stay at `100%` line coverage and `100%` branch coverage.
+
+## Validation scope
+
+- If the change only touches root library, engine, guides, or specs outside `examples/rails_8_1/`, run the root validation set.
+- If the change touches both root code and `examples/rails_8_1/`, run both validation sets.
+- If the change alters behavior, update the relevant guide examples as well as tests.
+
+## Test-writing rules
+
+- Use `bundle exec rake spec`, not `bundle exec rspec`.
+- Define a named subject with `subject(:name)`.
+- Keep `describe` / `context` nesting at three levels or fewer.
+- Prefer one expectation per example unless a matcher like `have_attributes` or a combined matcher makes the assertion a single behavior check.
+- Test through public APIs only; do not use `instance_variable_set`, `instance_variable_get`, or similar reflection in specs.
+- Keep `let` usage lean and split contexts when conditions differ.
+
+## Completion rule
+
+- Do not report a code change as ready until the required validation commands for the touched surfaces have succeeded.
+
+## Reference material
+
+- Use `guides/TESTING_STRATEGY.md` for testing guidance beyond the mandatory checks.
+- Use `guides/MONITORING_UI.md` and feature-specific guides when behavior-oriented expectations need clarification.
+
+## Unconfirmed items
+
+- Whether any additional external CI jobs run outside the repository-local Rake tasks

data/.agents/instructions/workflow.md

@@ -0,0 +1,39 @@
+# Workflow
+
+## Branches
+
+- Use a dedicated branch for work.
+- Preferred branch prefixes are `feature/<short-desc>` and `fix/<short-desc>`.
+
+## Commits
+
+- Use Conventional Commits in `type(scope): short summary` format.
+- Valid types include `feat`, `fix`, `refactor`, `docs`, `test`, `chore`, and `perf`.
+- Keep the subject line within 72 characters.
+- Update `CHANGELOG.md` before creating a commit when the change should be reflected in release notes.
+
+## Pull requests
+
+- Keep one PR focused on one logical change.
+- PR creation and merge are normally done by the user unless they explicitly delegate that step.
+- A completion handoff should include validation results, coverage, lint status, typecheck status, major changed files, and any recommended follow-up.
+
+## Before changing behavior
+
+- Check `guides/README.md` first to find the relevant feature guide.
+- Use feature guides such as `guides/DSL_BASICS.md`, `guides/TASK_OUTPUTS.md`, `guides/PARALLEL_PROCESSING.md`, `guides/DEPENDENCY_WAIT.md`, and `guides/MONITORING_UI.md` before changing semantics in those areas.
+
+## Release preparation
+
+- Record shipped changes in `CHANGELOG.md`, starting from `## [Unreleased]` before they are cut into a release section.
+- Keep version-related files in sync when preparing a release.
+
+## Review expectations
+
+- Explain why a change is necessary.
+- Keep PRs small enough to review without mixing unrelated work.
+
+## Unconfirmed items
+
+- Reviewer assignment rules outside the repository instructions
+- Whether release tagging always happens from `main`

data/.rubocop.yml

@@ -8,6 +8,7 @@ AllCops:
   TargetRubyVersion: 3.1
   NewCops: enable
   SuggestExtensions: false
+  UseProjectIndex: true
   Exclude:
     - examples/**/*
     - tmp/**/*
@@ -80,6 +81,15 @@ RSpec/ExampleLength:
 Style/Documentation:
   Enabled: false
 
+Style/RbsInline:
+  Mode: opt_out
+
+Style/RbsInline/MissingTypeAnnotation:
+  EnforcedStyle: method_type_signature
+
+Style/RbsInline/RedundantTypeAnnotation:
+  EnforcedStyle: method_type_signature
+
 Style/StringLiterals:
   EnforcedStyle: double_quotes
 

data/AGENTS.md

@@ -0,0 +1,23 @@
+# AGENTS.md
+
+This repository uses AGENTS.md as a thin hub. Read the required instructions files before making changes.
+
+## Required instructions
+
+- @.agents/instructions/general.md
+- @.agents/instructions/structure.md
+- @.agents/instructions/tech-stack.md
+- @.agents/instructions/security.md
+- @.agents/instructions/testing.md
+- @.agents/instructions/workflow.md
+
+## Load when needed
+
+- .agents/instructions/domain.md — JobWorkflow concepts and terminology
+- .agents/instructions/environment.md — local setup and example app workflow
+- .agents/instructions/coding-style.md — design rules, style constraints, and implementation patterns
+
+## Repository notes
+
+- Repository-specific reusable agent assets should live under `.agents/skills/`.
+- No subdirectory-level AGENTS.md files are defined at the moment. Use this root entry point unless a later task adds a more specific AGENTS.md.

data/CHANGELOG.md

@@ -1,5 +1,19 @@
 ## [Unreleased]
 
+## [0.6.2] - 2026-08-09
+
+### Changed
+
+- Extend RBS/Steep type checking to the monitoring Rails controllers and add signatures for Rails, Mission Control Jobs, and Solid Queue integrations
+- Upgrade the RBS inline, Steep, and RuboCop development tooling used by the type and lint checks
+- Keep the Rails 8.1 example app's lint task compatible with the root RuboCop configuration
+
+## [0.6.1] - 2026-06-01
+
+### Fixed
+
+- Fix monitoring UI task state rendering so tasks recorded as completed in ActiveJob continuation data no longer appear as `pending` when they have no outputs or sub-task statuses
+
 ## [0.6.0] - 2026-05-24
 
 ### Added

data/README.md

@@ -1,6 +1,6 @@
 # JobWorkflow
 
-> ⚠️ **Early Stage (v0.6.0):** This library is in active development. APIs and features may change in breaking ways without notice. Use in production at your own risk and expect potential breaking changes in future releases.
+> ⚠️ **Early Stage (v0.6.2):** This library is in active development. APIs and features may change in breaking ways without notice. Use in production at your own risk and expect potential breaking changes in future releases.
 
 ## Overview
 

data/Rakefile

@@ -40,7 +40,7 @@ namespace :rbs do
     require "rbs/inline"
     require "rbs/inline/cli"
     FileUtils.rm_r(File.expand_path("sig/generated", __dir__), secure: true)
-    RBS::Inline::CLI.new.run(%w[lib --output --opt-out])
+    RBS::Inline::CLI.new.run(%w[app lib --output --opt-out])
   end
 end
 

data/Steepfile

@@ -5,6 +5,10 @@
 target :lib do
   signature "sig"
   signature "sig-private"
+  library "actionpack"
+  library "actionview"
+  library "rack"
 
   check "lib"
+  check "app"
 end

data/app/controllers/job_workflow/monitoring/application_controller.rb

@@ -4,6 +4,7 @@ module JobWorkflow
   module Monitoring
     # Resolve the host app controller at load time so engine controllers inherit
     # the app's existing authentication and access control hooks.
+    # @rbs inherits ActionController::Base
     class ApplicationController < JobWorkflow::Monitoring.resolved_base_controller_class.constantize
       layout "job_workflow/monitoring/application"
     end

data/app/controllers/job_workflow/monitoring/executions_controller.rb

@@ -3,6 +3,12 @@
 module JobWorkflow
   module Monitoring
     class ExecutionsController < ApplicationController
+      # @rbs @workflow: singleton(DSL)
+      # @rbs @page: ExecutionPage
+      # @rbs @executions: Array[ExecutionViewModel]
+      # @rbs @execution: ExecutionViewModel
+
+      #:  () -> void
       def index
         @workflow = WorkflowRegistry.find(params[:workflow_job_class_name])
         return render plain: "Workflow definition not found.", status: :not_found if @workflow.nil?
@@ -14,6 +20,7 @@ module JobWorkflow
         @executions = @page.executions
       end
 
+      #:  () -> void
       def show
         @workflow = WorkflowRegistry.find(params[:workflow_job_class_name])
         return render plain: "Workflow definition not found.", status: :not_found if @workflow.nil?

data/app/controllers/job_workflow/monitoring/workflows_controller.rb

@@ -3,6 +3,9 @@
 module JobWorkflow
   module Monitoring
     class WorkflowsController < ApplicationController
+      # @rbs @workflows: Array[WorkflowDefinition]
+
+      #:  () -> void
       def index
         @workflows = Monitoring.workflows
       end

data/guides/PRODUCTION_DEPLOYMENT.md

@@ -1,6 +1,6 @@
 # Production Deployment
 
-> ⚠️ **Early Stage (v0.6.0):** JobWorkflow is still in early development. While this section outlines potential deployment patterns, please thoroughly test in your specific environment and monitor for any issues before relying on JobWorkflow in critical production systems.
+> ⚠️ **Early Stage (v0.6.2):** JobWorkflow is still in early development. While this section outlines potential deployment patterns, please thoroughly test in your specific environment and monitor for any issues before relying on JobWorkflow in critical production systems.
 
 This section covers suggested settings and patterns for running JobWorkflow in production-like environments.
 

data/guides/README.md

@@ -1,6 +1,6 @@
 # JobWorkflow Guides
 
-> ⚠️ **Early Stage (v0.6.0):** JobWorkflow is in active development. APIs and features may change. The following guides provide patterns and examples for building workflows, but be aware that implementations may need adjustment as the library evolves.
+> ⚠️ **Early Stage (v0.6.2):** JobWorkflow is in active development. APIs and features may change. The following guides provide patterns and examples for building workflows, but be aware that implementations may need adjustment as the library evolves.
 
 Welcome to the JobWorkflow documentation! This directory contains comprehensive guides to help you build robust workflows with JobWorkflow.
 

data/lib/job_workflow/auto_scaling/executor.rb

@@ -3,6 +3,8 @@
 module JobWorkflow
   module AutoScaling
     class Executor
+      # @rbs @adapter: Adapter::_InstanceMethods
+
       #:  (Configuration) -> void
       def initialize(config)
         @config = config
@@ -36,7 +38,7 @@ module JobWorkflow
 
       #:  () -> Array[Integer]
       def desired_count_list
-        config.min_count.step(config.max_count, config.step_count).to_a
+        config.min_count.step(config.max_count, config.step_count).to_a #: Array[Integer]
       end
     end
   end

data/lib/job_workflow/context.rb

@@ -120,8 +120,7 @@ module JobWorkflow
 
     #:  () -> String
     def job_id
-      local_job = job
-      raise "job is not set" if local_job.nil?
+      local_job = job || (raise "job is not set")
 
       local_job.job_id
     end
@@ -215,11 +214,8 @@ module JobWorkflow
 
     #:  (?Symbol?, ?fallback: untyped) { () -> untyped } -> untyped
     def skip_in_dry_run(dry_run_name = nil, fallback: nil)
-      local_job = job
-      task = task_context.task
-
-      raise "job is not set" if local_job.nil?
-      raise "skip_in_dry_run can be called only within with_task_context" if task.nil?
+      local_job = job || (raise "job is not set")
+      task_context.task || (raise "skip_in_dry_run can be called only within with_task_context")
 
       current_index = skip_in_dry_run_index
       self.skip_in_dry_run_index += 1
@@ -298,7 +294,8 @@ module JobWorkflow
 
     #:  () -> bool
     def each_task?
-      task_context.task.each?
+      task = task_context.task || (raise "each_task? can be called only in task")
+      task.each?
     end
 
     #:  (untyped) -> untyped

data/lib/job_workflow/dsl.rb

@@ -2,6 +2,8 @@
 
 module JobWorkflow
   module DSL
+    # @rbs @_context: Context?
+
     extend ActiveSupport::Concern
 
     include ActiveJob::Continuable
@@ -19,6 +21,8 @@ module JobWorkflow
     #
     #   def queue_name: () -> String
     #
+    #   def arguments: () -> Array[untyped]
+    #
     #   def set: (Hash[Symbol, untyped]) -> self
     #
     #   def step: (Symbol, ?start: ActiveJob::Continuation::_Succ, ?isolated: bool) -> void
@@ -33,13 +37,13 @@ module JobWorkflow
     #:  (Hash[untyped, untyped]) -> void
     def perform(arguments)
       self._context ||= Context.from_hash({ job: self, workflow: self.class._workflow })
-      context = self._context #: Context
+      context = _context #: Context
       Runner.new(context: context._update_arguments(arguments)).run
     end
 
     #:  () -> Output
     def output
-      context = self._context
+      context = _context
       raise "context is not set." if context.nil?
 
       context.output
@@ -211,14 +215,16 @@ module JobWorkflow
       #   ) -> void
       def workflow_concurrency(to:, key:, **opts)
         concurrency_key_proc = key
+        wrapped_key = proc {
+          # @type self: DSL
+          ctx = _context || Context.from_hash(
+            job: self, workflow: self.class._workflow
+          )._update_arguments((arguments.first || {}).symbolize_keys)
+          concurrency_key_proc.call(ctx)
+        } #: ^(untyped) -> untyped
         limits_concurrency(
           to:,
-          key: proc {
-            ctx = _context || Context.from_hash(
-              job: self, workflow: self.class._workflow
-            )._update_arguments((arguments.first || {}).symbolize_keys)
-            concurrency_key_proc.call(ctx)
-          },
+          key: wrapped_key,
           **opts
         )
       end

data/lib/job_workflow/instrumentation/opentelemetry_subscriber.rb

@@ -45,6 +45,7 @@ module JobWorkflow
       # @rbs!
       #   def self.subscriptions: () -> Array[untyped]
       #   def self.subscriptions=: (Array[untyped]) -> void
+
       cattr_accessor :subscriptions, instance_accessor: false, default: []
 
       class << self
@@ -110,6 +111,7 @@ module JobWorkflow
         [span, token]
       end
 
+      #:  (String, Hash[Symbol, untyped]) -> untyped
       def start_span(name, payload)
         span_name = build_span_name(name, payload)
         attributes = build_attributes(payload)

data/lib/job_workflow/logger.rb

@@ -14,8 +14,7 @@ module JobWorkflow
   #   JobWorkflow.logger.formatter = JobWorkflow::Logger::JsonFormatter.new(log_tags: [:request_id])
   #   ```
   module Logger
-    #:  (ActiveSupport::Logger) -> void
-    attr_writer :logger
+    attr_writer :logger #: ActiveSupport::Logger
 
     #:  () -> ActiveSupport::Logger
     def logger