|
|
# pip install dashscope python-dotenv
|
|
|
|
|
|
from pathlib import Path
|
|
|
|
|
|
import dashscope
|
|
|
from dashscope import Generation
|
|
|
from dashscope.api_entities.dashscope_response import DashScopeAPIResponse
|
|
|
|
|
|
|
|
|
class TestDeepSeek:
|
|
|
# 模型版本常量
|
|
|
R1 = "deepseek-r1"
|
|
|
V3 = "deepseek-v3"
|
|
|
# API密钥（建议从环境变量读取）
|
|
|
API_KEY = "sk-01d13a39e09844038322108ecdbd1bbc"
|
|
|
|
|
|
@staticmethod
|
|
|
def get_system_prompt() -> str:
|
|
|
"""读取系统提示文件"""
|
|
|
md_path = Path(r"D:\dsWork\QingLong\PptGenerator\md-file\readme\default.md")
|
|
|
if not md_path.exists():
|
|
|
raise FileNotFoundError(f"提示文件不存在: {md_path}")
|
|
|
return md_path.read_text(encoding='utf-8')
|
|
|
|
|
|
@classmethod
|
|
|
def call_with_message(cls) -> DashScopeAPIResponse:
|
|
|
"""调用生成API"""
|
|
|
system_prompt = (
|
|
|
"返回的教案内容请严格按照指定的markdown语法格式返回，用来生成ppt。"
|
|
|
"需要注意:1、最开头的Title:..,Author: ... Date:... 不能省略掉，有用处。"
|
|
|
"2、第一级用#,第二级用##,第二级是独立的PPT一个页，每页中的数据必须有> 开头,"
|
|
|
"如果是页面中的条目，必须用 - 开头,结果不要返回图片！请仔细阅读参考格式："
|
|
|
) + cls.get_system_prompt()
|
|
|
|
|
|
try:
|
|
|
return Generation.call(
|
|
|
model=cls.R1,
|
|
|
api_key=cls.API_KEY,
|
|
|
messages=[
|
|
|
{"role": "system", "content": system_prompt},
|
|
|
{"role": "user", "content": "请帮我生成一个小学数学内角和的教案，按标准格式输出。"}
|
|
|
],
|
|
|
result_format='message' # 与Java的GenerationParam.ResultFormat.MESSAGE对应
|
|
|
)
|
|
|
except Exception as e: # 根据实际SDK异常类型细化
|
|
|
raise RuntimeError(f"API调用失败: {str(e)}") from e
|
|
|
|
|
|
@classmethod
|
|
|
def main(cls):
|
|
|
"""主执行方法"""
|
|
|
try:
|
|
|
result = cls.call_with_message()
|
|
|
if result.status_code == 200 and result.output:
|
|
|
md_content = result.output.choices[0].message.content
|
|
|
print(md_content)
|
|
|
|
|
|
output_path = Path(r"D:\dsWork\QingLong\PptGenerator\md-file\readme\5.md")
|
|
|
output_path.parent.mkdir(parents=True, exist_ok=True)
|
|
|
output_path.write_text(md_content, encoding='utf-8')
|
|
|
print(f"结果已保存至: {output_path}")
|
|
|
else:
|
|
|
print(f"请求失败，状态码: {result.status_code}, 原因: {result.message}")
|
|
|
except Exception as e:
|
|
|
print(f"发生错误: {str(e)}")
|
|
|
print("请参考文档：https://help.aliyun.com/zh/model-studio/developer-reference/error-code")
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
dashscope.api_key = TestDeepSeek.API_KEY
|
|
|
TestDeepSeek.main() |
