JohnSmith9982 commited on
Commit
183a375
1 Parent(s): 2b718c8

Upload 39 files

Browse files
Files changed (2) hide show
  1. config.json +2 -0
  2. modules/models.py +5 -0
config.json ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ {
2
+ }
modules/models.py CHANGED
@@ -209,6 +209,11 @@ class OpenAIClient(BaseLLMModel):
209
  if error_msg:
210
  raise Exception(error_msg)
211
 
 
 
 
 
 
212
 
213
  class ChatGLM_Client(BaseLLMModel):
214
  def __init__(self, model_name) -> None:
 
209
  if error_msg:
210
  raise Exception(error_msg)
211
 
212
+ def set_key(self, new_access_key):
213
+ ret = super().set_key(new_access_key)
214
+ self._refresh_header()
215
+ return ret
216
+
217
 
218
  class ChatGLM_Client(BaseLLMModel):
219
  def __init__(self, model_name) -> None: