From c0e084aa86b2fced5d3f78c3351bdf38d0d84b54 Mon Sep 17 00:00:00 2001 From: Abi Raja Date: Wed, 21 Feb 2024 09:30:42 -0500 Subject: [PATCH] update evals --- backend/evals/core.py | 3 --- backend/run_evals.py | 2 +- 2 files changed, 1 insertion(+), 4 deletions(-) diff --git a/backend/evals/core.py b/backend/evals/core.py index 61db1a3..ba00c40 100644 --- a/backend/evals/core.py +++ b/backend/evals/core.py @@ -3,7 +3,6 @@ import os from llm import stream_openai_response from prompts import assemble_prompt from prompts.types import Stack -from utils import pprint_prompt 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_base_url = None - pprint_prompt(prompt_messages) - async def process_chunk(content: str): pass diff --git a/backend/run_evals.py b/backend/run_evals.py index ddb7eaa..75e76eb 100644 --- a/backend/run_evals.py +++ b/backend/run_evals.py @@ -23,7 +23,7 @@ async def main(): for filename in evals: filepath = os.path.join(INPUT_DIR, filename) 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) results = await asyncio.gather(*tasks)