fix query formatting and increase no retries

2 files changed, 10 insertions, 3 deletions

diff --git a/gn3/llms/client.py b/gn3/llms/client.py
index 4a43d2a..880df1d 100644
--- a/gn3/llms/client.py
+++ b/gn3/llms/client.py
@@ -12,6 +12,8 @@ from requests.packages.urllib3.util.retry import Retry
 from requests import HTTPError
 from requests import Session
 from requests.adapters import HTTPAdapter
+from urllib.request import urlretrieve
+from urllib.parse import quote
 
 basedir = os.path.join(os.path.dirname(__file__))
 
@@ -133,7 +135,7 @@ class GeneNetworkQAClient(Session):
 
     def custom_request(self, method, url, *args, **kwargs):
 
-        max_retries = 5
+        max_retries = 8
         retry_delay = 10
 
         response = super().request(method, url, *args, **kwargs)
@@ -195,7 +197,8 @@ class GeneNetworkQAClient(Session):
         return '?task_id=' + str(task_id['task_id'])
 
     def get_gnqa(self, query):
-        res, task_id = api_client.ask('?ask=' + query)
+        qstr = quote(query)
+        res, task_id = api_client.ask('?ask=' + qstr)
         res, success = api_client.get_answer(task_id)
 
         if success == 1:
@@ -205,3 +208,4 @@ class GeneNetworkQAClient(Session):
             return answer, context
         else:
             return res, "Unfortunately, I have nothing."
+
diff --git a/gn3/llms/process.py b/gn3/llms/process.py
index 58d565a..526c1b3 100644
--- a/gn3/llms/process.py
+++ b/gn3/llms/process.py
@@ -7,6 +7,9 @@ import time
 import string
 import json
 import os
+from urllib.request import urlretrieve
+from urllib.parse import quote
+
 from gn3.llms.client import GeneNetworkQAClient
 from gn3.llms.response import DocIDs
 
@@ -77,7 +80,7 @@ def filterResponseText(val):
 
 def getGNQA(query, auth_token):
     apiClient = GeneNetworkQAClient(requests.Session(), api_key=auth_token)
-    res, task_id = apiClient.ask('?ask=' + query, auth_token)
+    res, task_id = apiClient.ask('?ask=' + quote(query), auth_token)
     res, success = apiClient.get_answer(task_id)
 
     if (success == 1):