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-07-31 11:22:25 -04:00
parent c61a2ac772
commit 637c1b4c1d
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
backend/routes/generate_code.py
View File

@ -187,9 +187,7 @@ async def stream_code(websocket: WebSocket):
await send_message("status", "Generating code...", 0) await send_message("status", "Generating code...", 0)
await send_message("status", "Generating code...", 1) await send_message("status", "Generating code...", 1)
# TODO(*): Move down ### Prompt creation
async def process_chunk(content: str, variantIndex: int):
await send_message("chunk", content, variantIndex)
# Image cache for updates so that we don't have to regenerate images # Image cache for updates so that we don't have to regenerate images
image_cache: Dict[str, str] = {} image_cache: Dict[str, str] = {}
@ -206,6 +204,11 @@ async def stream_code(websocket: WebSocket):
pprint_prompt(prompt_messages) # type: ignore pprint_prompt(prompt_messages) # type: ignore
### Code generation
async def process_chunk(content: str, variantIndex: int):
await send_message("chunk", content, variantIndex)
if SHOULD_MOCK_AI_RESPONSE: if SHOULD_MOCK_AI_RESPONSE:
completions = [ completions = [
await mock_completion(process_chunk, input_mode=validated_input_mode) await mock_completion(process_chunk, input_mode=validated_input_mode)