diff options
| author | Alexander_Kabui | 2024-05-17 13:34:37 +0300 |
|---|---|---|
| committer | Alexander_Kabui | 2024-05-17 14:46:36 +0300 |
| commit | d3bc323fe3a965ee5b6917987c4fe7662056e560 (patch) | |
| tree | b0793a643ad30ae209d426866b80391f4143c29b /gn3/llms/process.py | |
| parent | 4d39c26b65aad3fa48d35fc11007f5f3afe1c112 (diff) | |
| download | genenetwork3-d3bc323fe3a965ee5b6917987c4fe7662056e560.tar.gz | |
Refactor custom request codebase.
Diffstat (limited to 'gn3/llms/process.py')
| -rw-r--r-- | gn3/llms/process.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/gn3/llms/process.py b/gn3/llms/process.py index 9cb09a1..4725bcb 100644 --- a/gn3/llms/process.py +++ b/gn3/llms/process.py @@ -5,7 +5,6 @@ import string import json import logging from urllib.parse import quote -import requests from gn3.llms.client import GeneNetworkQAClient @@ -118,7 +117,7 @@ def get_gnqa(query, auth_token, data_dir=""): references: contains doc_name,reference,pub_med_info """ - api_client = GeneNetworkQAClient(requests.Session(), api_key=auth_token) + api_client = GeneNetworkQAClient(api_key=auth_token) res, task_id = api_client.ask('?ask=' + quote(query), auth_token) if task_id == 0: raise RuntimeError(f"Error connecting to Fahamu Api: {str(res)}") @@ -135,7 +134,7 @@ def get_gnqa(query, auth_token, data_dir=""): return task_id, answer, references else: - return task_id, "Please try to rephrase your question to receive feedback", [] + return task_id, "We couldn't provide a response,Please try to rephrase your question to receive feedback", [] def fetch_query_results(query, user_id, redis_conn): |