Github/screenshot-to-code
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix TODO

Browse Source
This commit is contained in:
Abi Raja 2024-08-30 14:07:30 +02:00
parent 67e7066221
commit 5d33dd6bf3
1 changed files with 1 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
backend/routes/generate_code.py
View File

@ -221,13 +221,6 @@ async def stream_code(websocket: WebSocket):
f"Generating {stack} code in {input_mode} mode using {code_generation_model}..." f"Generating {stack} code in {input_mode} mode using {code_generation_model}..."
) )
# TODO(*): Do I still need this?
if not openai_api_key and is_openai_model(code_generation_model):
await throw_error(
"No OpenAI API key found. Please add your API key in the settings dialog or add it to backend/.env file. If you add it to .env, make sure to restart the backend server."
)
return
for i in range(NUM_VARIANTS): for i in range(NUM_VARIANTS):
await send_message("status", "Generating code...", i) await send_message("status", "Generating code...", i)
@ -285,7 +278,7 @@ async def stream_code(websocket: WebSocket):
variant_models = ["anthropic", "anthropic"] variant_models = ["anthropic", "anthropic"]
else: else:
await throw_error( await throw_error(
"No OpenAI or Anthropic API key found. Please add the environment variable OPENAI_API_KEY or ANTHROPIC_API_KEY to backend/.env or in the settings dialog" "No OpenAI or Anthropic API key found. Please add the environment variable OPENAI_API_KEY or ANTHROPIC_API_KEY to backend/.env or in the settings dialog. If you add it to .env, make sure to restart the backend server."
) )
raise Exception("No OpenAI or Anthropic key") raise Exception("No OpenAI or Anthropic key")