opentelemetry_aleph_alpha_client_instrumentation_quickstart.py

1from aleph_alpha_client import Client, CompletionRequest, Prompt
2from opentelemetry import trace
3from opentelemetry.instrumentation.alephalpha import AlephAlphaInstrumentor
4from opentelemetry.sdk.trace import TracerProvider
5from opentelemetry.sdk.trace.export import (
6    BatchSpanProcessor,
7    ConsoleSpanExporter,
8)
9
10# Initialize OpenTelemetry
11trace.set_tracer_provider(TracerProvider())
12trace.get_tracer_provider().add_span_processor(
13    BatchSpanProcessor(ConsoleSpanExporter())
14)
15
16# Instrument Aleph Alpha
17AlephAlphaInstrumentor().instrument()
18
19# Use the Aleph Alpha client as usual
20client = Client(token="YOUR_ALEPH_ALPHA_API_TOKEN")
21request = CompletionRequest(
22    prompt=Prompt.from_text("Explain quantum computing in one sentence."),
23    maximum_tokens=64,
24)
25response = client.complete(request, model="luminous-base")
26
27print(response.completions[0].completion)