protectai / rebuff

LLM Prompt Injection Detector
https://playground.rebuff.ai
Apache License 2.0
1.01k stars 67 forks source link

Add chroma tests #84

Closed ristomcgehee closed 7 months ago

ristomcgehee commented 7 months ago

This PR adds some basic tests for using Chroma as the vector store.

I pinned chromadb to 1.5.6 because I encountered the same issue as https://github.com/langchain-ai/langchain/issues/7260 on later versions.

I considered introducing Docker Compose since I'm running Chroma as a container, but I didn't think the project was quite complicated enough yet to merit Docker Compose.