import os
from mirage import MountMode, Workspace
from mirage.resource.chroma import ChromaConfig, ChromaResource
config = ChromaConfig(
host=os.environ.get("CHROMA_HOST", "localhost"),
port=int(os.environ.get("CHROMA_PORT", "8000")),
ssl=os.environ.get("CHROMA_SSL", "false").lower() == "true",
collection_name=os.environ["CHROMA_COLLECTION"],
slug_field=os.environ.get("CHROMA_SLUG_FIELD", "page_slug"),
chunk_index_field=os.environ.get("CHROMA_CHUNK_INDEX_FIELD", "chunk_index"),
)
resource = ChromaResource(config=config)
ws = Workspace({"/knowledge/": resource}, mode=MountMode.READ)