from opentelemetry import trace

from opentelemetry.sdk.trace import TracerProvider

from opentelemetry.sdk.trace.export import (
    SimpleSpanProcessor,
    ConsoleSpanExporter,
)

from exporter import SQLiteSpanExporter

provider = TracerProvider()

provider.add_span_processor(
    SimpleSpanProcessor(
        SQLiteSpanExporter("traces.db")
    )
)

trace.set_tracer_provider(provider)

tracer = trace.get_tracer("llm-zoomcamp")