mirror of
https://github.com/placeholder-soft/chroma.git
synced 2026-01-12 22:44:55 +08:00
896822231e
## Description of changes *Summarize the changes made by this PR.* - New functionality - Adds a basic pulsar producer, consumer and associated tests. As well as a docker compose for the distributed version of chroma. ## Test plan We added bin/cluster-test.sh, which starts pulsar and allows test_producer_consumer to run the pulsar fixture. ## Documentation Changes None required.
67 lines
1.4 KiB
YAML
67 lines
1.4 KiB
YAML
# This docker compose file is not meant to be used. It is a work in progress
|
|
# for the distributed version of Chroma. It is not yet functional.
|
|
|
|
version: '3.9'
|
|
|
|
networks:
|
|
net:
|
|
driver: bridge
|
|
|
|
services:
|
|
server:
|
|
image: server
|
|
build:
|
|
context: .
|
|
dockerfile: Dockerfile
|
|
volumes:
|
|
- ./:/chroma
|
|
- index_data:/index_data
|
|
command: uvicorn chromadb.app:app --reload --workers 1 --host 0.0.0.0 --port 8000 --log-config log_config.yml
|
|
environment:
|
|
- IS_PERSISTENT=TRUE
|
|
- CHROMA_PRODUCER_IMPL=chromadb.ingest.impl.pulsar.PulsarProducer
|
|
- CHROMA_CONSUMER_IMPL=chromadb.ingest.impl.pulsar.PulsarConsumer
|
|
- PULSAR_BROKER_URL=pulsar
|
|
- PULSAR_BROKER_PORT=6650
|
|
- PULSAR_ADMIN_PORT=8080
|
|
ports:
|
|
- 8000:8000
|
|
depends_on:
|
|
pulsar:
|
|
condition: service_healthy
|
|
networks:
|
|
- net
|
|
|
|
pulsar:
|
|
image: apachepulsar/pulsar
|
|
volumes:
|
|
- pulsardata:/pulsar/data
|
|
- pulsarconf:/pulsar/conf
|
|
command: bin/pulsar standalone
|
|
ports:
|
|
- 6650:6650
|
|
- 8080:8080
|
|
networks:
|
|
- net
|
|
healthcheck:
|
|
test:
|
|
[
|
|
"CMD",
|
|
"curl",
|
|
"-f",
|
|
"localhost:8080/admin/v2/brokers/health"
|
|
]
|
|
interval: 3s
|
|
timeout: 1m
|
|
retries: 10
|
|
|
|
volumes:
|
|
index_data:
|
|
driver: local
|
|
backups:
|
|
driver: local
|
|
pulsardata:
|
|
driver: local
|
|
pulsarconf:
|
|
driver: local
|