import os
from dotenv import load_dotenv

def setup_observability():
    """
    Connects the application to LangSmith for professional tracing and 
    portfolio-level observability.
    """
    load_dotenv()
    
    # We enable tracing if the API key is present
    if os.getenv("LANGCHAIN_API_KEY"):
        os.environ["LANGCHAIN_TRACING_V2"] = "true"
        os.environ["LANGCHAIN_ENDPOINT"] = "https://api.smith.langchain.com"
        os.environ["LANGCHAIN_PROJECT"] = os.getenv("LANGCHAIN_PROJECT", "aeis-executive-researcher")
        print("🚀 Observability: LangSmith tracing enabled.")
    else:
        print("⚠️ Observability: LANGCHAIN_API_KEY not found. Tracing disabled.")