(
self,
lang_in,
lang_out,
model,
base_url=None,
api_key=None,
envs=None,
prompt=None,
ignore_cache=False,
)
| 476 | CustomPrompt = True |
| 477 | |
| 478 | def __init__( |
| 479 | self, |
| 480 | lang_in, |
| 481 | lang_out, |
| 482 | model, |
| 483 | base_url=None, |
| 484 | api_key=None, |
| 485 | envs=None, |
| 486 | prompt=None, |
| 487 | ignore_cache=False, |
| 488 | ): |
| 489 | self.set_envs(envs) |
| 490 | base_url = self.envs["AZURE_OPENAI_BASE_URL"] |
| 491 | if not model: |
| 492 | model = self.envs["AZURE_OPENAI_MODEL"] |
| 493 | api_version = self.envs.get("AZURE_OPENAI_API_VERSION", "2024-06-01") |
| 494 | if api_key is None: |
| 495 | api_key = self.envs["AZURE_OPENAI_API_KEY"] |
| 496 | super().__init__(lang_in, lang_out, model, ignore_cache) |
| 497 | self.options = {"temperature": 0} |
| 498 | self.client = openai.AzureOpenAI( |
| 499 | azure_endpoint=base_url, |
| 500 | azure_deployment=model, |
| 501 | api_version=api_version, |
| 502 | api_key=api_key, |
| 503 | ) |
| 504 | self.prompttext = prompt |
| 505 | self.add_cache_impact_parameters("temperature", self.options["temperature"]) |
| 506 | self.add_cache_impact_parameters("prompt", self.prompt("", self.prompttext)) |
| 507 | |
| 508 | def do_translate(self, text) -> str: |
| 509 | response = self.client.chat.completions.create( |
nothing calls this directly
no test coverage detected