(
self, lang_in, lang_out, model, envs=None, prompt=None, ignore_cache=False
)
| 561 | CustomPrompt = True |
| 562 | |
| 563 | def __init__( |
| 564 | self, lang_in, lang_out, model, envs=None, prompt=None, ignore_cache=False |
| 565 | ): |
| 566 | self.set_envs(envs) |
| 567 | base_url = "https://open.bigmodel.cn/api/paas/v4" |
| 568 | api_key = self.envs["ZHIPU_API_KEY"] |
| 569 | if not model: |
| 570 | model = self.envs["ZHIPU_MODEL"] |
| 571 | super().__init__( |
| 572 | lang_in, |
| 573 | lang_out, |
| 574 | model, |
| 575 | base_url=base_url, |
| 576 | api_key=api_key, |
| 577 | ignore_cache=ignore_cache, |
| 578 | ) |
| 579 | self.prompttext = prompt |
| 580 | self.add_cache_impact_parameters("prompt", self.prompt("", self.prompttext)) |
| 581 | |
| 582 | def do_translate(self, text) -> str: |
| 583 | try: |
nothing calls this directly
no test coverage detected