diff options
| -rw-r--r-- | scripts/fetch_papers.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/scripts/fetch_papers.py b/scripts/fetch_papers.py index 68a9fe9..79246c9 100644 --- a/scripts/fetch_papers.py +++ b/scripts/fetch_papers.py @@ -36,9 +36,10 @@ def advanced_filter(entry): API_URL = "https://uiuc.chat/api/chat-api/chat" MODEL_NAME = "qwen2.5:14b-instruct-fp16" SYSTEM_PROMPT = ( - "Based on the given title and abstract, please determine if the paper " - "is relevant to both language models and bias (or fairness). " - "If yes, respond 1; otherwise respond 0." + "You are a helpful assistant. The user will give you a paper title and abstract. " + "Your task: Decide if this paper is about large language models (or generative text models) AND about bias/fairness. " + "If yes, respond with just a single character: 1. Otherwise, respond with a single character: 0. " + "No extra explanation, no punctuation—only the number." ) def is_relevant_by_api(title, summary, api_key): @@ -56,7 +57,7 @@ def is_relevant_by_api(title, summary, api_key): }, { "role": "user", - "content": f"Title: {title}\nAbstract: {summary}" + "content": SYSTEM_PROMPT + f"Title: {title}\nAbstract: {summary}" } ], "api_key": api_key, |