From 188f84ef3895de613e998c63da7ec2338e25a55c Mon Sep 17 00:00:00 2001
From: Alexander_Kabui
Date: Fri, 24 May 2024 16:30:59 +0300
Subject: Remove kwargs from LLMErrorr Exceptions and update relevant code.

---
 gn3/llms/client.py | 6 +++---
 gn3/llms/errors.py | 1 -
 2 files changed, 3 insertions(+), 4 deletions(-)

(limited to 'gn3/llms')

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")
-- 
cgit v1.2.3