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.10.1

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.10.1
File	Size	Uploaded
sie_chroma-0.10.1.tar.gz	10.9 kB	Oct 13, 2026	Details

Built distribution (wheel)

Table of built distributions (wheels) for sie-chroma 0.10.1
File	Interpreter	ABI	Platform	Reset
sie_chroma-0.10.1-py3-none-any.whl 4.4 kB Oct 13, 2026	Python 3	none	any	Details

Total release size: 15.3 kB

Release files / sie_chroma-0.10.1.tar.gz

Download URL	sie_chroma-0.10.1.tar.gz
Size	10.9 kB
Tags	Source
SHA-256 checksum How to use checksums	`8b1ffe331135ce65533be6dd4df8d2128c675a8b856abd2f6241bece32253e84`
BLAKE2b-256 checksum How to use checksums	`ac5feb45efff4e8cbc113123eed6b9db68cc846670f5122b3040c8c698363a63`
Upload date	Oct 13, 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/93325c4be4729f6d89ef29271c8e206a2af5f0ac

Workflow

Publishing configuration: github.com/superlinked/sie/blob/93325c4be4729f6d89ef29271c8e206a2af5f0ac/.github/workflows/release.yml
Publishing logs: github.com/superlinked/sie/actions/runs/38482618763/attempts/1

Artifact

Subject: sie_chroma-0.10.1.tar.gz
SHA-256 checksum: 8b1ffe331135ce65533be6dd4df8d2128c675a8b856abd2f6241bece32253e84 Verifying attestations

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

Download URL	sie_chroma-0.10.1-py3-none-any.whl
Size	4.4 kB
Tags	Python 3
SHA-256 checksum How to use checksums	`d82df21417d22cd0edfaec7121d1455d563761805319f9ff4bb049648a039972`
BLAKE2b-256 checksum How to use checksums	`c1e5c2c851b8cec29198bd3ad6b22e3d7004de88a2ff9442ccef61b73d3c5369`
Upload date	Oct 13, 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/93325c4be4729f6d89ef29271c8e206a2af5f0ac

Workflow

Publishing configuration: github.com/superlinked/sie/blob/93325c4be4729f6d89ef29271c8e206a2af5f0ac/.github/workflows/release.yml
Publishing logs: github.com/superlinked/sie/actions/runs/38482618763/attempts/1

Artifact

Subject: sie_chroma-0.10.1-py3-none-any.whl
SHA-256 checksum: d82df21417d22cd0edfaec7121d1455d563761805319f9ff4bb049648a039972 Verifying attestations