Simon Strandgaard commited on
Commit
5cdccba
·
1 Parent(s): cdf59a1

Prevent the user from clicking Submit/Retry, without an api key

Browse files
Files changed (1) hide show
  1. src/plan/app_text2plan.py +8 -0
src/plan/app_text2plan.py CHANGED
@@ -38,6 +38,7 @@ class Config:
38
  visible_open_output_dir_button: bool
39
  visible_openrouter_api_key_textbox: bool
40
  allow_only_openrouter_models: bool
 
41
  enable_purge_old_runs: bool
42
 
43
  CONFIG_LOCAL = Config(
@@ -46,6 +47,7 @@ CONFIG_LOCAL = Config(
46
  visible_open_output_dir_button=True,
47
  visible_openrouter_api_key_textbox=False,
48
  allow_only_openrouter_models=False,
 
49
  enable_purge_old_runs=False,
50
  )
51
  CONFIG_HUGGINGFACE_SPACES = Config(
@@ -54,6 +56,7 @@ CONFIG_HUGGINGFACE_SPACES = Config(
54
  visible_open_output_dir_button=False,
55
  visible_openrouter_api_key_textbox=True,
56
  allow_only_openrouter_models=True,
 
57
  enable_purge_old_runs=True,
58
  )
59
 
@@ -202,6 +205,11 @@ def run_planner(submit_or_retry_button, plan_prompt, browser_state, session_stat
202
  session_state.llm_model = settings.get("model_radio", session_state.llm_model)
203
  session_state.speedvsdetail = settings.get("speedvsdetail_radio", session_state.speedvsdetail)
204
 
 
 
 
 
 
205
  # Clear any previous stop signal.
206
  session_state.stop_event.clear()
207
 
 
38
  visible_open_output_dir_button: bool
39
  visible_openrouter_api_key_textbox: bool
40
  allow_only_openrouter_models: bool
41
+ run_planner_check_api_key_is_provided: bool
42
  enable_purge_old_runs: bool
43
 
44
  CONFIG_LOCAL = Config(
 
47
  visible_open_output_dir_button=True,
48
  visible_openrouter_api_key_textbox=False,
49
  allow_only_openrouter_models=False,
50
+ run_planner_check_api_key_is_provided=False,
51
  enable_purge_old_runs=False,
52
  )
53
  CONFIG_HUGGINGFACE_SPACES = Config(
 
56
  visible_open_output_dir_button=False,
57
  visible_openrouter_api_key_textbox=True,
58
  allow_only_openrouter_models=True,
59
+ run_planner_check_api_key_is_provided=True,
60
  enable_purge_old_runs=True,
61
  )
62
 
 
205
  session_state.llm_model = settings.get("model_radio", session_state.llm_model)
206
  session_state.speedvsdetail = settings.get("speedvsdetail_radio", session_state.speedvsdetail)
207
 
208
+ # Check if an OpenRouter API key is required and provided.
209
+ if CONFIG.run_planner_check_api_key_is_provided:
210
+ if session_state.openrouter_api_key is None or len(session_state.openrouter_api_key) == 0:
211
+ raise ValueError("An OpenRouter API key is required to use PlanExe. Please provide an API key in the Settings tab.")
212
+
213
  # Clear any previous stop signal.
214
  session_state.stop_event.clear()
215