INNER CODE UNIT · Python
generate_json_resume
IvanIsCoding/ResuLLMe · src/prompt_engineering/__init__.py:171
def generate_json_resume(cv_text, api_key, model="gpt-4o", model_type="OpenAI"):
"""Generate a JSON resume from a CV text"""
sections = []
if model_type == "OpenAI":
client = OpenAI(api_key=api_key)
elif model_type == "Gemini":
genai.configure(api_key=api_key)
model = genai.GenerativeModel(model)
for prompt in stqdm(
[
BASICS_PROMPT,
EDUCATION_PROMPT,
AWARDS_PROMPT,
PROJECTS_PROMPT,
SKILLS_PROMPT,
WORK_PROMPT,
],