Remove kwargs from LLMErrorr Exceptions and update relevant code.

2 files changed, 3 insertions, 4 deletions

diff --git a/gn3/llms/client.py b/gn3/llms/client.py
index 5c4fa0e..d29d2a1 100644
--- a/gn3/llms/client.py
+++ b/gn3/llms/client.py
@@ -117,11 +117,11 @@ class GeneNetworkQAClient(Session):
                     continue
                 return response
             else:
-                raise LLMError( f"Request error with code:\
+                raise LLMError(f"Request error with code:\
                 {response.status_code} occurred with reason:\
                 {response_msg.get(response.status_code,response.reason)}",
-                                query=self.query)
+                               self.query)
                 #time.sleep(retry_delay)
         raise LLMError("Time error: We couldn't provide a response,Please try\
         to rephrase your question to receive feedback",
-                       query=self.query)
+                       self.query)
diff --git a/gn3/llms/errors.py b/gn3/llms/errors.py
index c5439d6..77e0f9a 100644
--- a/gn3/llms/errors.py
+++ b/gn3/llms/errors.py
@@ -39,4 +39,3 @@ class LLMError(Exception):
     """custom exception for LLMErrorMIxins"""
     def __init__(self, *args, **kwargs):
         super().__init__(*args)
-        self.query = kwargs.get("query")