aboutsummaryrefslogtreecommitdiff
path: root/gn3/llms
diff options
context:
space:
mode:
authorAlexander_Kabui2024-03-25 16:27:37 +0300
committerAlexander_Kabui2024-03-25 16:27:37 +0300
commit6796e81483b30fac99410170a3abd1e8120fae6b (patch)
tree1865d08438af45bdd8f6a45a079d944aea1e39cb /gn3/llms
parentd797efb7c71e001e50830c10f30a89f1642cafd7 (diff)
downloadgenenetwork3-6796e81483b30fac99410170a3abd1e8120fae6b.tar.gz
pep8 formatting
Diffstat (limited to 'gn3/llms')
-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