fix get helper bug
Browse files- 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"]
|