File tree Expand file tree Collapse file tree 1 file changed +4
-0
lines changed Expand file tree Collapse file tree 1 file changed +4
-0
lines changed Original file line number Diff line number Diff line change @@ -525,6 +525,7 @@ class EmbeddingPlatform(Enum):
525525 HUGGINGFACE = "huggingface"
526526 OPENAI = "openai"
527527 AZURE = "azure"
528+ PRIVATEMODE_AI = "privatemode-ai"
528529
529530
530531class SampleProjectType (Enum ):
@@ -669,6 +670,7 @@ class LLMProvider(Enum):
669670 OPENAI = "Open AI"
670671 AZURE = "Azure"
671672 AZURE_FOUNDRY = "Azure Foundry"
673+ PRIVATEMODE_AI = "Privatemode AI"
672674
673675 @staticmethod
674676 def from_string (value : str ):
@@ -679,6 +681,8 @@ def from_string(value: str):
679681 return LLMProvider .AZURE
680682 elif changed_value == "AZURE_FOUNDRY" :
681683 return LLMProvider .AZURE_FOUNDRY
684+ elif changed_value == "PRIVATEMODE_AI" :
685+ return LLMProvider .PRIVATEMODE_AI
682686 raise ValueError ("Could not parse LLMProvider from string" )
683687
684688 def as_key (self ):
You can’t perform that action at this time.
0 commit comments