lib/ad/agent_architecture/db/models.rb in ad-agent_architecture-0.0.6 vs lib/ad/agent_architecture/db/models.rb in ad-agent_architecture-0.0.7

@@ -1,13 +1,13 @@
 # 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'
       end