martin/linkedIn_auto_jobs_applier_with_AI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

use gpt-4o-mini

Browse Source
This commit is contained in:
Maurice McCabe 2024-08-31 15:58:55 -07:00
parent 67543499ab
commit 369c23791d
1 changed files with 2 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
resume_yaml_generator.py
View File

@ -79,23 +79,13 @@ def generate_yaml_from_resume(resume_text: str, schema: Dict[str, Any], api_key:
Generate the YAML content that matches this schema based on the resume content provided, ensuring all format hints are followed and making educated guesses where necessary. Be sure to include best guesses for ALL fields, even if not explicitly mentioned in the resume.
Enclose your response in <resume_yaml> tags. Only include the YAML content within these tags, without any additional text or code block markers.
"""
model = "gpt-3.5-turbo-16k" # This model has a 16k token limit
tokens = num_tokens_from_string(prompt, model)
max_tokens = min(16385 - tokens, 4000) # Ensure we don't exceed model's limit
if tokens > 16385:
print(f"Warning: The input exceeds the model's context length. Tokens: {tokens}")
response = client.chat.completions.create(
model=model,
model="gpt-4o-mini",
messages=[
{"role": "system", "content": "You are a helpful assistant that generates structured YAML content from resume files, paying close attention to format requirements and schema structure."},
{"role": "user", "content": prompt}
],
max_tokens=max_tokens,
n=1,
stop=None,
temperature=0.5,
)
@ -106,8 +96,7 @@ def generate_yaml_from_resume(resume_text: str, schema: Dict[str, Any], api_key:
if match:
return match.group(1).strip()
else:
raise ValueError("YAML content not found in the expected format")
raise ValueError("YAML content not found in the expected format")
def save_yaml(data: str, output_file: str):
with open(output_file, 'w') as file:
file.write(data)