|
|
@@ -1,7 +1,7 @@
|
|
|
"""
|
|
|
@author: luojunhui
|
|
|
"""
|
|
|
-from openai import OpenAI
|
|
|
+from volcenginesdkarkruntime import Ark
|
|
|
|
|
|
from config import deep_seek_model
|
|
|
from config import deep_seek_default_model
|
|
|
@@ -12,10 +12,8 @@ def fetch_deepseek_response(model, prompt):
|
|
|
"""
|
|
|
deep_seek方法
|
|
|
"""
|
|
|
- client = OpenAI(
|
|
|
+ client = Ark(
|
|
|
api_key=deep_seek_api_key_byte_dance,
|
|
|
- base_url="https://ark.cn-beijing.volces.com/api/v3",
|
|
|
- # 深度推理模型耗费时间会较长，建议您设置一个较长的超时时间，推荐为30分钟
|
|
|
timeout=1800,
|
|
|
)
|
|
|
response = client.chat.completions.create(
|
