update

2026-05-28 19:13:14 -06:00
parent 6ab2886f77
commit fd7384980a
2 changed files with 29 additions and 12 deletions
--- a/utils/openai.py
+++ b/utils/openai.py
@@ -1,4 +1,4 @@
-from openai import OpenAI
+from openai import AsyncOpenAI, OpenAI

 from utils.shortcuts import get_env

@@ -11,3 +11,11 @@ def get_ai_client() -> OpenAI:
        raise Exception("缺少 AI_KEY")

    return OpenAI(api_key=key, base_url=BASE_URL)
+
+
+def get_async_ai_client() -> AsyncOpenAI:
+    key = get_env("AI_KEY")
+    if not key:
+        raise Exception("缺少 AI_KEY")
+
+    return AsyncOpenAI(api_key=key, base_url=BASE_URL)