attilasimko commited on
Commit
f46de5a
·
1 Parent(s): 338175b

Initial check

Browse files
Files changed (1) hide show
  1. evaluations/pitfalls.py +4 -2
evaluations/pitfalls.py CHANGED
@@ -7,5 +7,7 @@ def evaluate(verbose, llm, zip, readme):
7
 
8
  if (llm):
9
  for code in codebase:
10
- log(verbose, "LOG", code)
11
- log(verbose, "LOG", llm.predict("PITFALL", f"{code}\n Can you find any signs of common pitfalls in this code?"))
 
 
 
7
 
8
  if (llm):
9
  for code in codebase:
10
+ pitfall_check = llm.predict("STRICT", f"{codebase[code]}Do you find any signs of serious issues in this code?")
11
+ if (("Yes" in pitfall_check) & ("No" not in pitfall_check)):
12
+ log(verbose, "ERROR", f"Found possible issues in {code}")
13
+ log(verbose, "LOG", llm.predict("PITFALL", f"File name {code} file {codebase[code]}\n Can you find any signs of common pitfalls in this code?"))