main
黄海 5 months ago
parent 5bb2e22d42
commit 1bce02a4aa

@ -25,7 +25,7 @@ class MarkdownGenerator:
self.course_name = course_name
self.template_path = template_path or DEFAULT_TEMPLATE
self.output_path = output_path or DEFAULT_OUTPUT_DIR / f"{course_name}.md"
self.client = OpenAI(api_key=MODEL_API_KEY, base_url=MODEL_URL) # 初始化OpenAI客户端
self.client = OpenAI(api_key=MODEL_API_KEY, base_url=MODEL_API_URL) # 初始化OpenAI客户端
self._validate_paths()

@ -11,7 +11,7 @@ class ContentAnalyzer:
def __init__(
self,
api_key: str = MODEL_API_KEY,
base_url: str = MODEL_URL,
base_url: str = MODEL_API_URL,
model: str = MODEL_NAME,
max_retries: int = 10,
initial_timeout: int = 300

@ -9,7 +9,7 @@ class EnglishEssayAnalyzer:
def __init__(self):
self.client = OpenAI(
api_key=MODEL_API_KEY,
base_url=MODEL_URL
base_url=MODEL_API_URL
)
def _build_prompt(self, essay: str) -> str:

@ -5,7 +5,7 @@ from pathlib import Path
MODEL_API_KEY = "sk-01d13a39e09844038322108ecdbd1bbc"
MODEL_NAME = "deepseek-r1"
#MODEL_NAME='qwen-plus'
MODEL_URL='https://dashscope.aliyuncs.com/compatible-mode/v1'
MODEL_API_URL= 'https://dashscope.aliyuncs.com/compatible-mode/v1'
# 华为云
HW_API_KEY='2R7vrjfIZO1chcfk4V3zDscx2Ms4I4kdoY7XSFnxa8Z2RNfAztg8qrXXKkwJGZWvMtS4qnN782WmaY4xZfMhyg'

@ -10,7 +10,7 @@ from Config import *
class KnowledgeGraph:
def __init__(self, shiti_content: str):
self.shiti_content = shiti_content
self.client = OpenAI(api_key=HW_API_KEY, base_url=HW_API_URL)
self.client = OpenAI(api_key=MODEL_API_KEY, base_url=MODEL_API_URL)
def _generate_stream(self) -> Iterator[ChatCompletionChunk]:
"""流式生成内容"""

Loading…
Cancel
Save