cv / groq_api.py
saherPervaiz's picture
Update groq_api.py
9089c56 verified
raw
history blame
2.56 kB
# groq_api.py
import os
import requests
GROQ_API_KEY = "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"
def summarize_match(job_description, cv_names, cv_snippets):
if not GROQ_API_KEY:
return "❌ GROQ_API_KEY not set."
try:
# Truncate job description
job_description = job_description.strip()[:800] or "[No description provided]"
# Apply safe truncation (summary) to each CV
summarized_snippets = [(cv.strip()[:700] or "[No content]") for cv in cv_snippets]
cv_names = [name[:60] for name in cv_names]
# Build dynamic CV section
cv_section = ""
for i, (name, snippet) in enumerate(zip(cv_names, summarized_snippets), start=1):
cv_section += f"\n{i}. {name}:\n{snippet}\n"
prompt = f"""
You are an AI recruitment assistant helping to evaluate candidates for an open job role.
Below is the job description, followed by candidate CV summaries.
Your task is to assess how well each candidate aligns with the job. Be honest β€” if a candidate lacks the required skills or experience, clearly say they are not a good fit.
### Job Description:
{job_description}
### Candidate CVs:{cv_section}
### Instructions:
- Focus on **specific skills**, **technical tools**, **relevant experience**, and **industry fit**.
- Clearly identify **who is a good match**, and **why**.
- Clearly explain **who is not a match**, and **why not**.
- Avoid general language like "strong background" unless it's backed by actual experience.
End with a short ranked list of the candidates.
""".strip()
# Truncate if too long
if len(prompt) > 8000:
prompt = prompt[:8000]
response = requests.post(
url="https://api.groq.com/openai/v1/chat/completions",
headers={
"Authorization": f"Bearer {GROQ_API_KEY}",
"Content-Type": "application/json"
},
json={
"model": "llama3-8b-8192",
"messages": [
{"role": "system", "content": "You are a helpful recruitment assistant."},
{"role": "user", "content": prompt}
],
"temperature": 0.4
},
timeout=30
)
response.raise_for_status()
return response.json()["choices"][0]["message"]["content"]
except requests.exceptions.RequestException as e:
return f"❌ Groq API error: {e}"
except Exception as e:
return f"❌ Unexpected error: {e}"