Platform APIs
Neum Client
Interact with the Neum Cloud
The NeumClient
allows you to interact with the Neum AI Cloud through a python SDK. It supports the APIs listed below as methods in the client.
Initialize client
You will need a Neum AI Cloud API Key to initialize the client. Go to dashboard.neum.ai to create an account and get an API Key
Initialize NeumClient
from neumai.Client.NeumClient import NeumClient
neumClient = NeumClient(api_key = <INSERT NEUM AI KEY>)
Create pipeline
Configure your pipeline first using the Neum AI connectors.
Create pipeline
from neumai.Pipelines import Pipeline
pipeline = Pipeline(source=[...], embed=... , sink=...)
pipeline_id = neumClient.create_pipeline(pipeline=pipeline)
Get pipeline
Retrieve the pipeline configuration and latest run information:
Get pipelines
print(neumClient.get_pipeline(pipeline_id=pipeline_id))
Search pipeline
Query the data indexed using the pipeline configuration:
Create pipeline
print(neumClient.search_pipeline(pipeline_id=pipeline_id, query="What is ....", num_of_results=3, track=False, requested_by="John Doe"))
We also support search against a specific file within a pipeline using the search_file
method. Ex. if you ingested an S3 bucket with 10 files, you can pick one and only search against it.
Provide feedback on a retrieval
Capture feedback for a given search result. (Must have enabled retrieval tracking.)
Capture feedback
provide_retrieval_feedback(pipeline_id=pipeline_id, retrieval_id=retrieval_id, status="Add feedback ex. Good or Bad")
Get search retrievals for a pipeline
Query the retrieval events against a pipeline (Must have enabled retrieval tracking.)
Get retrievals for pipeline
print(neumClient.get_retrievals_by_pipeline_id(pipeline_id=pipeline_id))
Additional methods provided
trigger_pipeline
: Trigger a pipeline to runget_pipelines
: Get a list of all pipelines created by a userget_pipeline_runs
: Get a list of all the runs for a given pipelineget_pipeline_run
: Get a specific run for a pipelineget_files
: Get that statuses for all files available within a pipelineget_file
: Get the status for a specific file within a pipelineget_retrievals_by_file_id
: Get retrieval events for a specific file within a pipeline. (Must have tracking enabled)get_retrievals_by_pipeline_id_user_id
: Get retrieval events for a user within a pipeline. (Must have tracking enabled and passed arequested_by
parameter)get_retrievals_by_file_id_user_id
: Get retrieval events for a user within a file. (Must have tracking enabled and passed arequested_by
parameter)get_retrievals_by_user_id
: Get retrieval events for a user across pipelines. (Must have tracking enabled and passed arequested_by
parameter)
Was this page helpful?