VibecoderMcSwaggins commited on
Commit
da82e7e
·
1 Parent(s): b16e7a5

fix(ui): apply CodeRabbit review suggestions

Browse files

- Use ConfigurationError instead of ValueError for invalid API key
- Add env var override for server_name with nosec B104 comment

Files changed (1) hide show
  1. src/app.py +3 -2
src/app.py CHANGED
@@ -17,6 +17,7 @@ from src.tools.europepmc import EuropePMCTool
17
  from src.tools.pubmed import PubMedTool
18
  from src.tools.search_handler import SearchHandler
19
  from src.utils.config import settings
 
20
  from src.utils.models import OrchestratorConfig
21
 
22
 
@@ -72,7 +73,7 @@ def configure_orchestrator(
72
  model = OpenAIModel(settings.openai_model, provider=openai_provider)
73
  backend_info = "Paid API (OpenAI)"
74
  else:
75
- raise ValueError(
76
  "Invalid API key format. Expected sk-... (OpenAI) or sk-ant-... (Anthropic)"
77
  )
78
  judge_handler = JudgeHandler(model=model)
@@ -245,7 +246,7 @@ def main() -> None:
245
  """Run the Gradio app with MCP server enabled."""
246
  demo, _ = create_demo()
247
  demo.launch(
248
- server_name="0.0.0.0",
249
  server_port=7860,
250
  share=False,
251
  mcp_server=True,
 
17
  from src.tools.pubmed import PubMedTool
18
  from src.tools.search_handler import SearchHandler
19
  from src.utils.config import settings
20
+ from src.utils.exceptions import ConfigurationError
21
  from src.utils.models import OrchestratorConfig
22
 
23
 
 
73
  model = OpenAIModel(settings.openai_model, provider=openai_provider)
74
  backend_info = "Paid API (OpenAI)"
75
  else:
76
+ raise ConfigurationError(
77
  "Invalid API key format. Expected sk-... (OpenAI) or sk-ant-... (Anthropic)"
78
  )
79
  judge_handler = JudgeHandler(model=model)
 
246
  """Run the Gradio app with MCP server enabled."""
247
  demo, _ = create_demo()
248
  demo.launch(
249
+ server_name=os.getenv("GRADIO_SERVER_NAME", "0.0.0.0"), # nosec B104
250
  server_port=7860,
251
  share=False,
252
  mcp_server=True,