Add server support for chroma

protectai / rebuff

LLM Prompt Injection Detector

https://playground.rebuff.ai

Apache License 2.0

1.01k stars 67 forks source link

Add server support for chroma #66

Closed ristomcgehee closed 6 months ago

ristomcgehee commented 8 months ago

Before this PR can be merged it needs:

[x] https://github.com/protectai/rebuff/pull/63 to be merged
[x] Merge a minimal test case for chromadb Rebuff SDK instantiation (https://github.com/protectai/rebuff/pull/84)
[x] NPM release of rebuff
[x] Update server/package*

seanpmorgan commented 8 months ago

@ristomcgehee I'll publish 0.1.0 once we have a minimal test for chromadb SDK instantiation and querying. If you'd like to PR that lmk

ristomcgehee commented 8 months ago

@ristomcgehee I'll publish 0.1.0 once we have a minimal test for chromadb SDK instantiation and querying. If you'd like to PR that lmk

@seanpmorgan Yeah, I can make a PR for that. I'll be traveling for work all next week, so it might be a couple weeks till I get to it if that's all right. Since Chroma doesn't have a hosted version, should we spin up a Chroma server before running the tests, maybe with Docker Compose?

seanpmorgan commented 7 months ago

0.1.0 published: https://www.npmjs.com/package/rebuff

ristomcgehee commented 6 months ago

@seanpmorgan ready to be reviewed/merged