codeflash-admin/codeflash-internal
1
0
Fork 0
mirror of https://github.com/codeflash-ai/codeflash-internal.git synced 2026-05-04 18:25:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

ready to merge

Browse source
This commit is contained in:
aseembits93 2025-05-09 19:42:20 -07:00
parent ee805d6008
commit 7cd768e0e7
2 changed files with 1 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
django/aiservice/testgen/testgen.py
View file

@ -365,8 +365,6 @@ async def testgen(
return 500, TestGenErrorResponseSchema(error="Error generating tests. Internal server error.")
if hasattr(request, "should_log_features") and request.should_log_features:
# TODO: Update log features with perf instrumented tests
generated_test_source = "# TRY41 GENERATED WITH SINGLE PROMPT\n" + generated_test_source
instrumented_behavior_test_source = "# TRY41 GENERATED WITH SINGLE PROMPT\n" + instrumented_behavior_test_source
await log_features(
trace_id=data.trace_id,
user_id=request.user,

2
experiments/testgen_analysis.py
View file

@ -15,7 +15,7 @@ def load_data(
with engine.connect() as connection:
query = """
SELECT * FROM optimization_features WHERE
user_id = 'github|4725571' AND EXISTS (SELECT 1 FROM unnest(generated_test) AS elem WHERE elem LIKE '%%# TRY21 GENERATED WITH SINGLE PROMPT%%')
user_id = 'github|4725571' AND EXISTS (SELECT 1 FROM unnest(generated_test) AS elem WHERE elem LIKE '%%# TRY41 GENERATED WITH SINGLE PROMPT%%')
"""
return pd.read_sql_query(
query, connection