自动生成流程图

2026-01-05 10:23:02 +08:00
parent 7a0368227f
commit ed40019a14
5 changed files with 45 additions and 15 deletions
--- a/utils/openai.py
+++ b/utils/openai.py
@@ -0,0 +1,13 @@
+from openai import OpenAI
+
+from utils.shortcuts import get_env
+
+BASE_URL = "https://api.deepseek.com"
+
+
+def get_ai_client() -> OpenAI:
+    key = get_env("AI_KEY")
+    if not key:
+        raise Exception("缺少 AI_KEY")
+
+    return OpenAI(api_key=key, base_url=BASE_URL)