Spaces:
Paused
Paused
Update config_provider.py
Browse files- config_provider.py +13 -1
config_provider.py
CHANGED
|
@@ -13,12 +13,24 @@ from pydantic import BaseModel, Field, HttpUrl, ValidationError
|
|
| 13 |
from encryption_utils import decrypt
|
| 14 |
|
| 15 |
class GlobalConfig(BaseModel):
|
| 16 |
-
work_mode: str = Field("hfcloud", pattern=r"^(hfcloud|cloud|on-premise)$")
|
| 17 |
cloud_token: Optional[str] = None
|
| 18 |
spark_endpoint: HttpUrl
|
| 19 |
internal_prompt: Optional[str] = None # Yeni alan
|
| 20 |
users: List["UserConfig"] = []
|
| 21 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 22 |
def get_plain_token(self) -> Optional[str]:
|
| 23 |
if self.cloud_token:
|
| 24 |
# Lazy import to avoid circular dependency
|
|
|
|
| 13 |
from encryption_utils import decrypt
|
| 14 |
|
| 15 |
class GlobalConfig(BaseModel):
|
| 16 |
+
work_mode: str = Field("hfcloud", pattern=r"^(hfcloud|cloud|on-premise|gpt4o|gpt4o-mini)$")
|
| 17 |
cloud_token: Optional[str] = None
|
| 18 |
spark_endpoint: HttpUrl
|
| 19 |
internal_prompt: Optional[str] = None # Yeni alan
|
| 20 |
users: List["UserConfig"] = []
|
| 21 |
|
| 22 |
+
def is_gpt_mode(self) -> bool:
|
| 23 |
+
"""Check if running in GPT mode (any variant)"""
|
| 24 |
+
return self.work_mode in ("gpt4o", "gpt4o-mini")
|
| 25 |
+
|
| 26 |
+
def get_gpt_model(self) -> str:
|
| 27 |
+
"""Get the GPT model name for OpenAI API"""
|
| 28 |
+
if self.work_mode == "gpt4o":
|
| 29 |
+
return "gpt-4o"
|
| 30 |
+
elif self.work_mode == "gpt4o-mini":
|
| 31 |
+
return "gpt-4o-mini"
|
| 32 |
+
return None
|
| 33 |
+
|
| 34 |
def get_plain_token(self) -> Optional[str]:
|
| 35 |
if self.cloud_token:
|
| 36 |
# Lazy import to avoid circular dependency
|