from praisonai_tools.observability import obs

# Auto-detect provider and auto-instrument all agents
obs.init()

# That's it! Everything below is now traced automatically
from praisonaiagents import Agent

agent = Agent(name="Assistant", instructions="You are helpful.")
agent.chat("Hello!")  # Auto-traced to your provider

from praisonai_tools.observability import obs

# Specify provider explicitly
obs.init(provider="langfuse")

# Or disable auto-instrumentation for manual control
obs.init(auto_instrument=False)

from praisonai_tools.observability import obs
from praisonai_tools.observability.base import SpanKind

obs.init(provider="langfuse")

# Create a trace for a workflow
with obs.trace("my-workflow", session_id="user-123"):
    # Create spans for individual operations
    with obs.span("llm-call", kind=SpanKind.LLM) as span:
        span.model = "gpt-4o-mini"
        span.input_tokens = 100
        span.output_tokens = 50
        # ... your LLM call
    
    with obs.span("tool-call", kind=SpanKind.TOOL) as span:
        span.tool_name = "search"
        # ... your tool call

obs.log_llm_call(
    model="gpt-4o-mini",
    input_messages="What is 2+2?",
    output="4",
    input_tokens=10,
    output_tokens=1,
)

obs.log_tool_call(
    tool_name="search",
    tool_input={"query": "PraisonAI"},
    tool_output={"results": [...]},
)

@obs.decorator("my-function", SpanKind.CUSTOM)
def my_function():
    return "result"

from praisonai_tools.observability import obs
from praisonai_tools.observability.base import SpanKind

obs.init(provider="langfuse")

with obs.trace("multi-agent-workflow"):
    # Agent 1
    with obs.span("agent-1", kind=SpanKind.AGENT) as span:
        span.attributes["agent_name"] = "Researcher"
        # ... agent 1 work
    
    # Agent 2 (child of same trace)
    with obs.span("agent-2", kind=SpanKind.AGENT) as span:
        span.attributes["agent_name"] = "Writer"
        # ... agent 2 work

# Health check
python -m praisonai_tools.observability.cli doctor

# Health check (JSON output)
python -m praisonai_tools.observability.cli doctor --json

# Verify traces in backend (LangSmith)
python -m praisonai_tools.observability.cli verify --provider langsmith --project "My First App"

# Also available via praisonai wrapper
praisonai obs doctor
praisonai obs verify --project "My First App"

# Check observability status
print(obs.doctor())
# Output: {'enabled': True, 'provider': 'langfuse', 'connection_status': True, ...}