Skip to content

Commit 3e81bce

Browse files
committed
feat: summarizer
1 parent a7ff7ca commit 3e81bce

File tree

1 file changed

+12
-6
lines changed

1 file changed

+12
-6
lines changed

ovos_solver_openai_persona/engines.py

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,17 @@
1111

1212

1313
class OpenAICompletionsSolver(QuestionSolver):
14-
def __init__(self, config=None, enable_tx=False, enable_cache=False, priority=25):
15-
super().__init__(config, enable_tx=enable_tx, enable_cache=enable_cache, priority=priority)
14+
def __init__(self, config=None,
15+
translator: Optional[LanguageTranslator] = None,
16+
detector: Optional[LanguageDetector] = None,
17+
priority: int = 50,
18+
enable_tx: bool = False,
19+
enable_cache: bool = False,
20+
internal_lang: Optional[str] = None):
21+
super().__init__(config=config, translator=translator,
22+
detector=detector, priority=priority,
23+
enable_tx=enable_tx, enable_cache=enable_cache,
24+
internal_lang=internal_lang)
1625
self.api_url = f"{self.config.get('api_url', 'https://api.openai.com/v1')}/completions"
1726
self.engine = self.config.get("model", "text-davinci-002") # "ada" cheaper and faster, "davinci" better
1827
self.stop_token = "<|im_end|>"
@@ -76,13 +85,10 @@ def post_process_sentence(text: str) -> str:
7685

7786

7887
class OpenAIChatCompletionsSolver(ChatMessageSolver):
79-
enable_tx = False
80-
priority = 25
81-
8288
def __init__(self, config=None,
8389
translator: Optional[LanguageTranslator] = None,
8490
detector: Optional[LanguageDetector] = None,
85-
priority: int = 50,
91+
priority: int = 25,
8692
enable_tx: bool = False,
8793
enable_cache: bool = False,
8894
internal_lang: Optional[str] = None):

0 commit comments

Comments
 (0)