diff --git a/label_studio/ml_model_providers/migrations/0008_alter_modelproviderconnection_provider.py b/label_studio/ml_model_providers/migrations/0008_alter_modelproviderconnection_provider.py new file mode 100644 index 000000000000..7cbb55b158c6 --- /dev/null +++ b/label_studio/ml_model_providers/migrations/0008_alter_modelproviderconnection_provider.py @@ -0,0 +1,29 @@ +# Generated by Django 5.1.4 on 2025-01-10 20:15 + +from django.db import migrations, models + + +class Migration(migrations.Migration): + + dependencies = [ + ("ml_model_providers", "0007_alter_modelproviderconnection_provider"), + ] + + operations = [ + migrations.AlterField( + model_name="modelproviderconnection", + name="provider", + field=models.CharField( + choices=[ + ("OpenAI", "OpenAI"), + ("AzureOpenAI", "AzureOpenAI"), + ("VertexAI", "VertexAI"), + ("Gemini", "Gemini"), + ("Anthropic", "Anthropic"), + ("Custom", "Custom"), + ], + default="OpenAI", + max_length=255, + ), + ), + ] diff --git a/label_studio/ml_model_providers/models.py b/label_studio/ml_model_providers/models.py index 3ee837da0c03..72b6cdd8ce2b 100644 --- a/label_studio/ml_model_providers/models.py +++ b/label_studio/ml_model_providers/models.py @@ -13,6 +13,7 @@ class ModelProviders(models.TextChoices): AZURE_OPENAI = 'AzureOpenAI', _('AzureOpenAI') VERTEX_AI = 'VertexAI', _('VertexAI') GEMINI = 'Gemini', _('Gemini') + ANTHROPIC = 'Anthropic', _('Anthropic') CUSTOM = 'Custom', _('Custom') diff --git a/label_studio/ml_models/migrations/0015_alter_thirdpartymodelversion_provider.py b/label_studio/ml_models/migrations/0015_alter_thirdpartymodelversion_provider.py new file mode 100644 index 000000000000..cc1f678ec170 --- /dev/null +++ b/label_studio/ml_models/migrations/0015_alter_thirdpartymodelversion_provider.py @@ -0,0 +1,30 @@ +# Generated by Django 5.1.4 on 2025-01-10 20:15 + +from django.db import migrations, models + + +class Migration(migrations.Migration): + + dependencies = [ + ("ml_models", "0014_alter_thirdpartymodelversion_provider"), + ] + + operations = [ + migrations.AlterField( + model_name="thirdpartymodelversion", + name="provider", + field=models.CharField( + choices=[ + ("OpenAI", "OpenAI"), + ("AzureOpenAI", "AzureOpenAI"), + ("VertexAI", "VertexAI"), + ("Gemini", "Gemini"), + ("Anthropic", "Anthropic"), + ("Custom", "Custom"), + ], + default="OpenAI", + help_text="The model provider to use e.g. OpenAI", + max_length=255, + ), + ), + ]