fix: migration

This commit is contained in:
Timothy J. Baek 2024-05-24 01:43:04 -07:00
parent 110ed67468
commit 6dbd59c7c6

View File

@ -60,7 +60,7 @@ def migrate_modelfile_to_model(migrator: Migrator, database: pw.Database):
"ollama": {"modelfile": modelfile.modelfile.get("content")}, "ollama": {"modelfile": modelfile.modelfile.get("content")},
"suggestion_prompts": modelfile.modelfile.get("suggestionPrompts"), "suggestion_prompts": modelfile.modelfile.get("suggestionPrompts"),
"categories": modelfile.modelfile.get("categories"), "categories": modelfile.modelfile.get("categories"),
"user": {**modelfile.modelfile.get("user", {}), "community": "true"}, "user": {**modelfile.modelfile.get("user", {}), "community": True},
} }
) )