update get api help
Browse files- src/utils/llms_api.py +11 -7
src/utils/llms_api.py
CHANGED
|
@@ -1,8 +1,9 @@
|
|
| 1 |
from .api import HelperCompany
|
| 2 |
-
import openai
|
| 3 |
import re
|
|
|
|
| 4 |
from .header import get_dir, Prompt, ConfigReader
|
| 5 |
import traceback
|
|
|
|
| 6 |
|
| 7 |
TAG_moti = "Motivations:"
|
| 8 |
TAG_contr = "Details:"
|
|
@@ -32,14 +33,17 @@ class APIHelper(object):
|
|
| 32 |
super(APIHelper, self).__init__()
|
| 33 |
self.config = config
|
| 34 |
self.__checkout_config__()
|
| 35 |
-
self.summarizer = self.get_helper(
|
| 36 |
-
self.generator = self.get_helper(
|
| 37 |
self.prompt = Prompt(get_dir(config.ARTICLE.summarizing_prompt))
|
| 38 |
|
| 39 |
-
def get_helper(
|
| 40 |
-
|
| 41 |
-
|
| 42 |
-
|
|
|
|
|
|
|
|
|
|
| 43 |
)
|
| 44 |
|
| 45 |
def __checkout_config__(self):
|
|
|
|
| 1 |
from .api import HelperCompany
|
|
|
|
| 2 |
import re
|
| 3 |
+
import os
|
| 4 |
from .header import get_dir, Prompt, ConfigReader
|
| 5 |
import traceback
|
| 6 |
+
import openai
|
| 7 |
|
| 8 |
TAG_moti = "Motivations:"
|
| 9 |
TAG_contr = "Details:"
|
|
|
|
| 33 |
super(APIHelper, self).__init__()
|
| 34 |
self.config = config
|
| 35 |
self.__checkout_config__()
|
| 36 |
+
self.summarizer = self.get_helper()
|
| 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"]
|
| 44 |
+
BASE_URL = os.environ["BASE_URL"]
|
| 45 |
+
return HelperCompany.get()[MODEL_TYPE](
|
| 46 |
+
MODEL_API_KEY, MODEL_NAME, BASE_URL, timeout=None
|
| 47 |
)
|
| 48 |
|
| 49 |
def __checkout_config__(self):
|