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"]
|