104 lines
3.2 KiB
Python
104 lines
3.2 KiB
Python
|
"""候选知识点生成：调用 DeepSeek 分析文本，生成 ImportCandidate"""
|
|||
|
|
|
|||
|
|
import json
|
|||
|
|
import httpx
|
|||
|
|
from config import DEEPSEEK_API_KEY, DEEPSEEK_BASE_URL, DEEPSEEK_MODEL
|
|||
|
|
|
|||
|
|
MAX_CANDIDATES = 30
|
|||
|
|
MIN_CANDIDATES = 3
|
|||
|
|
CHARS_PER_CANDIDATE = 2000
|
|||
|
|
|
|||
|
|
_PROMPT = """你是一个学习助手。请分析以下文档内容，提取关键知识点。
|
|||
|
|
|
|||
|
|
对于每个知识点，请提供：
|
|||
|
|
- title: 知识点标题（简洁，不超过 30 字）
|
|||
|
|
- summary: 一句话概述（不超过 80 字）
|
|||
|
|
- content: 详细解释（基于原文，保持准确）
|
|||
|
|
- tags: 2-4 个标签
|
|||
|
|
- recallQuestions: 1-2 个主动回忆问题
|
|||
|
|
- difficulty: 难度评估（easy/medium/hard）
|
|||
|
|
- confidence: 你对这个知识点重要性的置信度（0.0-1.0）
|
|||
|
|
|
|||
|
|
请以 JSON 数组格式返回，每个元素是一个知识点：
|
|||
|
|
```json
|
|||
|
|
[{
|
|||
|
|
"title": "知识点标题",
|
|||
|
|
"summary": "一句话概述",
|
|||
|
|
"content": "详细解释...",
|
|||
|
|
"tags": ["标签1", "标签2"],
|
|||
|
|
"recallQuestions": ["问题1？", "问题2？"],
|
|||
|
|
"difficulty": "medium",
|
|||
|
|
"confidence": 0.85
|
|||
|
|
}]
|
|||
|
|
```
|
|||
|
|
|
|||
|
|
文档内容：
|
|||
|
|
{text}
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def generate_candidates(text: str) -> list[dict]:
|
|||
|
|
"""用 DeepSeek 生成候选知识点"""
|
|||
|
|
# 估算生成数量
|
|||
|
|
text_len = len(text)
|
|||
|
|
expected_count = max(MIN_CANDIDATES, min(MAX_CANDIDATES, text_len // CHARS_PER_CANDIDATE))
|
|||
|
|
|
|||
|
|
prompt = _PROMPT.format(text=text[:16000]) # 限制上下文长度
|
|||
|
|
|
|||
|
|
async with httpx.AsyncClient(timeout=120) as client:
|
|||
|
|
resp = await client.post(
|
|||
|
|
f"{DEEPSEEK_BASE_URL}/chat/completions",
|
|||
|
|
headers={"Authorization": f"Bearer {DEEPSEEK_API_KEY}"},
|
|||
|
|
json={
|
|||
|
|
"model": DEEPSEEK_MODEL,
|
|||
|
|
"messages": [
|
|||
|
|
{"role": "system", "content": "你是一个专业的学习内容分析师。请始终返回有效的 JSON 数组。"},
|
|||
|
|
{"role": "user", "content": prompt},
|
|||
|
|
],
|
|||
|
|
"temperature": 0.3,
|
|||
|
|
"max_tokens": 4096,
|
|||
|
|
},
|
|||
|
|
)
|
|||
|
|
if resp.status_code != 200:
|
|||
|
|
raise RuntimeError(f"DeepSeek API error: {resp.status_code} {resp.text}")
|
|||
|
|
|
|||
|
|
data = resp.json()
|
|||
|
|
raw = data["choices"][0]["message"]["content"]
|
|||
|
|
|
|||
|
|
# 提取 JSON
|
|||
|
|
return _parse_json_response(raw, expected_count)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _parse_json_response(raw: str, expected_count: int) -> list[dict]:
|
|||
|
|
"""从 AI 回复中提取 JSON 数组"""
|
|||
|
|
# 尝试直接解析
|
|||
|
|
try:
|
|||
|
|
candidates = json.loads(raw)
|
|||
|
|
if isinstance(candidates, list):
|
|||
|
|
return candidates[:MAX_CANDIDATES]
|
|||
|
|
except json.JSONDecodeError:
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
# 提取 ```json ... ``` 块
|
|||
|
|
import re
|
|||
|
|
m = re.search(r"```(?:json)?\s*\n?(.*?)\n?```", raw, re.DOTALL)
|
|||
|
|
if m:
|
|||
|
|
try:
|
|||
|
|
candidates = json.loads(m.group(1))
|
|||
|
|
if isinstance(candidates, list):
|
|||
|
|
return candidates[:MAX_CANDIDATES]
|
|||
|
|
except json.JSONDecodeError:
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
# 提取 [ ... ] 块
|
|||
|
|
m = re.search(r"\[.*\]", raw, re.DOTALL)
|
|||
|
|
if m:
|
|||
|
|
try:
|
|||
|
|
candidates = json.loads(m.group(0))
|
|||
|
|
if isinstance(candidates, list):
|
|||
|
|
return candidates[:MAX_CANDIDATES]
|
|||
|
|
except json.JSONDecodeError:
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
raise ValueError(f"无法解析 AI 候选知识点回复: {raw[:500]}")
|