protectai / rebuff

LLM Prompt Injection Detector

Home Page:https://playground.rebuff.ai

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

You haven't specified an API Key

rodrigo-pedro opened this issue · comments

Running on master commit 99e2f44.

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)

Thanks for surfacing another issue. PR to fix in #108.