aboutsummaryrefslogtreecommitdiff
path: root/gn3
diff options
context:
space:
mode:
authorAlexander_Kabui2024-05-16 14:05:30 +0300
committerAlexander_Kabui2024-05-16 14:05:30 +0300
commit69013d298c869a42059af13bc63bef1bbdc7393d (patch)
tree551bfa836f80b801860d7df70ca9f758d04196e4 /gn3
parent92ab53e9f0b10ebf7423626e37b3fe73eb8c0b65 (diff)
downloadgenenetwork3-69013d298c869a42059af13bc63bef1bbdc7393d.tar.gz
Update file to use correct import from response file
Diffstat (limited to 'gn3')
-rw-r--r--gn3/llms/process.py7
1 files changed, 4 insertions, 3 deletions
diff --git a/gn3/llms/process.py b/gn3/llms/process.py
index e38b73e..4edc238 100644
--- a/gn3/llms/process.py
+++ b/gn3/llms/process.py
@@ -49,8 +49,9 @@ def parse_context(context, get_info_func, format_bib_func):
def rate_document(task_id, doc_id, rating, auth_token):
- """This method is used to provide feedback for a document by making a rating."""
- # todo move this to clients
+ """This method is used to provide
+ feedback for a document by making a rating
+ """
try:
url = urljoin(BASE_URL,
f"""/feedback?task_id={task_id}&document_id={doc_id}&feedback={rating}""")
@@ -107,7 +108,7 @@ def get_gnqa(query, auth_token, tmp_dir=""):
answer = resp_text['data']['answer']
context = resp_text['data']['context']
references = parse_context(
- context, DocIDs().getInfo, format_bibliography_info)
+ context, DocIDs().get_info, format_bibliography_info)
references = fetch_pubmed(references, "pubmed.json", tmp_dir)
return task_id, answer, references