opentelemetry_milvus_instrumentation_console_trace_export_quickstart.py

1from pymilvus import connections, CollectionSchema, FieldSchema, DataType, Collection
2from opentelemetry import trace
3from opentelemetry.sdk.trace import TracerProvider
4from opentelemetry.sdk.trace.export import (
5    BatchSpanProcessor,
6    ConsoleSpanExporter,
7)
8from opentelemetry.instrumentation.milvus import MilvusInstrumentor
9
10# 1. Setup OpenTelemetry tracing
11provider = TracerProvider()
12processor = BatchSpanProcessor(ConsoleSpanExporter())
13provider.add_span_processor(processor)
14trace.set_tracer_provider(provider)
15
16# 2. Instrument Milvus
17MilvusInstrumentor().instrument()
18
19# 3. Use Milvus as usual (traces will be automatically generated)
20connections.connect("default", host="localhost", port="19530")
21
22fields = [
23    FieldSchema(name="pk", dtype=DataType.INT64, is_primary=True, auto_id=False),
24    FieldSchema(name="embeddings", dtype=DataType.FLOAT_VECTOR, dim=8)
25]
26schema = CollectionSchema(fields, "Example collection")
27collection = Collection("example_collection", schema)
28
29# This operation will generate a trace
30collection.insert([[1], [[0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8]]])
31
32connections.disconnect("default")