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

improve mock functionality

Browse Source
This commit is contained in:
Abi Raja 2024-03-08 05:02:23 -05:00
parent fa58f2ce8f
commit a0f5af0fdc
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
backend/mock_llm.py
View File

@ -4,6 +4,9 @@ from typing import Awaitable, Callable
from custom_types import InputMode from custom_types import InputMode
STREAM_CHUNK_SIZE = 100
async def mock_completion( async def mock_completion(
process_chunk: Callable[[str], Awaitable[None]], input_mode: InputMode process_chunk: Callable[[str], Awaitable[None]], input_mode: InputMode
) -> str: ) -> str:
@ -13,8 +16,8 @@ async def mock_completion(
else NO_IMAGES_NYTIMES_MOCK_CODE else NO_IMAGES_NYTIMES_MOCK_CODE
) )
for i in range(0, len(code_to_return), 100): for i in range(0, len(code_to_return), STREAM_CHUNK_SIZE):
await process_chunk(code_to_return[i : i + 100]) await process_chunk(code_to_return[i : i + STREAM_CHUNK_SIZE])
await asyncio.sleep(0.01) await asyncio.sleep(0.01)
return code_to_return return code_to_return