diff options
| author | Alexander_Kabui | 2024-05-16 14:05:30 +0300 |
|---|---|---|
| committer | Alexander_Kabui | 2024-05-16 14:05:30 +0300 |
| commit | 69013d298c869a42059af13bc63bef1bbdc7393d (patch) | |
| tree | 551bfa836f80b801860d7df70ca9f758d04196e4 /gn3/llms/process.py | |
| parent | 92ab53e9f0b10ebf7423626e37b3fe73eb8c0b65 (diff) | |
| download | genenetwork3-69013d298c869a42059af13bc63bef1bbdc7393d.tar.gz | |
Update file to use correct import from response file
Diffstat (limited to 'gn3/llms/process.py')
| -rw-r--r-- | gn3/llms/process.py | 7 |
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 |