Skip to content

Commit 49000cc

Browse files
committed
🎨 用户名生成代码适配更新后的模块
1 parent a39d25c commit 49000cc

File tree

3 files changed

+20
-6
lines changed

3 files changed

+20
-6
lines changed

Python/.gitignore

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,7 @@
11
__pypackages__/
22
.pdm.toml
33

4-
py310
4+
py310
5+
6+
# 配置文件
7+
config.toml

Python/config copy.toml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
deepseek_apikey = "sk-xxx"

Python/use_case/account/username/main.py

Lines changed: 15 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,18 @@
11
# 随机用户名生成器
22
import random
33
from typing import Optional, List
4-
from summer_modules.ai.deepseek import client
5-
from summer_modules.logger import init_and_get_logger
64
from pathlib import Path
5+
import toml
6+
7+
from summer_modules.ai.deepseek import DeepseekClient
8+
from summer_modules.logger import init_and_get_logger
79

810
CURRENT_DIR = Path(__file__).resolve().parent
911
logger = init_and_get_logger(CURRENT_DIR, "generate_username")
12+
CONFIG_TOML_FILEPATH = (CURRENT_DIR / "../../../config.toml").resolve()
13+
CONFIG = toml.load(CONFIG_TOML_FILEPATH)
14+
DEEPSEEK_APIKEY = CONFIG["deepseek_apikey"]
15+
DEEPSEEK_CLIENT = DeepseekClient(api_key=DEEPSEEK_APIKEY)
1016

1117

1218
def generate_username(style="default"):
@@ -172,7 +178,7 @@ def generate_username_with_deepseek(
172178
"""
173179

174180
try:
175-
response = client.chat.completions.create(
181+
response = DEEPSEEK_CLIENT.client.chat.completions.create(
176182
model="deepseek-chat",
177183
messages=[
178184
{"role": "system", "content": system_prompt},
@@ -185,8 +191,12 @@ def generate_username_with_deepseek(
185191
)
186192
content = response.choices[0].message.content
187193
# 将返回的用户名列表拆分成单个用户名
188-
usernames = content.split("\n")
189-
return usernames
194+
if content:
195+
usernames = content.split("\n")
196+
return usernames
197+
else:
198+
logger.error("DeepSeek API 返回的用户名为空")
199+
return []
190200

191201
except Exception as e:
192202
logger.error(f"DeepSeek API 生成用户名时发生未预期出错: {str(e)}")

0 commit comments

Comments
 (0)