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

13
resume_yaml_generator.py
View File

@ -80,22 +80,12 @@ def generate_yaml_from_resume(resume_text: str, schema: Dict[str, Any], api_key:
Enclose your response in <resume_yaml> tags. Only include the YAML content within these tags, without any additional text or code block markers. 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( response = client.chat.completions.create(
model=model, model="gpt-4o-mini",
messages=[ 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": "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} {"role": "user", "content": prompt}
], ],
max_tokens=max_tokens,
n=1,
stop=None,
temperature=0.5, temperature=0.5,
) )
@ -107,7 +97,6 @@ def generate_yaml_from_resume(resume_text: str, schema: Dict[str, Any], api_key:
return match.group(1).strip() return match.group(1).strip()
else: 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): def save_yaml(data: str, output_file: str):
with open(output_file, 'w') as file: with open(output_file, 'w') as file:
file.write(data) file.write(data)