lihuigu commited on
Commit
f48c751
·
1 Parent(s): 0b6004e

fix get helper bug

Browse files
Files changed (1) hide show
  1. src/utils/llms_api.py +1 -1
src/utils/llms_api.py CHANGED
@@ -37,7 +37,7 @@ class APIHelper(object):
37
  self.generator = self.get_helper()
38
  self.prompt = Prompt(get_dir(config.ARTICLE.summarizing_prompt))
39
 
40
- def get_helper():
41
  MODEL_TYPE = os.environ["MODEL_TYPE"]
42
  MODEL_NAME = os.environ["MODEL_NAME"]
43
  MODEL_API_KEY = os.environ["MODEL_API_KEY"]
 
37
  self.generator = self.get_helper()
38
  self.prompt = Prompt(get_dir(config.ARTICLE.summarizing_prompt))
39
 
40
+ def get_helper(self):
41
  MODEL_TYPE = os.environ["MODEL_TYPE"]
42
  MODEL_NAME = os.environ["MODEL_NAME"]
43
  MODEL_API_KEY = os.environ["MODEL_API_KEY"]