linhkid91 commited on
Commit
937bbef
·
1 Parent(s): cb7341f

Edit some comments

Browse files
Files changed (1) hide show
  1. src/relevancy.py +3 -5
src/relevancy.py CHANGED
@@ -31,7 +31,7 @@ def encode_prompt(query, prompt_papers):
31
  prompt += f"{idx + 1}. Authors: {authors}\n"
32
  prompt += f"{idx + 1}. Abstract: {abstract}\n"
33
  prompt += f"\n Generate response:\n1."
34
- #print(prompt)
35
  return prompt
36
 
37
 
@@ -45,8 +45,6 @@ def is_json(myjson):
45
 
46
  def post_process_chat_gpt_response(paper_data, response, threshold_score=8):
47
  selected_data = []
48
- print("HERE")
49
- print(response['message']['content'])
50
  if response is None:
51
  return []
52
  json_items = response['message']['content'].replace("\n\n", "\n").split("\n")
@@ -57,14 +55,14 @@ def post_process_chat_gpt_response(paper_data, response, threshold_score=8):
57
  json.loads(re.sub(pattern, "", line))
58
  for line in json_items if (is_json(line) and "relevancy score" in line.lower())]
59
  except Exception as e:
60
- #pprint.pprint([re.sub(pattern, "", line) for line in json_items if "relevancy score" in line.lower()])
61
  try:
62
  score_items = score_items[:-1]
63
  except Exception:
64
  score_items = []
65
  print(e)
66
  raise RuntimeError("failed")
67
- #pprint.pprint(score_items)
68
  scores = []
69
  for item in score_items:
70
  temp = item["Relevancy score"]
 
31
  prompt += f"{idx + 1}. Authors: {authors}\n"
32
  prompt += f"{idx + 1}. Abstract: {abstract}\n"
33
  prompt += f"\n Generate response:\n1."
34
+ print(prompt)
35
  return prompt
36
 
37
 
 
45
 
46
  def post_process_chat_gpt_response(paper_data, response, threshold_score=8):
47
  selected_data = []
 
 
48
  if response is None:
49
  return []
50
  json_items = response['message']['content'].replace("\n\n", "\n").split("\n")
 
55
  json.loads(re.sub(pattern, "", line))
56
  for line in json_items if (is_json(line) and "relevancy score" in line.lower())]
57
  except Exception as e:
58
+ pprint.pprint([re.sub(pattern, "", line) for line in json_items if "relevancy score" in line.lower()])
59
  try:
60
  score_items = score_items[:-1]
61
  except Exception:
62
  score_items = []
63
  print(e)
64
  raise RuntimeError("failed")
65
+ pprint.pprint(score_items)
66
  scores = []
67
  for item in score_items:
68
  temp = item["Relevancy score"]