Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat: DIA-1802: Anthropic support in Prompts #6891

Open
wants to merge 9 commits into
base: develop
Choose a base branch
from
+223 −0
Open

feat: DIA-1802: Anthropic support in Prompts #6891

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
1487640
feat: DIA-1715: VertexAI Gemini model support
Jan 8, 2025
46f7168
Merge branch 'develop' of github.com:heartexlabs/label-studio into fb…
Jan 8, 2025
773931c
Merge remote-tracking branch 'origin/develop' into fb-dia-1715
matt-bernstein Jan 9, 2025
03dc1ed
Merge branch 'develop' of github.com:heartexlabs/label-studio into fb…
Jan 9, 2025
f17eda8
Merge branch 'fb-dia-1715' of github.com:heartexlabs/label-studio int…
Jan 9, 2025
b0cbca0
feat: DIA-1716: Google AI Studio support in Prompts
matt-bernstein Jan 10, 2025
1baff06
add gemini to enum
matt-bernstein Jan 10, 2025
fbec9b6
feat: DIA-1802: Anthropic support in Prompts
matt-bernstein Jan 10, 2025
a349ae6
update enum
matt-bernstein Jan 10, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
62 changes: 62 additions & 0 deletions ...viders/migrations/0006_modelproviderconnection_google_application_credentials_and_more.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,62 @@
# Generated by Django 5.1.4 on 2025-01-03 20:58

from django.db import migrations, models


class Migration(migrations.Migration):

dependencies = [
(
"ml_model_providers",
"0005_modelproviderconnection_budget_alert_threshold_and_more",
),
]

operations = [
migrations.AddField(
model_name="modelproviderconnection",
name="google_application_credentials",
field=models.TextField(
blank=True,
help_text="The content of GOOGLE_APPLICATION_CREDENTIALS json file",
null=True,
verbose_name="google application credentials",
),
),
migrations.AddField(
model_name="modelproviderconnection",
name="google_location",
field=models.CharField(
blank=True,
help_text="Google project location",
max_length=255,
null=True,
verbose_name="google location",
),
),
migrations.AddField(
model_name="modelproviderconnection",
name="google_project_id",
field=models.CharField(
blank=True,
help_text="Google project ID",
max_length=255,
null=True,
verbose_name="google project id",
),
),
migrations.AlterField(
model_name="modelproviderconnection",
name="provider",
field=models.CharField(
choices=[
("OpenAI", "OpenAI"),
("AzureOpenAI", "AzureOpenAI"),
("VertexAI", "VertexAI"),
("Custom", "Custom"),
],
default="OpenAI",
max_length=255,
),
),
]
31 changes: 31 additions & 0 deletions label_studio/ml_model_providers/migrations/0007_alter_modelproviderconnection_provider.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,31 @@
# Generated by Django 5.1.4 on 2025-01-10 18:38

from django.db import migrations, models


class Migration(migrations.Migration):

dependencies = [
(
"ml_model_providers",
"0006_modelproviderconnection_google_application_credentials_and_more",
),
]

operations = [
migrations.AlterField(
model_name="modelproviderconnection",
name="provider",
field=models.CharField(
choices=[
("OpenAI", "OpenAI"),
("AzureOpenAI", "AzureOpenAI"),
("VertexAI", "VertexAI"),
("Gemini", "Gemini"),
("Custom", "Custom"),
],
default="OpenAI",
max_length=255,
),
),
]
29 changes: 29 additions & 0 deletions label_studio/ml_model_providers/migrations/0008_alter_modelproviderconnection_provider.py
View file
Open in desktop
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,
),
),
]
14 changes: 14 additions & 0 deletions label_studio/ml_model_providers/models.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,9 @@
class ModelProviders(models.TextChoices):
OPENAI = 'OpenAI', _('OpenAI')
AZURE_OPENAI = 'AzureOpenAI', _('AzureOpenAI')
VERTEX_AI = 'VertexAI', _('VertexAI')
GEMINI = 'Gemini', _('Gemini')
ANTHROPIC = 'Anthropic', _('Anthropic')
CUSTOM = 'Custom', _('Custom')


Expand All @@ -32,6 +35,17 @@ class ModelProviderConnection(models.Model):

endpoint = models.CharField(max_length=512, null=True, blank=True, help_text='Azure OpenAI endpoint')

google_application_credentials = models.TextField(
_('google application credentials'),
null=True,
blank=True,
help_text='The content of GOOGLE_APPLICATION_CREDENTIALS json file',
)

google_project_id = models.CharField(_('google project id'), max_length=255, null=True, blank=True, help_text='Google project ID')

google_location = models.CharField(_('google location'), max_length=255, null=True, blank=True, help_text='Google project location')

cached_available_models = models.CharField(
max_length=4096, null=True, blank=True, help_text='List of available models from the provider'
)
Expand Down
28 changes: 28 additions & 0 deletions label_studio/ml_models/migrations/0013_alter_thirdpartymodelversion_provider.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
# Generated by Django 5.1.4 on 2025-01-03 20:58

from django.db import migrations, models


class Migration(migrations.Migration):

dependencies = [
("ml_models", "0012_alter_thirdpartymodelversion_provider"),
]

operations = [
migrations.AlterField(
model_name="thirdpartymodelversion",
name="provider",
field=models.CharField(
choices=[
("OpenAI", "OpenAI"),
("AzureOpenAI", "AzureOpenAI"),
("VertexAI", "VertexAI"),
("Custom", "Custom"),
],
default="OpenAI",
help_text="The model provider to use e.g. OpenAI",
max_length=255,
),
),
]
29 changes: 29 additions & 0 deletions label_studio/ml_models/migrations/0014_alter_thirdpartymodelversion_provider.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@
# Generated by Django 5.1.4 on 2025-01-10 18:38

from django.db import migrations, models


class Migration(migrations.Migration):

dependencies = [
("ml_models", "0013_alter_thirdpartymodelversion_provider"),
]

operations = [
migrations.AlterField(
model_name="thirdpartymodelversion",
name="provider",
field=models.CharField(
choices=[
("OpenAI", "OpenAI"),
("AzureOpenAI", "AzureOpenAI"),
("VertexAI", "VertexAI"),
("Gemini", "Gemini"),
("Custom", "Custom"),
],
default="OpenAI",
help_text="The model provider to use e.g. OpenAI",
max_length=255,
),
),
]
30 changes: 30 additions & 0 deletions label_studio/ml_models/migrations/0015_alter_thirdpartymodelversion_provider.py
View file
Open in desktop
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,
),
),
]
Loading