feat: DIA-1868: Azure AI Studio support in Prompts (#7040)

Co-authored-by: fern-api <115122769+fern-api[bot]@users.noreply.github.com> Co-authored-by: Matt Bernstein <[email protected]> Co-authored-by: matt-bernstein <[email protected]>
HumanSignal · Feb 26, 2025 · a3af28e · a3af28e
1 parent 05bde58
commit a3af28e
Show file tree

Hide file tree

Showing 5 changed files with 67 additions and 4 deletions.
diff --git a/label_studio/ml_model_providers/migrations/0009_alter_modelproviderconnection_provider.py b/label_studio/ml_model_providers/migrations/0009_alter_modelproviderconnection_provider.py
@@ -0,0 +1,30 @@
+# Generated by Django 5.1.5 on 2025-02-07 16:47
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("ml_model_providers", "0008_alter_modelproviderconnection_provider"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="modelproviderconnection",
+            name="provider",
+            field=models.CharField(
+                choices=[
+                    ("OpenAI", "OpenAI"),
+                    ("AzureOpenAI", "AzureOpenAI"),
+                    ("AzureAIFoundry", "AzureAIFoundry"),
+                    ("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
@@ -11,6 +11,7 @@
 class ModelProviders(models.TextChoices):
     OPENAI = 'OpenAI', _('OpenAI')
     AZURE_OPENAI = 'AzureOpenAI', _('AzureOpenAI')
+    AZURE_AI_FOUNDRY = 'AzureAIFoundry', _('AzureAIFoundry')
     VERTEX_AI = 'VertexAI', _('VertexAI')
     GEMINI = 'Gemini', _('Gemini')
     ANTHROPIC = 'Anthropic', _('Anthropic')

diff --git a/label_studio/ml_models/migrations/0016_alter_thirdpartymodelversion_provider.py b/label_studio/ml_models/migrations/0016_alter_thirdpartymodelversion_provider.py
@@ -0,0 +1,31 @@
+# Generated by Django 5.1.5 on 2025-02-07 16:47
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("ml_models", "0015_alter_thirdpartymodelversion_provider"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="thirdpartymodelversion",
+            name="provider",
+            field=models.CharField(
+                choices=[
+                    ("OpenAI", "OpenAI"),
+                    ("AzureOpenAI", "AzureOpenAI"),
+                    ("AzureAIFoundry", "AzureAIFoundry"),
+                    ("VertexAI", "VertexAI"),
+                    ("Gemini", "Gemini"),
+                    ("Anthropic", "Anthropic"),
+                    ("Custom", "Custom"),
+                ],
+                default="OpenAI",
+                help_text="The model provider to use e.g. OpenAI",
+                max_length=255,
+            ),
+        ),
+    ]
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -209,7 +209,7 @@ djangorestframework-simplejwt = {extras = ["crypto"], version = "^5.4.0"}
 tldextract = ">=5.1.3"
 
 # Humansignal repo dependencies
-label-studio-sdk = {url = "https://github.com/HumanSignal/label-studio-sdk/archive/d953881b2b4567b9381883e35ebf378f9526a55e.zip"}
+label-studio-sdk = {url = "https://github.com/HumanSignal/label-studio-sdk/archive/324323c4e50581f25d4aedc7c42922e01a330d7c.zip"}
 
 [tool.poetry.group.test.dependencies]
 pytest = "7.2.2"