# 一個(gè)基本的 API 調(diào)用函數(shù)
from openai import OpenAI, APIConnectionError

def translate_text(text: str) -> str:
    message = [
        {'role': 'system', 'content': '您是一名頂級的字幕翻譯引擎。'},
        {'role': 'user', 'content': f'<INPUT>{text}</INPUT>'},
    ]
    model = OpenAI(api_key="YOUR_API_KEY", base_url="...") 

    try:
        response = model.chat.completions.create(model="gpt-4o", messages=message)
        if response.choices:
            return response.choices[0].message.content.strip()
        raise RuntimeError("API未返回有效結(jié)果")
    except APIConnectionError as e:
        print(f"網(wǎng)絡(luò)連接失敗: {e}。需要重試...")
        raise # 程序在這里崩潰
    except Exception as e:
        print(f"發(fā)生其他錯(cuò)誤: {e}")
        raise

# 手動實(shí)現(xiàn)重試
for attempt in range(3):
    try:
        # ... API 調(diào)用邏輯 ...
        return response.choices[0].message.content.strip()
    except APIConnectionError as e:
        print(f"第 {attempt + 1} 次嘗試失敗: {e}")
        if attempt == 2: # 檢查是否是最后一次嘗試
            raise
    # ... 對其他異常也要重復(fù)寫相似的邏輯 ...

from tenacity import retry

@retry
def translate_text(text: str) -> str:
    # ... 內(nèi)部邏輯和之前完全一樣，無需任何改動 ...

from tenacity import retry, stop_after_attempt

# 總共嘗試 3 次
@retry(stop=stop_after_attempt(3))
def translate_text(text: str) -> str:
    # ...

from tenacity import stop_after_delay

# 總次數(shù)達(dá)到 5 次或總耗時(shí)超過 30 秒，就停止
@retry(stop=(stop_after_attempt(5) | stop_after_delay(30)))
def translate_text(text: str) -> str:
    # ...

from tenacity import retry, wait_fixed

# 每次重試前都等待 2 秒
@retry(wait=wait_fixed(2))
def translate_text(text: str) -> str:
    # ...

from tenacity import wait_exponential

# 首次重試等 2^1=2s, 之后等 4s, 8s... 最多等到 10s
@retry(wait=wait_exponential(multiplier=1, min=2, max=10))
def translate_text(text: str) -> str:
    # ...

# 錯(cuò)誤的方式！這無法按預(yù)期工作
@retry(retry=retry_if_not_exception_type(AuthenticationError | PermissionDeniedError))

from openai import AuthenticationError, PermissionDeniedError

# 正確的方式！使用元組
@retry(retry=retry_if_not_exception_type((AuthenticationError, PermissionDeniedError)))

from tenacity import RetryCallState

def my_error_callback(retry_state: RetryCallState):
    # retry_state 對象包含了這次重試的所有信息
    print(f"所有 {retry_state.attempt_number} 次嘗試均失敗！")
    return "默認(rèn)的翻譯結(jié)果或錯(cuò)誤提示"

@retry(stop=stop_after_attempt(3), retry_error_callback=my_error_callback)
def translate_text(text: str) -> str:
    # ...

def return_last_value(retry_state: RetryCallState):
    # 危險(xiǎn)！這會重新拋出異常！
    return "失?。? + retry_state.outcome.result()

def return_last_value(retry_state: RetryCallState):
    # 安全！這只會返回異常對象
    last_exception = retry_state.outcome.exception()
    return f"經(jīng)過 {retry_state.attempt_number} 次嘗試后失敗。最后一次錯(cuò)誤是: {last_exception}"

class TTS:
    def __init__(self, name):
        self.name = name

    def _my_callback(self, retry_state):
        print(f"實(shí)例 {self.name} 的任務(wù)失敗了。")
        # ...

    # 這會失敗！NameError: name 'self' is not defined
    @retry(retry_error_callback=self._my_callback)
    def run(self):
        # ...

class TTS:
    def __init__(self, name):
        self.name = name

    def run(self):
        # 在這里，self 是可用的！
        @retry(
            # 因?yàn)檠b飾器在 run 方法內(nèi)部，它可以“捕獲”到 self
            retry_error_callback=self._my_callback
        )
        def _execute_task():
            # 這里是真正需要重試的邏輯
            print(f"正在為 {self.name} 執(zhí)行任務(wù)...")
            raise ValueError("任務(wù)失敗")

        # 調(diào)用被裝飾的內(nèi)部函數(shù)
        return _execute_task()

    def _my_callback(self, retry_state: RetryCallState):
        # ...

from openai import OpenAI, AuthenticationError, PermissionDeniedError
from tenacity import retry, stop_after_attempt, wait_fixed, retry_if_not_exception_type, RetryCallState

# 1. 定義一個(gè)通用的、可復(fù)用的異常處理類
class RetryRaise:
    # 定義不應(yīng)重試的致命異常
    NO_RETRY_EXCEPT = (AuthenticationError, PermissionDeniedError)

    @classmethod
    def _raise(cls, retry_state: RetryCallState):
        ex = retry_state.outcome.exception()
        if ex:
            # 根據(jù)不同異常類型，進(jìn)行日志記錄并拋出自定義的、更友好的 RuntimeError
            # ... 此處可以添加更復(fù)雜的異常分類邏輯 ...
            raise RuntimeError(f"重試 {retry_state.attempt_number} 次后最終失敗: {ex}") from ex
        raise RuntimeError(f"重試 {retry_state.attempt_number} 次后失敗，但未捕獲到異常。")

# 2. 實(shí)現(xiàn)模板父類
class BaseProvider:
    @retry(
        stop=stop_after_attempt(3),
        wait=wait_fixed(2),
        retry=retry_if_not_exception_type(RetryRaise.NO_RETRY_EXCEPT),
        retry_error_callback=RetryRaise._raise
    )
    def _exec(self) -> str:
        """這是模板方法，負(fù)責(zé)重試。子類不應(yīng)重寫它。"""
        # 調(diào)用鉤子方法，由子類實(shí)現(xiàn)
        return self._do_work()

    def _do_work(self) -> str:
        """這是鉤子方法，子類必須實(shí)現(xiàn)它。"""
        raise NotImplementedError("子類必須實(shí)現(xiàn) _do_work 方法")

# 3. 實(shí)現(xiàn)具體的子類
class DeepSeekProvider(BaseProvider):
    def __init__(self, api_key: str, base_url: str):
        self.api_key = api_key
        self.base_url = base_url
        self.model = OpenAI(api_key=self.api_key, base_url=self.base_url)

    def _do_work(self) -> str:
        """這里只關(guān)心核心業(yè)務(wù)邏輯，完全不用考慮重試。"""
        response = self.model.chat.completions.create(
            model="deepseek-chat",
            messages=[{'role': 'user', 'content': '你是誰？'}]
        )
        if response.choices:
            return response.choices[0].message.content.strip()
        raise RuntimeError(f"API未返回有效結(jié)果: {response}")

# --- 如何使用 ---
 provider = DeepSeekProvider(api_key="...", base_url="...")
 try:
     # 我們調(diào)用的是 _exec，它包含了重試邏輯
     result = provider._exec()
     print("執(zhí)行成功:", result)
 except RuntimeError as e:
     # 如果最終失敗，會捕獲到 RetryRaise 拋出的友好異常
     print("執(zhí)行失敗:", e)