Skip to content
This repository was archived by the owner on Jul 22, 2025. It is now read-only.

Commit e8edc0b

Browse files
committed
Migration for Mistral tokenizer rename
1 parent 9f77d1c commit e8edc0b

File tree

1 file changed

+43
-0
lines changed

1 file changed

+43
-0
lines changed
Lines changed: 43 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,43 @@
1+
# frozen_string_literal: true
2+
3+
class RenameMixtralTokenizerToMistralTokenizer < ActiveRecord::Migration[7.2]
4+
def up
5+
execute <<~SQL
6+
UPDATE
7+
llm_models
8+
SET
9+
tokenizer = 'DiscourseAi::Tokenizer::Mistral'
10+
WHERE
11+
tokenizer = 'DiscourseAi::Tokenizer::Mixtral'
12+
SQL
13+
14+
execute <<~SQL
15+
UPDATE
16+
embedding_definitions
17+
SET
18+
tokenizer_class = 'DiscourseAi::Tokenizer::Mistral'
19+
WHERE
20+
tokenizer_class = 'DiscourseAi::Tokenizer::Mistral'
21+
SQL
22+
end
23+
24+
def down
25+
execute <<~SQL
26+
UPDATE
27+
llm_models
28+
SET
29+
tokenizer = 'DiscourseAi::Tokenizer::Mixtral'
30+
WHERE
31+
tokenizer = 'DiscourseAi::Tokenizer::Mistral'
32+
SQL
33+
34+
execute <<~SQL
35+
UPDATE
36+
embedding_definitions
37+
SET
38+
tokenizer_class = 'DiscourseAi::Tokenizer::Mixtral'
39+
WHERE
40+
tokenizer_class = 'DiscourseAi::Tokenizer::Mistral'
41+
SQL
42+
end
43+
end

0 commit comments

Comments
 (0)