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.0.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.0-py3-none-any.whl (5.7 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: zeno_chroma-1.0.0.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.0.tar.gz
Algorithm Hash digest
SHA256 909654992926ef3dd06dd64c1bcf7b13189de4e86b2f1d7c66ec4ebfffec8481
MD5 577ab067f4a73cb78c865e7d25fd6aa0
BLAKE2b-256 5ef4e26f12317134fdce481d545f2e171dafe0fefc97315c71c8ca620d7f6de0

See more details on using hashes here.

Provenance

The following attestation bundles were made for zeno_chroma-1.0.0.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.0-py3-none-any.whl.

File metadata

  • Download URL: zeno_chroma-1.0.0-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.0-py3-none-any.whl
Algorithm Hash digest
SHA256 4b95a40ceaadaa0ce40e39fb7a754026b9a2413024f47a07d2c0d87f11c31fda
MD5 0d01e805f00100dd838cd6997b519bf3
BLAKE2b-256 c4e56a31ce944d482f94b15937d218abe0311e246235685d5b8b497a92f8230c

See more details on using hashes here.

Provenance

The following attestation bundles were made for zeno_chroma-1.0.0-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