xtrace_sdk.utils package

Submodules

xtrace_sdk.utils.config module

xtrace_sdk.utils.embedding module

class xtrace_sdk.utils.embedding.Embedding(model='mixedbread-ai/mxbai-embed-large-v1', dim=512)

Bases: object

This class is a simple wrapper around the sentence transformer library to binarize embeddings

bin_embed(query)

Given a query, return the binary embedding

Parameters:

query (str) – query

Returns:

binary embedding

Return type:

np.array

dumps(chunks, fname)

hamming(q1, q2)

Given two queries, return the hamming distance between them

Parameters:

• q1 (str) – query 1

• q2 (str) – query 2

Returns:

hamming distance between q1 and q2

Return type:

int

static real2bin(emb)

static sim_hash(emb)

xtrace_sdk.utils.types module

Module contents