ciyidogan commited on
Commit
536e6ec
·
verified ·
1 Parent(s): 27c784e

Update config_provider.py

Browse files
Files changed (1) hide show
  1. 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