openinference-instrumentation-mistralai

import os from opentelemetry import trace from opentelemetry.sdk.trace import TracerProvider from opentelemetry.sdk.trace.export import SimpleSpanProcessor from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter from openinference_instrumentation_mistralai import MistralAIInstrumentor # Set up OpenTelemetry (example with OTLP HTTP exporter) provider = TracerProvider() provider.add_span_processor(SimpleSpanProcessor(OTLPSpanExporter(endpoint="https://api.arize.com/v1/traces"))) trace.set_tracer_provider(provider) # Instrument MistralAI MistralAIInstrumentor().instrument() # Now use MistralAI as usual from mistralai import Mistral client = Mistral(api_key=os.environ.get("MISTRAL_API_KEY", "")) response = client.chat.complete(model="mistral-small-latest", messages=[{"role": "user", "content": "Hello"}]) print(response.choices[0].message.content)