Fix spec

patterns-ai-core · Aug 27, 2024 · fd6eec9 · fd6eec9
1 parent 668bf4e
commit fd6eec9
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/lib/langchainrb_rails/generators/langchainrb_rails/prompt_generator.rb b/lib/langchainrb_rails/generators/langchainrb_rails/prompt_generator.rb
@@ -14,6 +14,10 @@ def create_prompt_model
         template "prompt_model.rb", "app/models/prompt.rb"
         migration_template "create_prompts.rb", "db/migrate/create_prompts.rb"
       end
+
+      def migration_version
+        "[#{::ActiveRecord::VERSION::MAJOR}.#{::ActiveRecord::VERSION::MINOR}]"
+      end
     end
   end
 end