add dockerfile
separated code
هذا الالتزام موجود في:
93
filter.py
Normal file
93
filter.py
Normal file
@@ -0,0 +1,93 @@
|
||||
import logging
|
||||
import time
|
||||
from ai import generate
|
||||
import json
|
||||
from google.genai.errors import ServerError, ClientError
|
||||
|
||||
total_fail = 0
|
||||
total_overload = 0
|
||||
total_fail_overload = 0
|
||||
total_empty_response = 0
|
||||
total_fail_empty_response = 0
|
||||
|
||||
|
||||
def filter_jobs(jobs, cv, api_keys, good_fit_jobs):
|
||||
key_number = 0
|
||||
|
||||
for i, job in jobs.iterrows():
|
||||
# print("index is :", i) # for debugging
|
||||
|
||||
if (i + 1) % 10 == 0 and i != 0:
|
||||
logging.warning("sleeping to avoid API rate limits")
|
||||
time.sleep(60)
|
||||
try_count = 3
|
||||
|
||||
while try_count > 0:
|
||||
|
||||
try:
|
||||
cleaned_description = "\n".join(
|
||||
[line for line in job["description"].splitlines() if line.strip()]
|
||||
)
|
||||
ai_response = generate(cleaned_description, cv, api_keys[key_number])
|
||||
ai_response_dict = json.loads(ai_response)
|
||||
break
|
||||
|
||||
except json.JSONDecodeError as e:
|
||||
try_count -= 1
|
||||
total_empty_response += 1
|
||||
if try_count == 0:
|
||||
total_fail += 1
|
||||
total_fail_empty_response += 1
|
||||
|
||||
logging.warning("Sleeping after JSONDecodeError")
|
||||
time.sleep(6)
|
||||
|
||||
except ServerError as e:
|
||||
|
||||
if e.details["error"]["code"] == 503:
|
||||
try_count -= 1
|
||||
total_overload += 1
|
||||
if try_count == 0:
|
||||
total_fail += 1
|
||||
total_fail_overload += 1
|
||||
logging.warning("sleeping to after The model is overloaded.")
|
||||
print(e.details)
|
||||
time.sleep(10)
|
||||
else:
|
||||
logging.critical(e.details)
|
||||
return 1
|
||||
|
||||
except ClientError as e:
|
||||
if e.details["error"]["code"] == 429:
|
||||
logging.warning("api limit hit")
|
||||
key_number += 1
|
||||
if key_number > len(api_keys) - 1:
|
||||
logging.critical("All api keys hit the limit")
|
||||
return 1
|
||||
else:
|
||||
logging.critical(e.details)
|
||||
return 1
|
||||
|
||||
else:
|
||||
logging.critical("All attempts failed")
|
||||
continue
|
||||
|
||||
if ai_response_dict["percentage"] > 50:
|
||||
good_fit_jobs.append(
|
||||
{
|
||||
"title": job["title"],
|
||||
"url": job["job_url"],
|
||||
"percentage": ai_response_dict["percentage"],
|
||||
"why I'm I a good fit": ai_response_dict["why I'm I a good fit"],
|
||||
"what I'm I missing": ai_response_dict["what I'm I missing"],
|
||||
}
|
||||
)
|
||||
print_stats
|
||||
return good_fit_jobs
|
||||
|
||||
|
||||
def print_stats():
|
||||
stats = f"""total fail: {total_fail}
|
||||
total empty responses: {total_empty_response} fail: {total_fail_empty_response}
|
||||
Total overloads: {total_overload} fail: {total_fail_overload}"""
|
||||
print(stats)
|
المرجع في مشكلة جديدة
حظر مستخدم