aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--gn3/llms/process.py6
1 files changed, 4 insertions, 2 deletions
diff --git a/gn3/llms/process.py b/gn3/llms/process.py
index e33d3bc..b29d4c9 100644
--- a/gn3/llms/process.py
+++ b/gn3/llms/process.py
@@ -7,11 +7,11 @@ import json
from urllib.parse import urljoin
from urllib.parse import quote
+import logging
import requests
from gn3.llms.client import GeneNetworkQAClient
from gn3.llms.response import DocIDs
-from gn3.settings import TMPDIR
BASE_URL = 'https://genenetwork.fahamuai.com/api/tasks'
@@ -82,13 +82,15 @@ def fetch_pubmed(references, file_name, tmp_dir=""):
"""method to fetch and populate references with pubmed"""
try:
- pubmed = load_file(file_name)
+ pubmed = load_file(file_name, tmp_dir)
for reference in references:
if pubmed.get(reference["doc_id"]):
reference["pubmed"] = pubmed.get(reference["doc_id"])
return references
except FileNotFoundError:
+ logging.debug("failed to find pubmed_path for %s/%s",
+ tmp_dir, file_name)
return references