sie-chroma

SIE integration for ChromaDB.

Installation

pip install sie-chroma

Features

SIEEmbeddingFunction: Custom embedding function for ChromaDB collections

Quick Start

Basic Usage

import chromadb
from sie_chroma import SIEEmbeddingFunction

# Create SIE embedding function
embedding_function = SIEEmbeddingFunction(
    base_url="http://localhost:8080",
    model="BAAI/bge-m3",
)

# Create ChromaDB client and collection
client = chromadb.Client()
collection = client.create_collection(
    name="my_collection",
    embedding_function=embedding_function,
)

# Add documents (embeddings are generated automatically)
collection.add(
    documents=[
        "Machine learning enables pattern recognition.",
        "Deep learning uses neural networks.",
        "Natural language processing analyzes text.",
    ],
    ids=["doc1", "doc2", "doc3"],
)

# Query the collection
results = collection.query(
    query_texts=["What is deep learning?"],
    n_results=2,
)
print(results["documents"])

With Persistent Storage

import chromadb
from sie_chroma import SIEEmbeddingFunction

# Persistent client
client = chromadb.PersistentClient(path="./chroma_data")

embedding_function = SIEEmbeddingFunction(
    base_url="http://localhost:8080",
    model="BAAI/bge-m3",
)

# Get or create collection
collection = client.get_or_create_collection(
    name="research_papers",
    embedding_function=embedding_function,
)

# Add documents with metadata
collection.add(
    documents=["Paper about transformers...", "Study on attention mechanisms..."],
    metadatas=[{"year": 2023}, {"year": 2024}],
    ids=["paper1", "paper2"],
)

# Query with metadata filtering
results = collection.query(
    query_texts=["attention in neural networks"],
    n_results=5,
    where={"year": {"$gte": 2023}},
)

With LangChain or LlamaIndex

The SIEEmbeddingFunction works with ChromaDB's LangChain and LlamaIndex integrations:

# LangChain
from langchain_chroma import Chroma
from sie_chroma import SIEEmbeddingFunction

embedding_function = SIEEmbeddingFunction(model="BAAI/bge-m3")
vectorstore = Chroma(
    collection_name="docs",
    embedding_function=embedding_function,  # Works directly!
)

# LlamaIndex
from llama_index.vector_stores.chroma import ChromaVectorStore

# SIE can also be used via LlamaIndex's SIEEmbedding

SIE Server

Start the SIE server before using this integration:

mise run serve -- -d cpu -p 8080

Testing

# Unit tests (no server required)
pytest

# Integration tests (requires running server)
pytest -m integration

Metadata

Release files for sie-chroma 0.8.2

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for sie-chroma 0.8.2
File	Size	Uploaded
sie_chroma-0.8.2.tar.gz	10.4 kB	Sep 23, 2026	Details

Built distribution (wheel)

Table of built distributions (wheels) for sie-chroma 0.8.2
File	Interpreter	ABI	Platform	Reset
sie_chroma-0.8.2-py3-none-any.whl 4.2 kB Sep 23, 2026	Python 3	none	any	Details

Total release size: 14.6 kB

Release files / sie_chroma-0.8.2.tar.gz

Download URL	sie_chroma-0.8.2.tar.gz
Size	10.4 kB
Tags	Source
SHA-256 checksum How to use checksums	`ee8a87a6dab931caf4c64b1d0552930e53c5a31d7d1d83ac4cbd3097fe9a8524`
BLAKE2b-256 checksum How to use checksums	`98570fca12a669b46913485951e01219787da6d969bb1d185feb150210593d8c`
Upload date	Sep 23, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	Yes
Uploaded via	`twine/7.0.0 CPython/3.13.14`

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

Identity

Publishing platform: GitHub Actions
Publishing repository: github.com/superlinked/sie
Publishing commit: github.com/superlinked/sie/tree/66daf371748b463f9f2bbe308ffb83f488f4cd8a

Workflow

Publishing configuration: github.com/superlinked/sie/blob/66daf371748b463f9f2bbe308ffb83f488f4cd8a/.github/workflows/release.yml
Publishing logs: github.com/superlinked/sie/actions/runs/35908401744/attempts/1

Artifact

Subject: sie_chroma-0.8.2.tar.gz
SHA-256 checksum: ee8a87a6dab931caf4c64b1d0552930e53c5a31d7d1d83ac4cbd3097fe9a8524 Verifying attestations

Release files / sie_chroma-0.8.2-py3-none-any.whl

Download URL	sie_chroma-0.8.2-py3-none-any.whl
Size	4.2 kB
Tags	Python 3
SHA-256 checksum How to use checksums	`e720bfcafa8dd1be99d901eb59aeb8b9849fd80813b191cc22548086fd648ca9`
BLAKE2b-256 checksum How to use checksums	`842b07477b3924647e54d9f7168b131054308845e417978a28d0ef39a8324265`
Upload date	Sep 23, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	Yes
Uploaded via	`twine/7.0.0 CPython/3.13.14`

Provenance

Identity

Publishing platform: GitHub Actions
Publishing repository: github.com/superlinked/sie
Publishing commit: github.com/superlinked/sie/tree/66daf371748b463f9f2bbe308ffb83f488f4cd8a

Workflow

Publishing configuration: github.com/superlinked/sie/blob/66daf371748b463f9f2bbe308ffb83f488f4cd8a/.github/workflows/release.yml
Publishing logs: github.com/superlinked/sie/actions/runs/35908401744/attempts/1

Artifact

Subject: sie_chroma-0.8.2-py3-none-any.whl
SHA-256 checksum: e720bfcafa8dd1be99d901eb59aeb8b9849fd80813b191cc22548086fd648ca9 Verifying attestations