ciyidogan commited on
Commit
4760d1c
·
verified ·
1 Parent(s): ae0c665

Update admin_routes.py

Browse files
Files changed (1) hide show
  1. admin_routes.py +4 -1
admin_routes.py CHANGED
@@ -65,6 +65,7 @@ class EnvironmentUpdate(BaseModel):
65
  work_mode: str
66
  cloud_token: Optional[str] = None
67
  spark_endpoint: str
 
68
 
69
  class ProjectCreate(BaseModel):
70
  name: str
@@ -291,7 +292,8 @@ async def get_environment(username: str = Depends(verify_token)):
291
  return {
292
  "work_mode": env_config.get("work_mode", "on-premise"),
293
  "cloud_token": env_config.get("cloud_token", ""),
294
- "spark_endpoint": env_config.get("spark_endpoint", "http://localhost:7861")
 
295
  }
296
 
297
  @router.put("/environment")
@@ -306,6 +308,7 @@ async def update_environment(
306
  config["config"]["work_mode"] = update.work_mode
307
  config["config"]["cloud_token"] = update.cloud_token or ""
308
  config["config"]["spark_endpoint"] = update.spark_endpoint
 
309
  config["config"]["last_update_date"] = get_timestamp()
310
  config["config"]["last_update_user"] = username
311
 
 
65
  work_mode: str
66
  cloud_token: Optional[str] = None
67
  spark_endpoint: str
68
+ internal_prompt: Optional[str] = None
69
 
70
  class ProjectCreate(BaseModel):
71
  name: str
 
292
  return {
293
  "work_mode": env_config.get("work_mode", "on-premise"),
294
  "cloud_token": env_config.get("cloud_token", ""),
295
+ "spark_endpoint": env_config.get("spark_endpoint", "http://localhost:7861"),
296
+ "internal_prompt": env_config.get("internal_prompt", "")
297
  }
298
 
299
  @router.put("/environment")
 
308
  config["config"]["work_mode"] = update.work_mode
309
  config["config"]["cloud_token"] = update.cloud_token or ""
310
  config["config"]["spark_endpoint"] = update.spark_endpoint
311
+ config["config"]["internal_prompt"] = update.internal_prompt or "" # Yeni alan
312
  config["config"]["last_update_date"] = get_timestamp()
313
  config["config"]["last_update_user"] = username
314