SarahXia0405 commited on
Commit
8a1fd1b
·
verified ·
1 Parent(s): aeb7491

Update clare_core.py

Browse files
Files changed (1) hide show
  1. clare_core.py +10 -1
clare_core.py CHANGED
@@ -563,8 +563,17 @@ def build_messages(
563
  messages.append({"role": "user", "content": user_message})
564
  return messages
565
 
 
 
 
 
 
 
 
 
 
 
566
 
567
- @traceable(run_type="chain", name="chat_with_clare")
568
  def chat_with_clare(
569
  message: str,
570
  history: List[Tuple[str, str]],
 
563
  messages.append({"role": "user", "content": user_message})
564
  return messages
565
 
566
+ # 装饰器
567
+ @traceable(
568
+ run_type="chain",
569
+ name="chat_with_clare",
570
+ metadata={
571
+ "learning_mode": lambda *args, **kwargs: kwargs.get("learning_mode"),
572
+ "language_preference": lambda *args, **kwargs: kwargs.get("language_preference"),
573
+ "doc_type": lambda *args, **kwargs: kwargs.get("doc_type"),
574
+ },
575
+ )
576
 
 
577
  def chat_with_clare(
578
  message: str,
579
  history: List[Tuple[str, str]],