Skip to content

Commit

Permalink
update enum
Browse files Browse the repository at this point in the history
  • Loading branch information
matt-bernstein committed Jan 10, 2025
1 parent fbec9b6 commit a349ae6
Show file tree
Hide file tree
Showing 3 changed files with 60 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -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,
),
),
]
1 change: 1 addition & 0 deletions label_studio/ml_model_providers/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -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')


Expand Down
Original file line number Diff line number Diff line change
@@ -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,
),
),
]

0 comments on commit a349ae6

Please sign in to comment.