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

update evals

Browse Source
This commit is contained in:
Abi Raja 2024-02-21 09:30:42 -05:00
parent 34c1f9a959
commit c0e084aa86
2 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
backend/evals/core.py
View File

@ -3,7 +3,6 @@ import os
from llm import stream_openai_response from llm import stream_openai_response
from prompts import assemble_prompt from prompts import assemble_prompt
from prompts.types import Stack from prompts.types import Stack
from utils import pprint_prompt
async def generate_code_core(image_url: str, stack: Stack) -> str: async def generate_code_core(image_url: str, stack: Stack) -> str:
@ -11,8 +10,6 @@ async def generate_code_core(image_url: str, stack: Stack) -> str:
openai_api_key = os.environ.get("OPENAI_API_KEY") openai_api_key = os.environ.get("OPENAI_API_KEY")
openai_base_url = None openai_base_url = None
pprint_prompt(prompt_messages)
async def process_chunk(content: str): async def process_chunk(content: str):
pass pass

2
backend/run_evals.py
View File

@ -23,7 +23,7 @@ async def main():
for filename in evals: for filename in evals:
filepath = os.path.join(INPUT_DIR, filename) filepath = os.path.join(INPUT_DIR, filename)
data_url = await image_to_data_url(filepath) data_url = await image_to_data_url(filepath)
task = generate_code_core(data_url, "vue_tailwind") task = generate_code_core(data_url, "html_tailwind")
tasks.append(task) tasks.append(task)
results = await asyncio.gather(*tasks) results = await asyncio.gather(*tasks)