Spaces:
Building
Building
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
|