Rohan Kataria commited on
Commit
d7cbf82
·
1 Parent(s): 576e29e

added clear api function

Browse files
Files changed (1) hide show
  1. src/main.py +5 -1
src/main.py CHANGED
@@ -36,6 +36,10 @@ class ChatWrapper:
36
  raise ValueError(f'Invalid chain_type: {chain_type}')
37
  else:
38
  self.chain = None
 
 
 
 
39
 
40
  def __call__(self, inp: str):
41
  self.lock.acquire()
@@ -49,7 +53,7 @@ class ChatWrapper:
49
  except Exception as e:
50
  self.history.append((inp, f"An error occurred: {e}"))
51
  finally:
52
- self.api_key = '' # API key is cleared after running each chain
53
  self.lock.release()
54
 
55
  return self.history
 
36
  raise ValueError(f'Invalid chain_type: {chain_type}')
37
  else:
38
  self.chain = None
39
+
40
+ def clear_api_key(self):
41
+ if hasattr(self, 'api_key'):
42
+ del self.api_key
43
 
44
  def __call__(self, inp: str):
45
  self.lock.acquire()
 
53
  except Exception as e:
54
  self.history.append((inp, f"An error occurred: {e}"))
55
  finally:
56
+ self.clear_api_key() # API key is cleared after running each chain in the class
57
  self.lock.release()
58
 
59
  return self.history