Skip to content

Commit a169a62

Browse files
authored
prototype for privatemode ai (#177)
* Enum for privatemode * small name change * Adds embedding platform
1 parent b41145a commit a169a62

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

enums.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff 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

530531
class 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):

0 commit comments

Comments
 (0)