Spaces:
Sleeping
Sleeping
Update clare_core.py
Browse files- clare_core.py +7 -3
clare_core.py
CHANGED
|
@@ -415,7 +415,7 @@ def build_messages(
|
|
| 415 |
course_outline: Optional[List[str]],
|
| 416 |
weaknesses: Optional[List[str]],
|
| 417 |
cognitive_state: Optional[Dict[str, int]],
|
| 418 |
-
rag_context: Optional[str] = None, #
|
| 419 |
) -> List[Dict[str, str]]:
|
| 420 |
messages: List[Dict[str, str]] = [
|
| 421 |
{"role": "system", "content": CLARE_SYSTEM_PROMPT}
|
|
@@ -538,7 +538,7 @@ def build_messages(
|
|
| 538 |
}
|
| 539 |
)
|
| 540 |
|
| 541 |
-
#
|
| 542 |
if rag_context:
|
| 543 |
messages.append(
|
| 544 |
{
|
|
@@ -584,7 +584,7 @@ def chat_with_clare(
|
|
| 584 |
course_outline: Optional[List[str]],
|
| 585 |
weaknesses: Optional[List[str]],
|
| 586 |
cognitive_state: Optional[Dict[str, int]],
|
| 587 |
-
rag_context: Optional[str] = None, #
|
| 588 |
) -> Tuple[str, List[Tuple[str, str]]]:
|
| 589 |
# 构建 messages
|
| 590 |
messages = build_messages(
|
|
@@ -641,6 +641,9 @@ def export_conversation(
|
|
| 641 |
|
| 642 |
|
| 643 |
# ---------- 生成 3 个 quiz 题目 ----------
|
|
|
|
|
|
|
|
|
|
| 644 |
def generate_quiz_from_history(
|
| 645 |
history: List[Tuple[str, str]],
|
| 646 |
course_outline: List[str],
|
|
@@ -711,6 +714,7 @@ def generate_quiz_from_history(
|
|
| 711 |
|
| 712 |
|
| 713 |
# ---------- 概念总结(知识点摘要) ----------
|
|
|
|
| 714 |
def summarize_conversation(
|
| 715 |
history: List[Tuple[str, str]],
|
| 716 |
course_outline: List[str],
|
|
|
|
| 415 |
course_outline: Optional[List[str]],
|
| 416 |
weaknesses: Optional[List[str]],
|
| 417 |
cognitive_state: Optional[Dict[str, int]],
|
| 418 |
+
rag_context: Optional[str] = None, # 新增:RAG 检索结果
|
| 419 |
) -> List[Dict[str, str]]:
|
| 420 |
messages: List[Dict[str, str]] = [
|
| 421 |
{"role": "system", "content": CLARE_SYSTEM_PROMPT}
|
|
|
|
| 538 |
}
|
| 539 |
)
|
| 540 |
|
| 541 |
+
# RAG 检索结果
|
| 542 |
if rag_context:
|
| 543 |
messages.append(
|
| 544 |
{
|
|
|
|
| 584 |
course_outline: Optional[List[str]],
|
| 585 |
weaknesses: Optional[List[str]],
|
| 586 |
cognitive_state: Optional[Dict[str, int]],
|
| 587 |
+
rag_context: Optional[str] = None, # 新增
|
| 588 |
) -> Tuple[str, List[Tuple[str, str]]]:
|
| 589 |
# 构建 messages
|
| 590 |
messages = build_messages(
|
|
|
|
| 641 |
|
| 642 |
|
| 643 |
# ---------- 生成 3 个 quiz 题目 ----------
|
| 644 |
+
from langsmith import traceable
|
| 645 |
+
|
| 646 |
+
@traceable(run_type="chain", name="generate_quiz_from_history")
|
| 647 |
def generate_quiz_from_history(
|
| 648 |
history: List[Tuple[str, str]],
|
| 649 |
course_outline: List[str],
|
|
|
|
| 714 |
|
| 715 |
|
| 716 |
# ---------- 概念总结(知识点摘要) ----------
|
| 717 |
+
@traceable(run_type="chain", name="summarize_conversation")
|
| 718 |
def summarize_conversation(
|
| 719 |
history: List[Tuple[str, str]],
|
| 720 |
course_outline: List[str],
|