feat: DIA-1802: Anthropic support in Prompts

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,
+            ),
+        ),
+    ]

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')
 
 

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,
+            ),
+        ),
+    ]

pyproject.toml

@@ -208,7 +208,7 @@ django-migration-linter = "^5.1.0"
 setuptools = ">=75.4.0"
 
 # Humansignal repo dependencies
-label-studio-sdk = {url = "https://github.com/HumanSignal/label-studio-sdk/archive/09995cf0c72398322af949bb13de034a7bbb785f.zip"}
+label-studio-sdk = {url = "https://github.com/HumanSignal/label-studio-sdk/archive/5352eb4b3147669b638e34e9efabc85ee62d7fdc.zip"}
 
 [tool.poetry.group.test.dependencies]
 pytest = "7.2.2"