engineering/opencompletion.com
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

make count_tokens public on Message

Browse source 
modified:   app.py
	modified:   migrations/versions/190d5ef26e20_add_token_count_to_message.py
This commit is contained in:
Russell Ballestrini 2023-12-07 10:16:01 -05:00
parent caf4c76ceb
commit c2aaac9768
2 changed files with 9 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

5
migrations/versions/190d5ef26e20_add_token_count_to_message.py
View file

@ -30,16 +30,13 @@ def upgrade():
# Assuming the content column is not nullable and always has a value
messages = session.query(Message).all()
for message in messages:
message.token_count = len(message.content.split())
message.count_tokens()
session.add(message)
session.commit()
# ### end Alembic commands ###
def downgrade():
# ### commands auto generated by Alembic - please adjust! ###
with op.batch_alter_table('message', schema=None) as batch_op:
batch_op.drop_column('token_count')
# ### end Alembic commands ###