Update pipeline.py

pipeline.py

@@ -77,11 +77,13 @@ logger = logging.getLogger(__name__)
 from langchain_core.tracers import LangChainTracer
 from langsmith import Client
 
-os.environ["LANGCHAIN_TRACING"] = "true"
+
+os.environ["LANGCHAIN_TRACING_V2"]="true"
 os.environ["LANGSMITH_ENDPOINT"]="https://api.smith.langchain.com"
-langsmith_client = Client()
-os.environ["LANGSMITH_PROJECT"]="Healthy_ai_expert"
-tracer = LangChainTracer(project_name=os.environ.get("LANGCHAIN_PROJECT", "healthy_ai_expert"))
+# langsmith_client = Client()
+os.environ["LANGCHAIN_API_KEY"]=os.getenv("LANGCHAIN_API_KEY")
+os.environ["LANGCHAIN_PROJECT"]=os.getenv("LANGCHAIN_PROJECT")
+# tracer = LangChainTracer(project_name=os.environ.get("LANGCHAIN_PROJECT", "healthy_ai_expert"))
 
 
 
@@ -592,17 +594,17 @@ def is_ethical_conflict(query: str) -> bool:
 # Main Pipeline
 # -------------------------------------------------------
 def run_with_chain(query: str) -> str:
-    """
-    Overall flow:
-    1) Validate & rate-limit
-    2) Mistral moderation => 
-       - If self-harm => self_harm_chain
-       - If hate => refusal
-       - If violence/dangerous => we STILL produce a guided response (ethics) unless it's extreme
-    3) If not refused, check if query is aggression/ethical => route to chain
-    4) Otherwise classify => brand/wellness/out-of-scope => RAG => tailor
-    """
-    # with tracer.new_trace(name="wellness_pipeline_run") as run:
+        """
+        Overall flow:
+        1) Validate & rate-limit
+        2) Mistral moderation => 
+           - If self-harm => self_harm_chain
+           - If hate => refusal
+           - If violence/dangerous => we STILL produce a guided response (ethics) unless it's extreme
+        3) If not refused, check if query is aggression/ethical => route to chain
+        4) Otherwise classify => brand/wellness/out-of-scope => RAG => tailor
+        """
+        # with tracer.new_trace(name="wellness_pipeline_run") as run:
         start_time = time.time()
         try:
             # 1) Validate