feat: DIA-1802: Anthropic support in Prompts (#407)

Co-authored-by: fern-api <115122769+fern-api[bot]@users.noreply.github.com> Co-authored-by: Nikita Belonogov <[email protected]> Co-authored-by: Matt Bernstein <[email protected]>
HumanSignal · Feb 6, 2025 · c539a97 · c539a97
1 parent 7328142
commit c539a97
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/.mock/definition/__package__.yml b/.mock/definition/__package__.yml
@@ -2342,6 +2342,7 @@ types:
       - AzureOpenAI
       - VertexAI
       - Gemini
+      - Anthropic
       - Custom
     inline: true
     source:
@@ -2550,6 +2551,7 @@ types:
       - AzureOpenAI
       - VertexAI
       - Gemini
+      - Anthropic
       - Custom
     inline: true
     source:

diff --git a/src/label_studio_sdk/types/model_provider_connection_provider.py b/src/label_studio_sdk/types/model_provider_connection_provider.py
@@ -3,5 +3,5 @@
 import typing
 
 ModelProviderConnectionProvider = typing.Union[
-    typing.Literal["OpenAI", "AzureOpenAI", "VertexAI", "Gemini", "Custom"], typing.Any
+    typing.Literal["OpenAI", "AzureOpenAI", "VertexAI", "Gemini", "Anthropic", "Custom"], typing.Any
 ]
diff --git a/src/label_studio_sdk/types/prompt_version_provider.py b/src/label_studio_sdk/types/prompt_version_provider.py
@@ -3,5 +3,5 @@
 import typing
 
 PromptVersionProvider = typing.Union[
-    typing.Literal["OpenAI", "AzureOpenAI", "VertexAI", "Gemini", "Custom"], typing.Any
+    typing.Literal["OpenAI", "AzureOpenAI", "VertexAI", "Gemini", "Anthropic", "Custom"], typing.Any
 ]