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
- 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
|
| 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,
|