SarahXia0405 commited on
Commit
b0e752a
·
verified ·
1 Parent(s): 4fabb3a

Update clare_core.py

Browse files
Files changed (1) hide show
  1. 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, # 👈 新增:RAG 检索结果
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
- # RAG 检索结果(如果有)
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],