ccm commited on
Commit
ac7bad6
·
verified ·
1 Parent(s): faa6332

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -96,7 +96,7 @@ def preprocess(query: str, k: int) -> tuple[str, str]:
96
  prompt = prompt.replace("{{ABSTRACTS_GO_HERE}}", research_abstracts)
97
  prompt = prompt.replace("{{QUERY_GOES_HERE}}", query)
98
 
99
- return prompt, "\n\n### References\n\n"+"\n".join([str(i+1)+". " + ref + "\n\t- " + "\n\t- ".join(["", *references[ref]]) for i, ref in enumerate(references.keys())])
100
 
101
  def postprocess(response: str, bypass_from_preprocessing: str) -> str:
102
  """
 
96
  prompt = prompt.replace("{{ABSTRACTS_GO_HERE}}", research_abstracts)
97
  prompt = prompt.replace("{{QUERY_GOES_HERE}}", query)
98
 
99
+ return prompt, "\n\n### References\n\n"+"\n".join([str(i+1)+". " + ref + "\n\t- ".join(["", *references[ref].replace("\n", " ")]) for i, ref in enumerate(references.keys())])
100
 
101
  def postprocess(response: str, bypass_from_preprocessing: str) -> str:
102
  """