protectai / rebuff

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

You haven't specified an API Key #107

Closed rodrigo-pedro closed 5 months ago

rodrigo-pedro commented 6 months ago

Running on master commit 99e2f44a100d66b39e225dc967a26eea3b791b95.

File "/venv/lib/python3.10/site-packages/rebuff/sdk.py", line 83, in detect_injection self.initialize_pinecone() File "/venv/lib/python3.10/site-packages/rebuff/sdk.py", line 40, in initialize_pinecone self.vector_store = init_pinecone( File "/venv/lib/python3.10/site-packages/rebuff/detect_pi_vectorbase.py", line 72, in init_pinecone vector_store = Pinecone.from_existing_index( File "/venv/lib/python3.10/site-packages/langchain_community/vectorstores/pinecone.py", line 462, in from_existing_index pinecone_index = cls.get_pinecone_index(index_name, pool_threads) File "/venv/lib/python3.10/site-packages/langchain_community/vectorstores/pinecone.py", line 369, in get_pinecone_index pinecone_instance = pinecone.Pinecone( File "/venv/lib/python3.10/site-packages/pinecone/control/pinecone.py", line 95, in init self.config = PineconeConfig.build(api_key=api_key, host=host, additional_headers=additional_headers, kwargs) File "/venv/lib/python3.10/site-packages/pinecone/config/pinecone_config.py", line 12, in build return ConfigBuilder.build(api_key=api_key, host=host, additional_headers=additional_headers, kwargs) File "/venv/lib/python3.10/site-packages/pinecone/config/config.py", line 45, in build raise PineconeConfigurationError("You haven't specified an Api-Key.") pinecone.exceptions.PineconeConfigurationError: You haven't specified an Api-Key.

My code:

rb = RebuffSdk( openai_apikey=openai_apikey, pinecone_apikey=pinecone_apikey, pinecone_index=pinecone_index, openai_model=openai_model )

rb.detect_injection(prompt)

ristomcgehee commented 6 months ago

Thanks for surfacing another issue. PR to fix in https://github.com/protectai/rebuff/pull/108.