Qdrant

pip install qdrant-client langtrace-python-sdk

from langtrace_python_sdk import langtrace, with_langtrace_root_span
from qdrant_client import QdrantClient

# Initialize Qdrant client
qdrant_client = QdrantClient("localhost", port=6333)

# Initialize Langtrace
langtrace.init(api_key='your_langtrace_api_key_here')

@with_langtrace_root_span("add_points")
def add_points_to_collection(collection_name, points):
    qdrant_client.upsert(
        collection_name=collection_name,
        points=points
    )

@with_langtrace_root_span("search")
def search_collection(collection_name, query_vector, limit=10):
    return qdrant_client.search(
        collection_name=collection_name,
        query_vector=query_vector,
        limit=limit
    )

@with_langtrace_root_span("create_collection")
def create_collection(collection_name, vector_size):
    qdrant_client.recreate_collection(
        collection_name=collection_name,
        vectors_config=models.VectorParams(size=vector_size, distance=models.Distance.COSINE)
    )

   @with_langtrace_root_span("qdrant_operation")
   def qdrant_operation():
       try:
           # Qdrant operation here
           pass
       except Exception as e:
           add_attribute_to_current_span("error", str(e))
           raise