File size: 2,198 Bytes
692bd71 55b70aa c21f3cc 55b70aa c21f3cc 692bd71 c21f3cc 692bd71 55b70aa 692bd71 dcb6098 c21f3cc 692bd71 c21f3cc 692bd71 55b70aa 692bd71 62c8baf 692bd71 dcb6098 692bd71 55b70aa 692bd71 c92f6c4 55b70aa c92f6c4 692bd71 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 |
import gemini
import llama_groq
from prompts import *
#model = gemini.GeminiModel()
model=llama_groq.LlamaModel()
def review_code(code, c_prompt=None):
if code is None or len(code) < 5 or code.isspace():
raise InvalidCodeException("No code provided")
if c_prompt is not None and len(c_prompt) > 30:
prompt = custom_review_prompt(code.strip(), c_prompt.strip())
else:
#prompt = validation_prompt(code.strip())
prompt = default_review_prompt1(code.strip())
try:
res = model.execute(prompt)
except Exception as e:
raise CodeReviewException(str(e))
return res
def refine_code(code, c_prompt=None):
if code is None or len(code) < 5 or code.isspace():
raise InvalidCodeException("No code provided")
if c_prompt is not None and len(c_prompt) > 30:
prompt = custom_review_prompt(code.strip(), c_prompt.strip())
else:
#prompt = validation_prompt(code.strip())
prompt = default_refine_prompt(code.strip())
try:
res = model.execute(prompt)
except Exception as e:
raise CodeReviewException(str(e))
return res
def generate_documentation(code,c_prompt):
if code is None or len(code) < 5 or code.isspace():
raise InvalidCodeException("No code provided")
if c_prompt is not None and len(c_prompt) > 30:
prompt = custom_review_prompt(code.strip(), c_prompt.strip())
else:
prompt = default_doc_prompt(code.strip())
try:
res = model.execute(prompt)
except Exception as e:
raise CodeReviewException(str(e))
return res
def resume_writer(code,c_prompt):
if code is None or len(code) < 5 or code.isspace():
raise InvalidCodeException("No code provided")
if c_prompt is not None and len(c_prompt) > 30:
prompt = custom_review_prompt(code.strip(), c_prompt.strip())
else:
prompt = resume_prompt(code.strip())
try:
res = model.execute(prompt)
except Exception as e:
raise CodeReviewException(str(e))
return res
class InvalidCodeException(Exception):
pass
class CodeReviewException(Exception):
pass |