Update api/clare_core.py

api/clare_core.py

@@ -24,6 +24,12 @@ ENABLE_TRACING = os.getenv("CLARE_ENABLE_TRACING", "0").strip() == "1"
 if ENABLE_TRACING:
     from langsmith import traceable  # type: ignore
     from langsmith.run_helpers import set_run_metadata  # type: ignore
+
+    try:
+        # Available in newer langsmith versions
+        from langsmith.run_helpers import get_current_run_tree  # type: ignore
+    except Exception:
+        get_current_run_tree = None  # type: ignore
 else:
     # no-op decorators / funcs
     def traceable(*args, **kwargs):  # type: ignore
@@ -34,6 +40,8 @@ else:
     def set_run_metadata(**kwargs):  # type: ignore
         return None
 
+    get_current_run_tree = None  # type: ignore
+
 
 # ----------------------------
 # Speed knobs (simple + stable)
@@ -478,6 +486,28 @@ def model_name_or_default(x: str) -> str:
     return (x or "").strip() or DEFAULT_MODEL
 
 
+def get_langsmith_run_id() -> Optional[str]:
+    """
+    从 traceable 上下文里获取当前 run_id（用于把 UI feedback 挂到同一个 run 上）
+    若 tracing 关闭或环境不支持，则返回 None
+    """
+    if not ENABLE_TRACING:
+        return None
+    try:
+        if get_current_run_tree is None:
+            return None
+        rt = get_current_run_tree()
+        if not rt:
+            return None
+        rid = getattr(rt, "id", None)
+        if not rid:
+            return None
+        return str(rid)
+    except Exception as e:
+        print(f"[LangSmith get run id error] {repr(e)}")
+        return None
+
+
 @traceable(run_type="chain", name="chat_with_clare")
 def chat_with_clare(
     message: str,
@@ -490,7 +520,7 @@ def chat_with_clare(
     weaknesses: Optional[List[str]],
     cognitive_state: Optional[Dict[str, int]],
     rag_context: Optional[str] = None,
-) -> Tuple[str, List[Tuple[str, str]]]:
+) -> Tuple[str, List[Tuple[str, str]], Optional[str]]:
     # avoid any tracing overhead when disabled (set_run_metadata is no-op in that case)
     try:
         set_run_metadata(
@@ -523,8 +553,11 @@ def chat_with_clare(
         max_tokens=DEFAULT_MAX_OUTPUT_TOKENS,
     )
 
+    # Get run_id AFTER the run exists (works when tracing enabled; otherwise None)
+    run_id = get_langsmith_run_id()
+
     history = history + [(message, answer)]
-    return answer, history
+    return answer, history, run_id
 
 
 def export_conversation(