🤖 Retrieval Agent

Classic Retrieval

To retrieve the most relevant knowledge passages for a given query from a data srouce:

results = client.retrieve(query='What is RAG?', top_k=5, sources=['default'])

Retrieval Agent

To use our intelligent Retrieval Agent for retrieval:

results = client.retrieve(query='What is RAG?', top_k=5, sources=['default'], agent='on')

Adding OpenAI Key

To add an OpenAI key to the user account (for enabling the intelligent retrieval agent):

client.add_openai_key(key='YOUR_OPENAI_API_KEY')

To remove the added key:

client.remove_openai_key()