Skip to main content

ChromaDB-backed KnowledgeStore adapter for Zeno.

Project description

zeno-chroma

ChromaDB-backed KnowledgeStore adapter for Zeno.

Persists facts and documents in a local Chroma collection with each row stamped by user_id so multi-tenant isolation is enforced at query time via Chroma's where filter rather than post-fetch filtering. Plugs into the ThreeLayer memory binder as the knowledge and (optionally) the user_memory backing.

Install

uv add 'zeno-framework[chroma]'
# or, without the AI package:
uv add zeno-chroma

Features

  • Persistent local store via chromadb.PersistentClient — no server to run, just a directory
  • user_id-scoped search using Chroma's where filter (S7 isolation is structural, not advisory)
  • Cosine similarity, distance → similarity score conversion (1.0 - distance, clamped to [0, 1])
  • Synchronous Chroma client wrapped in asyncio.to_thread so embeddings and queries don't block the event loop
  • Default Chroma embedding function; pass embedding_function= to override
  • Pairs with VectorBackedUserMemoryStore to also serve user-memory reads/writes from the same collection

Minimal usage

import asyncio
from pathlib import Path

from zeno.agent import Agent
from zeno.app import ZenoApp
from zeno.builtin_tools import recall, remember
from zeno.channels.cli import CliChannel
from zeno.chroma import ChromaKnowledgeStore
from zeno.memory.sqlite.conversation_store import SqliteConversationStore
from zeno.memory.sqlite.session_store import SqliteSessionStore
from zeno.memory.three_layer import ThreeLayer
from zeno.memory.vector_backed.user_memory_store import VectorBackedUserMemoryStore
from zeno.testing import FakeProvider

async def main() -> None:
    data_dir = Path("./.zeno")
    data_dir.mkdir(exist_ok=True)

    knowledge = ChromaKnowledgeStore(persist_dir=data_dir / "chroma")
    memory = ThreeLayer(
        session=SqliteSessionStore(data_dir / "sessions.db"),
        user_memory=VectorBackedUserMemoryStore(knowledge_store=knowledge),
        knowledge=knowledge,
        conversation=SqliteConversationStore(data_dir / "conversations.db"),
    )

    app = ZenoApp(
        agent=Agent(
            name="root",
            instructions="Use `recall` to look things up and `remember` to store facts.",
            tools=[recall, remember],
        ),
        channels=[CliChannel()],
        provider=FakeProvider(),
        memory=memory,
    )
    await app.run()

asyncio.run(main())

The first call to remember writes a row tagged with the active user_id. Subsequent recall calls only see rows that user inserted — cross-user isolation is enforced by the where={"user_id": ...} filter on every query.

Testing

uv run pytest packages/zeno-chroma

Tests run against a temporary Chroma directory so no global state leaks between cases. The contract suite shared with zeno-qdrant guarantees both adapters honour the same KnowledgeStore semantics.

Part of the Zeno framework.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

zeno_chroma-1.0.1.tar.gz (5.2 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

zeno_chroma-1.0.1-py3-none-any.whl (5.7 kB view details)

Uploaded Python 3

File details

Details for the file zeno_chroma-1.0.1.tar.gz.

File metadata

  • Download URL: zeno_chroma-1.0.1.tar.gz
  • Upload date:
  • Size: 5.2 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for zeno_chroma-1.0.1.tar.gz
Algorithm Hash digest
SHA256 cd5f490e512054a7347c95edbd7d8cb3e909fdc7130644387fb0809eb7ee9eb3
MD5 7c9b320c54f7ccf904ebcdfc14b3c484
BLAKE2b-256 15793048fb8dcebd68c26a176a162a16ff6053c9efc3e3be163d5448f28ce9e0

See more details on using hashes here.

Provenance

The following attestation bundles were made for zeno_chroma-1.0.1.tar.gz:

Publisher: publish.yml on nkootstra/zeno

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file zeno_chroma-1.0.1-py3-none-any.whl.

File metadata

  • Download URL: zeno_chroma-1.0.1-py3-none-any.whl
  • Upload date:
  • Size: 5.7 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for zeno_chroma-1.0.1-py3-none-any.whl
Algorithm Hash digest
SHA256 8e6d086f6fc6fcbd51ce3934254c5372fd4b67ee64d961829f6e77cc41b05431
MD5 68c9f3a416944c432c218ef60e49b82e
BLAKE2b-256 8bf9d708bbc1e67d3f272656127e57f87b52672b6be64d8f7774ecab0402a39d

See more details on using hashes here.

Provenance

The following attestation bundles were made for zeno_chroma-1.0.1-py3-none-any.whl:

Publisher: publish.yml on nkootstra/zeno

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page